First commit

tests/unit/test_token_cost_enterprise.py

@@ -0,0 +1,23 @@
+from types import SimpleNamespace
+from agent_framework.observability.token_cost import TokenUsageCollector, TokenUsage
+
+
+def test_token_usage_extracts_cached_and_reasoning_tokens():
+    usage = TokenUsage.from_openai_usage({
+        "prompt_tokens": 1000,
+        "completion_tokens": 500,
+        "total_tokens": 1600,
+        "prompt_tokens_details": {"cached_tokens": 250},
+        "completion_tokens_details": {"reasoning_tokens": 100},
+    })
+    assert usage.prompt_tokens == 1000
+    assert usage.cached_tokens == 250
+    assert usage.reasoning_tokens == 100
+    assert usage.total_tokens == 1600
+
+
+def test_cost_tracker_uses_model_prices_json():
+    settings = SimpleNamespace(MODEL_PRICES_JSON='{"my-model":{"input_per_1m":"1","output_per_1m":"2","cached_input_per_1m":"0.1"}}', USD_BRL_RATE='5')
+    enriched = TokenUsageCollector(settings).enrich("my-model", {"prompt_tokens": 1000, "completion_tokens": 1000, "prompt_tokens_details": {"cached_tokens": 500}})
+    assert enriched["cost_usd"] > 0
+    assert abs(enriched["cost_brl"] - enriched["cost_usd"] * 5) < 1e-9