inscript 0.4.0__tar.gz → 0.5.0__tar.gz

{inscript-0.4.0 → inscript-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inscript
-Version: 0.4.0
+Version: 0.5.0
 Summary: Universal agent activity ledger. Records what AI agents do.
 Author: Andrew Park
 License-Expression: MIT

{inscript-0.4.0 → inscript-0.5.0}/inscript.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inscript
-Version: 0.4.0
+Version: 0.5.0
 Summary: Universal agent activity ledger. Records what AI agents do.
 Author: Andrew Park
 License-Expression: MIT

{inscript-0.4.0 → inscript-0.5.0}/inscript_pkg/__init__.py

@@ -17,7 +17,7 @@ import time
 from datetime import datetime, timezone
 from pathlib import Path
 
-__version__ = "0.4.0"
+__version__ = "0.5.0"
 
 INSCRIPT_DIR = Path.home() / ".inscript"
 ACTIVE_PROJECT_FILE = INSCRIPT_DIR / "active_project"
@@ -86,6 +86,15 @@ def session_dir(session_id: str) -> Path:
     return SESSIONS_DIR / session_id
 
 
+def _format_tokens(n: int) -> str:
+    """Format token count: 1234 -> 1.2k, 1234567 -> 1.2M."""
+    if n >= 1_000_000:
+        return f"{n / 1_000_000:.1f}M"
+    elif n >= 1_000:
+        return f"{n / 1_000:.1f}k"
+    return str(n)
+
+
 def list_sessions() -> list[dict]:
     sessions = []
     if not SESSIONS_DIR.exists():
@@ -279,7 +288,20 @@ def _cmd_log(session_id: str | None):
             except json.JSONDecodeError:
                 pass
 
-    print(f"  {len(files_seen)} files, {edits} edits, {len(prompts)} prompts")
+    # Show token usage if summary exists
+    summary_file = sdir / "summary.json"
+    token_str = ""
+    if summary_file.exists():
+        try:
+            s = json.loads(summary_file.read_text())
+            tokens = s.get("tokens")
+            if tokens:
+                total = tokens.get("total_tokens", 0)
+                token_str = f", {_format_tokens(total)} tokens"
+        except (json.JSONDecodeError, OSError):
+            pass
+
+    print(f"  {len(files_seen)} files, {edits} edits, {len(prompts)} prompts{token_str}")
 
 
 def _cmd_tag(tag_name: str | None):
@@ -529,10 +551,18 @@ def _cmd_export(session_id: str):
     if summary_file.exists():
         s = json.loads(summary_file.read_text())
         print(f"- **Ended**: {s.get('end_time', '?')}")
+        if s.get("duration_seconds"):
+            print(f"- **Duration**: {_format_duration(s['duration_seconds'])}")
         print(f"- **Prompts**: {s.get('prompts', '?')}")
         print(f"- **Files read**: {s.get('files_read', '?')}")
         print(f"- **Files written**: {s.get('files_written', '?')}")
         print(f"- **Total edits**: {s.get('total_edits', '?')}")
+        tokens = s.get("tokens")
+        if tokens:
+            print(f"- **Model**: {tokens.get('model', '?')}")
+            print(f"- **Tokens**: {_format_tokens(tokens.get('total_tokens', 0))} total ({_format_tokens(tokens.get('input_tokens', 0))} in, {_format_tokens(tokens.get('output_tokens', 0))} out)")
+            if tokens.get("cache_read_tokens"):
+                print(f"- **Cache**: {_format_tokens(tokens['cache_read_tokens'])} read, {_format_tokens(tokens.get('cache_write_tokens', 0))} written")
 
     # Load prompts, touches, and diffs
     prompts = _load_prompts(sdir)

{inscript-0.4.0 → inscript-0.5.0}/inscript_pkg/hook.py

@@ -137,6 +137,12 @@ def handle_session_start(data: dict) -> None:
         "project": proj,
         "status": "active",
     }
+
+    # Save transcript path if provided (for token counting on Stop)
+    transcript_path = data.get("transcript_path")
+    if transcript_path:
+        meta["transcript_path"] = transcript_path
+
     (sdir / "meta.json").write_text(json.dumps(meta, indent=2))
 
     # Mark as active session
@@ -216,17 +222,23 @@ def handle_post_tool_use(data: dict) -> None:
 
     sdir = SESSIONS_DIR / session_id
 
