token-tracker 0.3.7__tar.gz → 0.3.8__tar.gz

{token_tracker-0.3.7/token_tracker.egg-info → token_tracker-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.3.7
+Version: 0.3.8
 Summary: Track token usage across local AI agents (Claude Code, Codex)
 Author-email: stormzhang <stormzhang.dev@gmail.com>
 License: MIT License

{token_tracker-0.3.7 → token_tracker-0.3.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "token-tracker"
-version = "0.3.7"
+version = "0.3.8"
 description = "Track token usage across local AI agents (Claude Code, Codex)"
 readme = "README.md"
 requires-python = ">=3.11"

{token_tracker-0.3.7 → token_tracker-0.3.8}/src/analyzer/cost.py

@@ -25,8 +25,12 @@ _FAMILY_FALLBACK = (
     ("claude-opus", "claude-opus-4-8"),
     ("claude-sonnet", "claude-sonnet-4-6"),
     ("claude-haiku", "claude-haiku-4-5-20251001"),
+    ("claude-fable", "claude-fable-5"),
 )
 
+# 解析不到定价的模型只提示一次，避免聚合时每条 entry 刷屏
+_warned_unknown_models: set[str] = set()
+
 
 def get_pricing() -> dict:
     global _pricing
@@ -44,6 +48,7 @@ def calculate_cost(entry: UsageEntry) -> float:
     pricing = get_pricing()
     model_key = _resolve_model_key(entry.model, pricing)
     if model_key is None:
+        _warn_unknown_model_once(entry.model)
         return 0.0
 
     info = pricing[model_key]
@@ -85,6 +90,17 @@ def _resolve_model_key(model: str, pricing: dict) -> str | None:
     return None
 
 
+def _warn_unknown_model_once(model: str) -> None:
+    # 全新系列（非 opus/sonnet/haiku/fable）连系列兜底都接不住，成本会按 $0 计；显形以免静默少算
+    if model and model not in _warned_unknown_models:
+        _warned_unknown_models.add(model)
+        print(
+            f"token-tracker: 未知模型 {model!r} 缺少定价，本次成本按 $0 计；"
+            "litellm 收录后自动恢复，或在 cost.py 的 _fallback_pricing 补内置价",
+            file=sys.stderr,
+        )
+
+
 def _load_pricing() -> dict:
     cached = _read_cache()
     if cached is not None and not _cache_stale():
@@ -169,9 +185,18 @@ _OPUS_PRICING = {
     "cache_read_input_token_cost": 0.5e-6,
 }
 
+# Fable 5 / Mythos 5 同价，是 Opus 档的 2 倍（$10/$50 每百万 token）
+_FABLE_PRICING = {
+    "input_cost_per_token": 10e-6,
+    "output_cost_per_token": 50e-6,
+    "cache_creation_input_token_cost": 12.5e-6,
+    "cache_read_input_token_cost": 1.0e-6,
+}
+
 
 def _fallback_pricing() -> dict:
     return {
+        "claude-fable-5": _FABLE_PRICING,
         "claude-opus-4-8": _OPUS_PRICING,
         "claude-opus-4-7": _OPUS_PRICING,
         "claude-opus-4-6": _OPUS_PRICING,

{token_tracker-0.3.7 → token_tracker-0.3.8}/src/ui/format.py

@@ -8,8 +8,10 @@ AGENT_SHORT = {"claude-code": "CC", "codex": "Codex"}
 AGENT_LABEL = {"claude-code": "Claude Code", "codex": "Codex"}
 
 MODEL_SHORT = {
+    "claude-fable-5": "Fable 5",
     "claude-opus-4-6": "Opus 4.6",
     "claude-opus-4-7": "Opus 4.7",
+    "claude-opus-4-8": "Opus 4.8",
     "claude-sonnet-4-6": "Sonnet 4.6",
     "claude-sonnet": "Sonnet",
     "claude-haiku-4-5-20251001": "Haiku 4.5",

{token_tracker-0.3.7 → token_tracker-0.3.8}/tests/test_cost.py

@@ -114,6 +114,48 @@ def test_fallback_pricing_includes_openai_models():
         assert pricing[k].get("input_cost_per_token", 0) > 0
 
 
+def test_fallback_pricing_includes_fable():
+    # Fable 5 是全新系列，必须有专属兜底价（不能退回 Opus，价格差一倍）
+    info = cost._fallback_pricing()["claude-fable-5"]
+    assert info["input_cost_per_token"] == pytest.approx(10e-6)
+    assert info["output_cost_per_token"] == pytest.approx(50e-6)
+
+
+def test_fable_cost_is_double_opus(monkeypatch):
+    monkeypatch.setattr(cost, "_pricing", cost._fallback_pricing())
+    entry = make_entry(
+        model="claude-fable-5",
+        input_tokens=1_000_000,
+        output_tokens=1_000_000,
+        cache_creation_tokens=1_000_000,
+        cache_read_tokens=1_000_000,
+    )
+    # 10 + 50 + 12.5 + 1.0
+    assert cost.calculate_cost(entry) == pytest.approx(73.5)
+
+
+def test_fable_dated_variant_resolves_via_prefix(monkeypatch):
+    monkeypatch.setattr(cost, "_pricing", cost._fallback_pricing())
+    entry = make_entry(model="claude-fable-5-20260601", input_tokens=1_000_000)
+    assert cost.calculate_cost(entry) == pytest.approx(10.0)
+
+
+def test_unknown_fable_variant_falls_back_to_family(monkeypatch):
+    # 未来的 fable-6 即便 litellm 未收录，也按系列退回 fable-5，不归零
+    monkeypatch.setattr(cost, "_pricing", cost._fallback_pricing())
+    entry = make_entry(model="claude-fable-6-20270101", input_tokens=1_000_000)
+    assert cost.calculate_cost(entry) == pytest.approx(10.0)
+
+
+def test_unknown_model_warns_once(fixed_pricing, monkeypatch, capsys):
+    # 全新系列接不住时按 $0 计，但必须显形提醒；同一模型只提示一次
+    monkeypatch.setattr(cost, "_warned_unknown_models", set())
+    assert cost.calculate_cost(make_entry(model="claude-quartz-1", input_tokens=1_000_000)) == 0.0
+    assert cost.calculate_cost(make_entry(model="claude-quartz-1", input_tokens=500_000)) == 0.0
+    err = capsys.readouterr().err
+    assert err.count("claude-quartz-1") == 1
+
+
 def test_fresh_cache_is_used_without_fetching(tmp_path, monkeypatch):
     cache = tmp_path / "pricing_cache.json"
     cache.write_text('{"gpt-5": {"input_cost_per_token": 1e-6}}', encoding="utf-8")

{token_tracker-0.3.7 → token_tracker-0.3.8/token_tracker.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.3.7
+Version: 0.3.8
 Summary: Track token usage across local AI agents (Claude Code, Codex)
 Author-email: stormzhang <stormzhang.dev@gmail.com>
 License: MIT License