diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 5029a57b782..9fbb8aa1fbc 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -38536,6 +38536,7 @@ "max_tokens": 262144, "mode": "chat", "output_cost_per_token": 4.4e-06, + "source": "https://docs.z.ai/guides/overview/pricing", "supports_function_calling": true, "supports_reasoning": true }, @@ -38548,6 +38549,7 @@ "max_tokens": 262144, "mode": "chat", "output_cost_per_token": 4e-06, + "source": "https://platform.kimi.ai/docs/pricing/chat-k27-code", "supports_function_calling": true, "supports_reasoning": true }, @@ -38560,6 +38562,7 @@ "max_tokens": 262144, "mode": "chat", "output_cost_per_token": 8e-06, + "source": "https://platform.kimi.ai/docs/pricing/chat-k27-code", "supports_function_calling": true, "supports_reasoning": true }, diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 03741c81443..de501e14e53 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -3073,6 +3073,7 @@ def test_model_info_for_openai_kimi_and_glm_aliases(local_model_cost_map): assert kimi["input_cost_per_token"] == 9.5e-07 assert kimi["output_cost_per_token"] == 4e-06 assert kimi["cache_read_input_token_cost"] == 1.9e-07 + assert kimi["source"] == "https://platform.kimi.ai/docs/pricing/chat-k27-code" assert kimi["max_input_tokens"] == 262144 kimi_highspeed = litellm.get_model_info( @@ -3083,6 +3084,7 @@ def test_model_info_for_openai_kimi_and_glm_aliases(local_model_cost_map): assert kimi_highspeed["input_cost_per_token"] == 1.9e-06 assert kimi_highspeed["output_cost_per_token"] == 8e-06 assert kimi_highspeed["cache_read_input_token_cost"] == 3.8e-07 + assert kimi_highspeed["source"] == "https://platform.kimi.ai/docs/pricing/chat-k27-code" assert kimi_highspeed["max_input_tokens"] == 262144 glm = litellm.get_model_info(model="openai/glm-5.2", custom_llm_provider="openai") @@ -3091,6 +3093,7 @@ def test_model_info_for_openai_kimi_and_glm_aliases(local_model_cost_map): assert glm["input_cost_per_token"] == 1.4e-06 assert glm["output_cost_per_token"] == 4.4e-06 assert glm["cache_read_input_token_cost"] == 2.6e-07 + assert glm["source"] == "https://docs.z.ai/guides/overview/pricing" assert glm["max_input_tokens"] == 262144