From b3fc4223276df3171fb93e459e68f3cb342bbf81 Mon Sep 17 00:00:00 2001
From: Alex-wuhu <yanglongwei06@gmail.com>
Date: Wed, 25 Mar 2026 00:40:17 +0800
Subject: [PATCH] Add Novita AI as LLM provider for cost tracking

- Add NOVITA_PROVIDER constant
- Add Novita models to MODEL_TOKEN_PRICING with pricing:
  - moonshotai/kimi-k2.5: /bin/bash.6/M input, /M output, /bin/bash.1/M cache
  - zai-org/glm-5: /M input, .2/M output, /bin/bash.2/M cache
  - minimax/minimax-m2.5: /bin/bash.3/M input, .2/M output, /bin/bash.03/M cache
- Update _estimate_provider_cost to handle Novita (OpenAI-compatible)
- Update provider validation in extract_api_cost
---
 src/art/api_costs.py | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/art/api_costs.py b/src/art/api_costs.py
index 0ed5e067..8d4ba435 100644
--- a/src/art/api_costs.py
+++ b/src/art/api_costs.py
@@ -10,6 +10,7 @@
 
 OPENAI_PROVIDER = "openai"
 ANTHROPIC_PROVIDER = "anthropic"
+NOVITA_PROVIDER = "novita"
 
 P = ParamSpec("P")
 R = TypeVar("R")
@@ -54,6 +55,21 @@ class _AnthropicTokenUsage:
         cache_creation_per_million=3.75,
         cache_read_per_million=0.30,
     ),
+    "novita/moonshotai/kimi-k2.5": TokenPricing(
+        prompt_per_million=0.6,
+        completion_per_million=3.0,
+        cached_prompt_per_million=0.1,
+    ),
+    "novita/zai-org/glm-5": TokenPricing(
+        prompt_per_million=1.0,
+        completion_per_million=3.2,
+        cached_prompt_per_million=0.2,
+    ),
+    "novita/minimax/minimax-m2.5": TokenPricing(
+        prompt_per_million=0.3,
+        completion_per_million=1.2,
+        cached_prompt_per_million=0.03,
+    ),
 }
 
 
@@ -264,7 +280,7 @@ def _estimate_provider_cost(
     response: Any,
     pricing: TokenPricing,
 ) -> float | None:
-    if provider_name == OPENAI_PROVIDER:
+    if provider_name in {OPENAI_PROVIDER, NOVITA_PROVIDER}:
         return _estimate_openai_cost(_extract_openai_token_counts(response), pricing)
     if provider_name == ANTHROPIC_PROVIDER:
         return _estimate_anthropic_cost(
@@ -438,7 +454,7 @@ def extract_api_cost(
     if provider_cost is not None:
         return provider_cost
 
-    if provider_name in {OPENAI_PROVIDER, ANTHROPIC_PROVIDER}:
+    if provider_name in {OPENAI_PROVIDER, ANTHROPIC_PROVIDER, NOVITA_PROVIDER}:
         raise ValueError(
             f"Response usage does not match provider '{provider_name}'. "
             "Pass the correct provider/model pair or register a custom cost extractor."