pisama-claude-code 0.3.4__tar.gz → 0.3.6__tar.gz

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/CHANGELOG.md

@@ -7,6 +7,13 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.3.5] - 2025-01-05
+
+### Fixed
+- Correctly extract user input from transcript (skip tool_result blocks)
+- Collect reasoning/output from multiple assistant messages (Claude Code splits them)
+- Look back 200 lines to find user prompts
+
 ## [0.3.4] - 2025-01-05
 
 ### Fixed
@@ -84,7 +91,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Support for Bash, Read, Write, Edit, Grep, Glob tools
 - Local storage in `~/.claude/pisama/traces/`
 
-[Unreleased]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.4...HEAD
+[Unreleased]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.5...HEAD
+[0.3.5]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.4...v0.3.5
 [0.3.4]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.3...v0.3.4
 [0.3.3]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.2...v0.3.3
 [0.3.2]: https://github.com/tn-pisama/pisama-claude-code/compare/v0.3.1...v0.3.2

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pisama-claude-code
-Version: 0.3.4
+Version: 0.3.6
 Summary: Trace capture for Claude Code sessions - sync to PISAMA platform for analysis
 Project-URL: Homepage, https://pisama.dev
 Project-URL: Documentation, https://pisama.dev/docs/claude-code
@@ -57,6 +57,10 @@ Description-Content-Type: text/markdown
 [![Downloads/month](https://img.shields.io/pypi/dm/pisama-claude-code)](https://pypistats.org/packages/pisama-claude-code)
 [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
 
+## Demo
+
+![pisama-claude-code demo](assets/demo.gif)
+
 ## Why PISAMA?
 
 When working with Claude Code, have you ever wondered:

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/README.md

@@ -10,6 +10,10 @@
 [![Downloads/month](https://img.shields.io/pypi/dm/pisama-claude-code)](https://pypistats.org/packages/pisama-claude-code)
 [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
 
+## Demo
+
+![pisama-claude-code demo](assets/demo.gif)
+
 ## Why PISAMA?
 
 When working with Claude Code, have you ever wondered:

pisama_claude_code-0.3.6/demo/demo.cast

@@ -0,0 +1,153 @@
+{"version":3,"term":{"cols":80,"rows":24,"type":"xterm-256color"},"timestamp":1767597795,"command":"./demo/demo-simulated.sh","env":{"SHELL":"/bin/zsh"}}
+[0.011, "o", "\u001b[3J\u001b[H\u001b[2J"]
+[0.000, "o", "\u001b[1;36m"]
+[0.003, "o", "╔═══════════════════════════════════════════════════╗\r\n║   pisama-claude-code                              ║\r\n║   Track token costs for Claude Code sessions      ║\r\n╚═══════════════════════════════════════════════════╝\r\n"]
+[0.000, "o", "\u001b[0m\r\n"]
+[2.004, "o", "\u001b[1;33m# Install\u001b[0m\r\n"]
+[0.000, "o", "\u001b[0;32m$\u001b[0m "]
+[0.000, "o", "p"]
+[0.048, "o", "i"]
+[0.046, "o", "p"]
+[0.046, "o", " "]
+[0.043, "o", "i"]
+[0.043, "o", "n"]
+[0.048, "o", "s"]
+[0.043, "o", "t"]
+[0.045, "o", "a"]
+[0.047, "o", "l"]
+[0.047, "o", "l"]
+[0.046, "o", " "]
+[0.048, "o", "p"]
+[0.046, "o", "i"]
+[0.047, "o", "s"]
+[0.047, "o", "a"]
+[0.043, "o", "m"]
+[0.044, "o", "a"]
+[0.046, "o", "-"]
+[0.049, "o", "c"]
+[0.045, "o", "l"]
+[0.047, "o", "a"]
+[0.048, "o", "u"]
+[0.043, "o", "d"]
+[0.048, "o", "e"]
+[0.047, "o", "-"]
+[0.045, "o", "c"]
+[0.047, "o", "o"]
+[0.048, "o", "d"]
+[0.044, "o", "e"]
+[0.044, "o", "\r\n"]
+[0.207, "o", "Successfully installed pisama-claude-code-0.3.2\r\n"]
+[1.508, "o", "\r\n\u001b[1;33m# Set up hooks\u001b[0m\r\n\u001b[0;32m$\u001b[0m "]
+[0.000, "o", "p"]
+[0.047, "o", "i"]
+[0.045, "o", "s"]
+[0.046, "o", "a"]
+[0.046, "o", "m"]
+[0.045, "o", "a"]
+[0.047, "o", "-"]
+[0.043, "o", "c"]
+[0.047, "o", "c"]
+[0.047, "o", " "]
+[0.048, "o", "i"]
+[0.043, "o", "n"]
+[0.046, "o", "s"]
+[0.044, "o", "t"]
+[0.046, "o", "a"]
+[0.047, "o", "l"]
+[0.045, "o", "l"]
+[0.047, "o", "\r\n"]
+[0.209, "o", "✅ Installed pisama-capture.py\r\n✅ Installed pisama-pre.sh\r\n✅ Installed pisama-post.sh\r\n✅ Updated ~/.claude/settings.local.json\r\n\r\nHooks installed! Traces will be captured automatically.\r\n"]
+[2.005, "o", "\r\n\u001b[1;33m# After a coding session, check status\u001b[0m\r\n\u001b[0;32m$\u001b[0m "]
+[0.000, "o", "p"]
+[0.045, "o", "i"]
+[0.047, "o", "s"]
+[0.046, "o", "a"]
+[0.048, "o", "m"]
+[0.045, "o", "a"]
+[0.045, "o", "-"]
+[0.045, "o", "c"]
+[0.048, "o", "c"]
+[0.048, "o", " "]
+[0.048, "o", "s"]
+[0.043, "o", "t"]
+[0.044, "o", "a"]
+[0.049, "o", "t"]
+[0.047, "o", "u"]
+[0.045, "o", "s"]
+[0.043, "o", "\r\n"]
+[0.211, "o", "📊 PISAMA Status\r\n========================================\r\n\r\n🔧 Hook Installation:\r\n   ✅ pisama-capture.py\r\n   ✅ pisama-pre.sh\r\n   ✅ pisama-post.sh\r\n   All hooks installed\r\n\r\n🔗 Platform Connection:\r\n   ❌ Not connected (local-only mode)\r\n\r\n📁 Local Traces: 1,847\r\n   Input tokens:  142,580\r\n   Output tokens: 891,234\r\n   Total cost:    $52.34\r\n   Models: claude-opus-4-5-20251101\r\n\r\n📝 Claude Code Settings:\r\n   ✅ PISAMA hooks configured in settings\r\n"]
+[2.509, "o", "\r\n\u001b[1;33m# View recent tool calls\u001b[0m\r\n\u001b[0;32m$\u001b[0m "]
+[0.000, "o", "p"]
+[0.046, "o", "i"]
+[0.043, "o", "s"]
+[0.043, "o", "a"]
+[0.043, "o", "m"]
+[0.043, "o", "a"]
+[0.044, "o", "-"]
+[0.047, "o", "c"]
+[0.046, "o", "c"]
+[0.046, "o", " "]
+[0.042, "o", "t"]
+[0.043, "o", "r"]
+[0.046, "o", "a"]
+[0.047, "o", "c"]
+[0.048, "o", "e"]
+[0.045, "o", "s"]
+[0.045, "o", " "]
+[0.047, "o", "-"]
+[0.047, "o", "v"]
+[0.047, "o", " "]
+[0.043, "o", "-"]
+[0.045, "o", "-"]
+[0.048, "o", "l"]
+[0.047, "o", "a"]
+[0.046, "o", "s"]
+[0.046, "o", "t"]
+[0.048, "o", " "]
+[0.043, "o", "5"]
+[0.045, "o", "\r\n"]
+[0.207, "o", "📋 Recent Traces (5 shown)\r\n======================================================================\r\n2025-01-04T14:23:45 | Bash         | a3f8c2d1 | post |   2341i  8923o | $0.7012\r\n2025-01-04T14:23:38 | Read         | a3f8c2d1 | post |   1823i  2341o | $0.1892\r\n2025-01-04T14:23:31 | Edit         | a3f8c2d1 | post |   3421i 12893o | $1.0234\r\n2025-01-04T14:23:22 | Grep         | a3f8c2d1 | post |    892i  1234o | $0.1023\r\n2025-01-04T14:23:15 | Glob         | a3f8c2d1 | post |    423i   891o | $0.0712\r\n──────────────────────────────────────────────────────────────────────\r\nTotals: 8,900 input + 26,282 output tokens = $2.0873\r\n"]
+[2.504, "o", "\r\n\u001b[1;33m# Detailed breakdown\u001b[0m\r\n"]
+[0.000, "o", "\u001b[0;32m$\u001b[0m "]
+[0.000, "o", "p"]
+[0.045, "o", "i"]
+[0.043, "o", "s"]
+[0.045, "o", "a"]
+[0.044, "o", "m"]
+[0.045, "o", "a"]
+[0.048, "o", "-"]
+[0.044, "o", "c"]
+[0.047, "o", "c"]
+[0.043, "o", " "]
+[0.043, "o", "u"]
+[0.043, "o", "s"]
+[0.047, "o", "a"]
+[0.048, "o", "g"]
+[0.046, "o", "e"]
+[0.042, "o", " "]
+[0.044, "o", "-"]
+[0.043, "o", "-"]
+[0.046, "o", "b"]
+[0.046, "o", "y"]
+[0.045, "o", "-"]
+[0.046, "o", "m"]
+[0.046, "o", "o"]
+[0.046, "o", "d"]
+[0.047, "o", "e"]
+[0.044, "o", "l"]
+[0.048, "o", " "]
+[0.044, "o", "-"]
+[0.048, "o", "-"]
+[0.045, "o", "b"]
+[0.045, "o", "y"]
+[0.044, "o", "-"]
+[0.048, "o", "t"]
+[0.042, "o", "o"]
+[0.047, "o", "o"]
+[0.046, "o", "l"]
+[0.046, "o", "\r\n"]
+[0.210, "o", "📊 Token Usage Summary (last 100 traces)\r\n==================================================\r\nInput tokens:           142,580\r\nOutput tokens:          891,234\r\nCache read tokens:    1,234,567\r\nTotal tokens:         1,033,814\r\nTotal cost:        $       52.34\r\n\r\n📈 By Model:\r\n--------------------------------------------------\r\n  claude-opus-4-5-20251101              $52.34\r\n\r\n🔧 By Tool:\r\n--------------------------------------------------\r\n  Bash                   145 calls  $23.45\r\n  Read                    89 calls  $12.34\r\n  Edit                    67 calls   $8.92\r\n  Write                   34 calls   $4.21\r\n  Grep                    23 calls   $2.12\r\n  Glob                    12 calls   $1.30\r\n"]
+[2.002, "o", "\r\n\u001b[1;36m"]
+[0.003, "o", "════════════════════════════════════════════════════\r\n  ✓ Track every token. Know your costs.\r\n\r\n  pip install pisama-claude-code\r\n════════════════════════════════════════════════════\r\n"]
+[0.000, "o", "\u001b[0m"]
+[3.008, "x", "0"]

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pisama-claude-code"
-version = "0.3.4"
+version = "0.3.6"
 description = "Trace capture for Claude Code sessions - sync to PISAMA platform for analysis"
 readme = "README.md"
 license = "MIT"

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/src/pisama_claude_code/__init__.py

@@ -1,6 +1,6 @@
 """PISAMA Claude Code Integration - Trace capture, failure detection, and self-healing."""
 
-__version__ = "0.3.4"
+__version__ = "0.3.5"
 
 # Lazy imports to avoid loading everything at startup
 def install(force: bool = False):

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/src/pisama_claude_code/cli.py

@@ -53,12 +53,15 @@ def save_config(config: dict):
 
 
 @click.group()
-@click.version_option(version="0.3.4")
+@click.version_option(version="0.3.6")
 def main():
     """PISAMA Claude Code - Trace capture and sync."""
     pass
 
 
+GITHUB_URL = "https://github.com/tn-pisama/pisama-claude-code"
+
+
 @main.command()
 @click.option("--force", "-f", is_flag=True, help="Overwrite existing hooks")
 def install(force: bool):
@@ -66,6 +69,13 @@ def install(force: bool):
     from pisama_claude_code.install import install as do_install
     do_install(force=force)
 
+    # Star reminder after install
+    click.echo("")
+    click.echo("=" * 50)
+    click.echo("If pisama-claude-code is useful, please star us!")
+    click.echo(f"   {GITHUB_URL}")
+    click.echo("=" * 50)
+
 
 @main.command()
 def uninstall():
@@ -439,6 +449,10 @@ def status():
     else:
         click.echo("   ❌ No settings file found")
 
+    # Star reminder
+    click.echo("")
+    click.echo(f"Star us on GitHub: {GITHUB_URL}")
+
 
 @main.command()
 @click.option("--last", default=100, help="Number of traces to analyze")
@@ -675,7 +689,7 @@ def prepare_sync_payload(traces: list, include_outputs: bool) -> dict:
 
     return {
         "source": "claude-code",
-        "version": "0.3.4",
+        "version": "0.3.5",
         "uploaded_at": datetime.now(timezone.utc).isoformat(),
         "trace_count": len(clean_traces),
         "traces": clean_traces,

{pisama_claude_code-0.3.4 → pisama_claude_code-0.3.6}/src/pisama_claude_code/hooks/capture_hook.py

@@ -185,7 +185,14 @@ def extract_content_parts(content_blocks: list) -> dict:
 
 
 def get_last_user_message(transcript_path: str) -> dict:
-    """Read transcript and get the last user message (input)."""
+    """Read transcript and get the last user text message (input).
+
+    Note: Claude Code "user" messages can be either:
+    1. Actual user text input (content is a string or has type="text" blocks)
+    2. Tool results (content has type="tool_result" blocks)
+
+    We want the actual user input, not tool results.
+    """
     try:
         from pathlib import Path
         transcript = Path(transcript_path)
@@ -193,29 +200,47 @@ def get_last_user_message(transcript_path: str) -> dict:
             return {}
 
         with open(transcript) as f:
-            lines = f.readlines()[-50:]  # Look back further for user message
+            lines = f.readlines()[-200:]  # Look back further for user message
 
         for line in reversed(lines):
             try:
                 entry = json.loads(line)
-                if entry.get("type") == "human" and "message" in entry:
+                # Claude Code uses "user" type
+                if entry.get("type") in ("user", "human") and "message" in entry:
                     msg = entry["message"]
                     content = msg.get("content", [])
 
-                    # Extract text from content blocks
-                    if isinstance(content, list):
+                    # Content can be a direct string (actual user input)
+                    if isinstance(content, str):
+                        # Skip system messages and interrupts
+                        if not content.startswith("[Request") and not content.startswith("[System"):
+                            return {"content": content, "role": "user"}
+
+                    # Content can be a list of blocks
+                    elif isinstance(content, list):
                         text_parts = []
+                        is_tool_result = False
+
                         for block in content:
-                            if isinstance(block, dict) and block.get("type") == "text":
-                                text_parts.append(block.get("text", ""))
+                            if isinstance(block, dict):
+                                block_type = block.get("type", "")
+                                if block_type == "tool_result":
+                                    is_tool_result = True
+                                    break  # This is a tool result, not user input
+                                elif block_type == "text":
+                                    text = block.get("text", "")
+                                    if text and not text.startswith("[Request"):
+                                        text_parts.append(text)
                             elif isinstance(block, str):
                                 text_parts.append(block)
-                        return {
-                            "content": "\n".join(text_parts) if text_parts else None,
-                            "role": "user",
-                        }
-                    elif isinstance(content, str):
-                        return {"content": content, "role": "user"}
+
+                        # Only return if this is actual user text, not tool results
+                        if text_parts and not is_tool_result:
+                            return {
+                                "content": "\n".join(text_parts),
+                                "role": "user",
+                            }
+
             except json.JSONDecodeError:
                 continue
         return {}
@@ -224,14 +249,17 @@ def get_last_user_message(transcript_path: str) -> dict:
 
 
 def get_last_assistant_message(transcript_path: str) -> dict:
-    """Read transcript and get the last assistant message with usage.
+    """Read transcript and get recent assistant messages with usage.
+
+    Claude Code separates thinking, text, and tool_use into different messages,
+    so we need to collect content from multiple recent assistant messages.
 
-    Now extracts:
+    Extracts:
     - model: The Claude model used
     - usage: Token counts (input, output, cache)
     - input: User's input message
-    - reasoning: Extended thinking content
-    - output: Text response content
+    - reasoning: Extended thinking content (collected from recent messages)
+    - output: Text response content (collected from recent messages)
     - tool_calls: Any tool calls made
     """
     try:
@@ -240,45 +268,72 @@ def get_last_assistant_message(transcript_path: str) -> dict:
         if not transcript.exists():
             return {}
 
-        # Read last 50 lines to find messages
+        # Read last 100 lines to find messages
         with open(transcript) as f:
-            lines = f.readlines()[-50:]
+            lines = f.readlines()[-100:]
+
+        # Collect content from recent assistant messages (up to 10)
+        # Claude Code sends thinking, text, tool_use as separate messages
+        all_reasoning = []
+        all_output = []
+        all_tool_calls = []
+        model = None
+        usage = {}
+        stop_reason = None
+        messages_checked = 0
+        max_messages = 10
 
-        assistant_msg = None
         for line in reversed(lines):
+            if messages_checked >= max_messages:
+                break
             try:
                 entry = json.loads(line)
                 if entry.get("type") == "assistant" and "message" in entry:
+                    messages_checked += 1
                     msg = entry["message"]
-                    assistant_msg = msg
+
+                    # Get model and usage from first (most recent) message
+                    if model is None:
+                        model = msg.get("model")
+                        usage = msg.get("usage", {})
+                        stop_reason = msg.get("stop_reason")
+
+                    # Extract content from this message
+                    content = msg.get("content", [])
+                    if isinstance(content, list):
+                        parts = extract_content_parts(content)
+                        if parts.get("reasoning", {}).get("content"):
+                            all_reasoning.append(parts["reasoning"]["content"])
+                        if parts.get("output", {}).get("content"):
+                            all_output.append(parts["output"]["content"])
+                        if parts.get("tool_calls"):
+                            all_tool_calls.extend(parts["tool_calls"])
+
+                # Stop if we hit a user message (different turn)
+                elif entry.get("type") == "human":
                     break
+
             except json.JSONDecodeError:
                 continue
 
-        if not assistant_msg:
-            return {}
-
-        # Extract content parts (reasoning, output, tool_calls)
-        content = assistant_msg.get("content", [])
-        content_parts = extract_content_parts(content) if isinstance(content, list) else {}
-
         # Get user input
         user_msg = get_last_user_message(transcript_path)
 
-        # Get usage breakdown
-        usage = assistant_msg.get("usage", {})
+        # Combine collected content (reverse to get chronological order)
+        reasoning_text = "\n\n".join(reversed(all_reasoning)) if all_reasoning else None
+        output_text = "\n\n".join(reversed(all_output)) if all_output else None
 
         return {
-            "model": assistant_msg.get("model"),
+            "model": model,
             "usage": usage,
-            "stop_reason": assistant_msg.get("stop_reason"),
+            "stop_reason": stop_reason,
             # Structured content
             "input": user_msg.get("content"),
-            "reasoning": content_parts.get("reasoning", {}).get("content"),
-            "output": content_parts.get("output", {}).get("content"),
-            "tool_calls": content_parts.get("tool_calls"),
+            "reasoning": reasoning_text,
+            "output": output_text,
+            "tool_calls": all_tool_calls if all_tool_calls else None,
             # Legacy field for backward compatibility
-            "content": content,
+            "content": [],
         }
     except Exception:
         return {}