fix: track token savings in get_file_tree and search_text

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: jgravelle

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "jcodemunch-mcp"
-version = "0.2.11"
+version = "0.2.12"
 description = "Token-efficient MCP server for source code exploration via tree-sitter AST parsing"
 readme = "README.md"
 requires-python = ">=3.10"

src/jcodemunch_mcp/tools/get_file_tree.py

@@ -4,7 +4,7 @@
 import time
 from typing import Optional
 
-from ..storage import IndexStore
+from ..storage import IndexStore, record_savings, estimate_savings, cost_avoided
 from ._utils import resolve_repo
 
 
@@ -53,13 +53,29 @@ def get_file_tree(
 
     elapsed = (time.perf_counter() - start) * 1000
 
+    # Token savings: sum of raw file sizes vs compact tree response
+    store2 = IndexStore(base_path=storage_path)
+    content_dir = store2._content_dir(owner, name)
+    raw_bytes = 0
+    for f in files:
+        try:
+            raw_bytes += os.path.getsize(content_dir / f)
+        except OSError:
+            pass
+    response_bytes = len(str(tree).encode())
+    tokens_saved = estimate_savings(raw_bytes, response_bytes)
+    total_saved = record_savings(tokens_saved)
+
     return {
         "repo": f"{owner}/{name}",
         "path_prefix": path_prefix,
         "tree": tree,
         "_meta": {
             "timing_ms": round(elapsed, 1),
             "file_count": len(files),
+            "tokens_saved": tokens_saved,
+            "total_tokens_saved": total_saved,
+            **cost_avoided(tokens_saved, total_saved),
         },
     }
 

src/jcodemunch_mcp/tools/search_text.py

@@ -1,9 +1,10 @@
 """Full-text search across indexed file contents."""
 
+import os
 import time
 from typing import Optional
 
-from ..storage import IndexStore
+from ..storage import IndexStore, record_savings, estimate_savings, cost_avoided
 from ._utils import resolve_repo
 
 
@@ -81,6 +82,17 @@ def search_text(
 
     elapsed = (time.perf_counter() - start) * 1000
 
+    # Token savings: raw bytes of searched files vs matched lines returned
+    raw_bytes = 0
+    for file_path in files[:files_searched]:
+        try:
+            raw_bytes += os.path.getsize(content_dir / file_path)
+        except OSError:
+            pass
+    response_bytes = sum(len(m["text"].encode()) for m in matches)
+    tokens_saved = estimate_savings(raw_bytes, response_bytes)
+    total_saved = record_savings(tokens_saved)
+
     return {
         "repo": f"{owner}/{name}",
         "query": query,
@@ -90,5 +102,8 @@ def search_text(
             "timing_ms": round(elapsed, 1),
             "files_searched": files_searched,
             "truncated": len(matches) >= max_results,
+            "tokens_saved": tokens_saved,
+            "total_tokens_saved": total_saved,
+            **cost_avoided(tokens_saved, total_saved),
         },
     }