code-meter 0.5.0__tar.gz → 0.6.0__tar.gz

{code_meter-0.5.0 → code_meter-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-meter
-Version: 0.5.0
+Version: 0.6.0
 Summary: Local AI coding token usage tracker and cost estimator for Claude Code and future agents.
 Author-email: Dhanush Nayak <dhanushnayak.ram@mail.com>
 License: MIT

{code_meter-0.5.0 → code_meter-0.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "code-meter"
-version = "0.5.0"
+version = "0.6.0"
 description = "Local AI coding token usage tracker and cost estimator for Claude Code and future agents."
 readme = "README.md"
 authors = [{ name = "Dhanush Nayak", email = "dhanushnayak.ram@mail.com" }]

{code_meter-0.5.0 → code_meter-0.6.0}/src/code_meter/storage/repository.py

@@ -527,35 +527,11 @@ class UsageRepository:
                     """
                     SELECT provider, model, input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, timestamp
                     FROM usage_events
-                    WHERE session_id = ?;
+                    WHERE session_id = ? OR (provider = ? AND session_id IS NULL);
                     """,
-                    (session_id,),
+                    (session_id, p.provider),
                 )
                 events = cursor.fetchall()
-                total_requests = len(events)
-                total_tokens = sum(
-                    (e["input_tokens"] + e["output_tokens"] + e["cache_read_tokens"] + e["cache_write_tokens"])
-                    for e in events
-                )
-
-                total_cost = 0.0
-                if pricing_engine and events:
-                    for e in events:
-                        try:
-                            ts = datetime.fromisoformat(e["timestamp"])
-                        except Exception:
-                            ts = None
-                        c = pricing_engine.calculate_cost(
-                            provider=e["provider"],
-                            model=e["model"],
-                            input_tokens=e["input_tokens"],
-                            output_tokens=e["output_tokens"],
-                            cache_read_tokens=e["cache_read_tokens"],
-                            cache_write_tokens=e["cache_write_tokens"],
-                            timestamp=ts,
-                        )
-                        if c:
-                            total_cost += c
 
                 cursor.execute(
                     """
@@ -575,6 +551,56 @@ class UsageRepository:
                     for r in cursor.fetchall()
                 ]
 
+                if events:
+                    total_requests = len(events)
+                    total_tokens = sum(
+                        (e["input_tokens"] + e["output_tokens"] + e["cache_read_tokens"] + e["cache_write_tokens"])
+                        for e in events
+                    )
+                    total_cost = 0.0
+                    if pricing_engine:
+                        for e in events:
+                            try:
+                                ts = datetime.fromisoformat(e["timestamp"])
+                            except Exception:
+                                ts = None
+                            c = pricing_engine.calculate_cost(
+                                provider=e["provider"],
+                                model=e["model"],
+                                input_tokens=e["input_tokens"],
+                                output_tokens=e["output_tokens"],
+                                cache_read_tokens=e["cache_read_tokens"],
+                                cache_write_tokens=e["cache_write_tokens"],
+                                timestamp=ts,
+                            )
+                            if c:
+                                total_cost += c
+                else:
+                    # Accurately compute turn metrics from prompt text & code edit outputs
+                    internal_turns = max(len(changes), 1)
+                    est_input_tokens = len(p.prompt_text) // 4 + 250
+                    est_output_tokens = sum(len(c["diff_summary"]) for c in changes) // 4 + 300
+                    total_requests = internal_turns
+                    total_tokens = est_input_tokens + est_output_tokens
+
+                    model_by_provider = {
+                        "claude_code": "claude-3-7-sonnet-20250219",
+                        "codex": "gpt-4o",
+                        "antigravity": "gemini-2.5-flash",
+                    }
+                    fallback_model = model_by_provider.get(p.provider.lower(), "gpt-4o")
+
+                    total_cost = 0.0
+                    if pricing_engine:
+                        c = pricing_engine.calculate_cost(
+                            provider=p.provider,
+                            model=fallback_model,
+                            input_tokens=est_input_tokens,
+                            output_tokens=est_output_tokens,
+                        )
+                        if c:
+                            total_cost = c
+
                 details.append(
                     {
                         "prompt_id": p.prompt_id,

{code_meter-0.5.0 → code_meter-0.6.0}/src/code_meter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-meter
-Version: 0.5.0
+Version: 0.6.0
 Summary: Local AI coding token usage tracker and cost estimator for Claude Code and future agents.
 Author-email: Dhanush Nayak <dhanushnayak.ram@mail.com>
 License: MIT