-    # Update session meta with project if it changed
-    if root:
-        meta_file = sdir / "meta.json"
-        if meta_file.exists():
-            try:
-                meta = json.loads(meta_file.read_text())
-                if meta.get("project") != str(root):
-                    meta["project"] = str(root)
-                    meta_file.write_text(json.dumps(meta, indent=2))
-            except (json.JSONDecodeError, OSError):
-                pass
+    # Update session meta with project + transcript_path if needed
+    meta_file = sdir / "meta.json"
+    if meta_file.exists():
+        try:
+            meta = json.loads(meta_file.read_text())
+            changed = False
+            if root and meta.get("project") != str(root):
+                meta["project"] = str(root)
+                changed = True
+            transcript_path = data.get("transcript_path")
+            if transcript_path and not meta.get("transcript_path"):
+                meta["transcript_path"] = transcript_path
+                changed = True
+            if changed:
+                meta_file.write_text(json.dumps(meta, indent=2))
+        except (json.JSONDecodeError, OSError):
+            pass
 
     # Relativize path for readability
     display_path = file_path
@@ -314,6 +326,52 @@ def _check_overlap(current_session: str, project: str, file_path: str) -> None:
 # Stop handler
 # ---------------------------------------------------------------------------
 
+def _compute_token_usage(transcript_path: str) -> dict | None:
+    """Read a Claude Code transcript JSONL and sum token usage."""
+    try:
+        tp = Path(transcript_path)
+        if not tp.exists():
+            return None
+
+        total_input = 0
+        total_output = 0
+        total_cache_read = 0
+        total_cache_write = 0
+        model = None
+
+        for line in tp.open():
+            try:
+                d = json.loads(line)
+                msg = d.get("message", {})
+                usage = msg.get("usage")
+                if not usage:
+                    continue
+
+                if model is None:
+                    model = msg.get("model")
+
+                total_input += usage.get("input_tokens", 0)
+                total_output += usage.get("output_tokens", 0)
+                total_cache_read += usage.get("cache_read_input_tokens", 0)
+                total_cache_write += usage.get("cache_creation_input_tokens", 0)
+            except json.JSONDecodeError:
+                continue
+
+        if total_input == 0 and total_output == 0:
+            return None
+
+        return {
+            "model": model,
+            "input_tokens": total_input,
+            "output_tokens": total_output,
+            "cache_read_tokens": total_cache_read,
+            "cache_write_tokens": total_cache_write,
+            "total_tokens": total_input + total_output,
+        }
+    except (OSError, Exception):
+        return None
+
+
 def handle_stop(data: dict) -> None:
     """Finalize the current session with a summary."""
     session_id = active_session()
@@ -343,13 +401,18 @@ def handle_stop(data: dict) -> None:
             except json.JSONDecodeError:
                 pass
 
-    # Read start time from meta
+    # Read start time and transcript path from meta
     meta_file = sdir / "meta.json"
     start_time = None
+    transcript_path = None
     if meta_file.exists():
         try:
             meta = json.loads(meta_file.read_text())
             start_time = meta.get("start_time")
+            transcript_path = meta.get("transcript_path")
+            # Also check the Stop hook data for transcript_path
+            if not transcript_path:
+                transcript_path = data.get("transcript_path")
             meta["status"] = "completed"
             meta_file.write_text(json.dumps(meta, indent=2))
         except (json.JSONDecodeError, OSError):
@@ -369,7 +432,7 @@ def handle_stop(data: dict) -> None:
     prompts_file = sdir / "prompts.jsonl"
     prompt_count = _count_lines(prompts_file)
 
-    summary = {
+    summary: dict = {
         "end_time": end_time,
         "duration_seconds": duration,
         "prompts": prompt_count,
@@ -381,6 +444,12 @@ def handle_stop(data: dict) -> None:
         "total_lines_changed": total_lines,
     }
 
+    # Compute token usage from transcript
+    if transcript_path:
+        token_usage = _compute_token_usage(transcript_path)
+        if token_usage:
+            summary["tokens"] = token_usage
+
     (sdir / "summary.json").write_text(json.dumps(summary, indent=2))
 
     # Clear active session

{inscript-0.4.0 → inscript-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "inscript"
-version = "0.4.0"
+version = "0.5.0"
 description = "Universal agent activity ledger. Records what AI agents do."
 readme = "README.md"
 license = "MIT"