conduct-cli 0.4.10__tar.gz → 0.4.12__tar.gz

{conduct_cli-0.4.10 → conduct_cli-0.4.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conduct-cli
-Version: 0.4.10
+Version: 0.4.12
 Summary: CLI for Conduct AI — install agents, manage projects, run tests
 Author-email: Conduct AI <hello@conductai.ai>
 License: MIT

{conduct_cli-0.4.10 → conduct_cli-0.4.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "conduct-cli"
-version = "0.4.10"
+version = "0.4.12"
 description = "CLI for Conduct AI — install agents, manage projects, run tests"
 readme = "README.md"
 license = { text = "MIT" }

{conduct_cli-0.4.10 → conduct_cli-0.4.12}/src/conduct_cli/guard.py

@@ -198,13 +198,30 @@ def _post_usage(session_id, tool_name, tokens_input, tokens_output, duration_ms)
     )
 
 
+def _tail_lines(path, n=200):
+    """Read last n lines of a file efficiently without loading the whole file."""
+    size = path.stat().st_size
+    if size == 0:
+        return []
+    chunk = min(size, n * 300)  # ~300 bytes per line estimate
+    with open(path, "rb") as f:
+        f.seek(max(0, size - chunk))
+        raw = f.read()
+    text = raw.decode("utf-8", errors="ignore")
+    lines = text.splitlines()
+    # If we didn't seek to start, first line may be partial — drop it
+    if size > chunk:
+        lines = lines[1:]
+    return lines
+
+
 def _read_tokens_from_transcript(transcript_path, tool_use_id):
-    """Scan transcript JSONL backwards for the message that produced tool_use_id."""
+    """Read token counts from Claude Code transcript (matched by tool_use_id)."""
     try:
         path = Path(transcript_path)
         if not path.exists() or not tool_use_id:
             return 0, 0
-        lines = path.read_text(encoding="utf-8", errors="ignore").splitlines()
+        lines = _tail_lines(path)
         for line in reversed(lines):
             if not line.strip() or "tool_use" not in line:
                 continue
@@ -232,6 +249,35 @@ def _read_tokens_from_transcript(transcript_path, tool_use_id):
     return 0, 0
 
 
+def _read_codex_tokens():
+    """Read last_token_usage from the most recently modified Codex session file."""
+    try:
+        sessions_dir = Path.home() / ".codex" / "sessions"
+        if not sessions_dir.exists():
+            return 0, 0
+        files = sorted(sessions_dir.rglob("*.jsonl"), key=lambda p: p.stat().st_mtime, reverse=True)
+        if not files:
+            return 0, 0
+        lines = _tail_lines(files[0])
+        for line in reversed(lines):
+            if "token_count" not in line:
+                continue
+            try:
+                entry = json.loads(line)
+                if entry.get("type") == "event_msg":
+                    info = entry.get("payload", {}).get("info", {})
+                    usage = info.get("last_token_usage", {})
+                    if usage:
+                        total_in  = usage.get("input_tokens", 0)
+                        total_out = usage.get("output_tokens", 0) + usage.get("reasoning_output_tokens", 0)
+                        return total_in, total_out
+            except Exception:
+                continue
+    except Exception:
+        pass
+    return 0, 0
+
+
 def post_usage_main():
     """PostToolUse hook entrypoint — captures token usage and duration."""
     try:
@@ -242,7 +288,10 @@ def post_usage_main():
     tool_name       = (data.get("tool_name") or "").lower()
     tool_use_id     = data.get("tool_use_id")
     transcript_path = data.get("transcript_path")
-    tokens_input, tokens_output = _read_tokens_from_transcript(transcript_path, tool_use_id)
+    if transcript_path:
+        tokens_input, tokens_output = _read_tokens_from_transcript(transcript_path, tool_use_id)
+    else:
+        tokens_input, tokens_output = _read_codex_tokens()
     _post_usage(session_id, tool_name, tokens_input, tokens_output, None)
     sys.exit(0)
 

{conduct_cli-0.4.10 → conduct_cli-0.4.12}/src/conduct_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conduct-cli
-Version: 0.4.10
+Version: 0.4.12
 Summary: CLI for Conduct AI — install agents, manage projects, run tests
 Author-email: Conduct AI <hello@conductai.ai>
 License: MIT