Skip to content

Parse "Kept model as" and Claude Code 2.1.x transcript format #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
egerev wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Parse "Kept model as" and Claude Code 2.1.x transcript format #5

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bc3ecf1
feat: parse "Kept model as" and Claude Code 2.1.x transcript format
egerev Apr 17, 2026
042849e
fix: parse Claude effort command output
egerev Apr 24, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 24 additions & 5 deletions status_line.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -204,13 +204,22 @@ def _get_env(key: str) -> str | None:

# Matches: "Set model to ... with <effort> effort" after stripping ANSI
_MODEL_SET_RE = re.compile(r"Set model to .+ with (\w+) effort")
# Also matches: "Kept model as ... with <effort> effort"
_MODEL_KEPT_RE = re.compile(r"Kept model as .+ with (\w+) effort")
# Matches: "Set effort level to max (this session only): ..."
_EFFORT_SET_RE = re.compile(r"Set effort (?:level )?to ([\w-]+)\b")


def _effort_from_transcript(transcript_path: str | None) -> str | None:
"""Parse the most recent /model command output from the session transcript.

The transcript is JSONL. /model writes entries with content like:
The transcript is JSONL. /model and /effort write entries in a few formats:
- assistant message: entry["message"]["content"] (old format)
- system local_command: entry["content"] (new format since ~2.1.x)

Content looks like:
"<local-command-stdout>Set model to ... with max effort</local-command-stdout>"
"<local-command-stdout>Set effort level to max (this session only): ...</local-command-stdout>"

Strategy: grep finds JSONL lines with the right tag, then we JSON-parse
the last match and extract effort from the decoded content string.
Expand All @@ -221,9 +230,9 @@ def _effort_from_transcript(transcript_path: str | None) -> str | None:
p = Path(transcript_path)
if not p.exists():
return None
# Find JSONL lines that are actual /model output (not code/comments)
# Find JSONL lines that are actual /model or /effort output (not code/comments).
result = subprocess.run(
["grep", "local-command-stdout>Set model to", str(p)],
["grep", "-E", "local-command-stdout>(Set model to|Kept model as|Set effort)", str(p)],
capture_output=True, text=True, timeout=2,
)
if result.returncode != 0 or not result.stdout.strip():
Expand All @@ -233,12 +242,22 @@ def _effort_from_transcript(transcript_path: str | None) -> str | None:
for line in reversed(lines):
try:
entry = json.loads(line)
content = entry.get("message", {}).get("content", "")
# New format: system entries have "content" directly
# Old format: assistant entries have "message.content"
content = entry.get("content", "")
if not content or not isinstance(content, str):
msg = entry.get("message")
if isinstance(msg, dict):
content = msg.get("content", "")
if not isinstance(content, str):
continue
# Content is decoded — ANSI escapes are real bytes now
clean = re.sub(r"\x1b\[[0-9;]*m", "", content)
m = _MODEL_SET_RE.search(clean)
m = (
_MODEL_SET_RE.search(clean)
or _MODEL_KEPT_RE.search(clean)
or _EFFORT_SET_RE.search(clean)
)
if m:
return m.group(1)
except (json.JSONDecodeError, AttributeError):
Expand Down