unique_toolkit 2026.46.0.dev3__py3-none-any.whl → 2026.46.0.dev4__py3-none-any.whl

unique_toolkit/language_model/model_costs.py

@@ -145,7 +145,12 @@ def calculate_invocation_cost_usd(
     token_usage: LanguageModelTokenUsage,
     catalog: ModelCostCatalog | None = None,
 ) -> float | None:
-    """Calculate invocation cost using prompt and completion token totals."""
+    """Calculate invocation cost, pricing cache reads and writes at their own rates.
+
+    Mirrors node-chat's ``ModelCostService.computeSpent`` so debug-info spend and
+    the platform ledger agree. Cache reads and writes are counted inside
+    ``prompt_tokens``; a row without a cache rate bills them at ``input``.
+    """
 
     if token_usage.prompt_tokens is None or token_usage.completion_tokens is None:
         return None
@@ -163,10 +168,32 @@ def calculate_invocation_cost_usd(
     if model_cost is None:
         return None
 
-    input_multiplier, completion_multiplier = model_cost.multipliers_for(
-        token_usage.prompt_tokens
+    prompt_tokens = token_usage.prompt_tokens
+    cache_read_tokens = min(token_usage.cached_tokens or 0, prompt_tokens)
+    cache_write_tokens = min(
+        token_usage.cache_write_tokens or 0, prompt_tokens - cache_read_tokens
+    )
+    uncached_tokens = prompt_tokens - cache_read_tokens - cache_write_tokens
+    cache_read_rate = (
+        model_cost.cached_input
+        if model_cost.cached_input is not None
+        else model_cost.input
     )
+    # The usage payload does not say which TTL a write used; node-chat bills an
+    # unsplit write at the 5-minute rate too.
+    cache_write_rate = (
+        model_cost.cache_write
+        if model_cost.cache_write is not None
+        else model_cost.input
+    )
+
+    input_multiplier, completion_multiplier = model_cost.multipliers_for(prompt_tokens)
     return (
-        token_usage.prompt_tokens * model_cost.input * input_multiplier
+        (
+            uncached_tokens * model_cost.input
+            + cache_read_tokens * cache_read_rate
+            + cache_write_tokens * cache_write_rate
+        )
+        * input_multiplier
         + token_usage.completion_tokens * model_cost.completion * completion_multiplier
     ) / _TOKENS_PER_MILLION

{unique_toolkit-2026.46.0.dev3.dist-info → unique_toolkit-2026.46.0.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.46.0.dev3
+Version: 2026.46.0.dev4
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-2026.46.0.dev3.dist-info → unique_toolkit-2026.46.0.dev4.dist-info}/RECORD

@@ -402,7 +402,7 @@ unique_toolkit/language_model/errors.py,sha256=0nm4psGCGJM2B7LIyKXy7WuswRzwARIgI
 unique_toolkit/language_model/functions.py,sha256=6Le6dDm9su6xk3sQ8IPiyUn2tf1qvzxHVUhyTTgP8Kw,28109
 unique_toolkit/language_model/infos.py,sha256=IAwSmmFEdX51bYXsx62XQqJ4wwaJMz7fGf3nsNcHlZE,183833
 unique_toolkit/language_model/invocation_stats.py,sha256=CQbpUz394IsVd8eNTB1gPSMt5f0UXmkN1amNsqtAW6U,5857
-unique_toolkit/language_model/model_costs.py,sha256=ShuDAzMew1fUkJ3tzPgU5Zqjp6ivUZaG97BFh93tEn4,6358
+unique_toolkit/language_model/model_costs.py,sha256=AVHAuMTzsglzusWsVF_xwBVEndST-TVCVi0H49OhdvE,7419
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/provider_token_count.py,sha256=mS597Mvrc7EsAc9iJSmASIEwvETSPLtIEsWxusPrQVc,5337
 unique_toolkit/language_model/reference.py,sha256=T3wr2jP_uRg5wKVIbKqLbxJlcgxfbZQGq3XdjbQ2kSo,6984
@@ -430,6 +430,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.46.0.dev3.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.46.0.dev3.dist-info/METADATA,sha256=hwd2lpcpuRDDzxudDerD9o_Jqd9wIvAbUadPJu-Ytn4,7821
-unique_toolkit-2026.46.0.dev3.dist-info/RECORD,,
+unique_toolkit-2026.46.0.dev4.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.46.0.dev4.dist-info/METADATA,sha256=S3pb1MeBajfmMuiG1-DzzHjYwlu1YKRqAbgpw0XhhWA,7821
+unique_toolkit-2026.46.0.dev4.dist-info/RECORD,,