scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

fix: load kimi-k2.6 registry tests from local backup instead of remote cost map

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-21 19:44:14 -07:00
parent 28ff436a52
commit f2c38de65a
No known key found for this signature in database
1 changed files with 17 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/test_litellm/llms/moonshot/test_moonshot_chat_transformation.py
View file

@ -18,6 +18,7 @@ import pytest
import litellm
import litellm.utils
from litellm import completion
from litellm.litellm_core_utils.get_model_cost_map import GetModelCostMap
from litellm.llms.moonshot.chat.transformation import MoonshotChatConfig
@ -658,35 +659,39 @@ class TestMoonshotConfig:
class TestKimiK26ModelRegistry:
"""Tests that kimi-k2.6 is correctly registered in the model registry."""
def test_kimi_k26_in_model_cost_map(self):
"""kimi-k2.6 should be present in the model cost map."""
model_info = litellm.model_cost.get("moonshot/kimi-k2.6")
assert model_info is not None, "moonshot/kimi-k2.6 not found in model_cost"
@pytest.fixture(autouse=True)
def model_cost_map(self):
"""Load directly from the bundled backup so tests don't depend on remote fetch."""
return GetModelCostMap.load_local_model_cost_map()
def test_kimi_k26_pricing(self):
def test_kimi_k26_in_model_cost_map(self, model_cost_map):
"""kimi-k2.6 should be present in the model cost map."""
assert "moonshot/kimi-k2.6" in model_cost_map, "moonshot/kimi-k2.6 not found in model_cost"
def test_kimi_k26_pricing(self, model_cost_map):
"""kimi-k2.6 pricing should match official Kimi API rates."""
model_info = litellm.model_cost["moonshot/kimi-k2.6"]
model_info = model_cost_map["moonshot/kimi-k2.6"]
assert model_info["input_cost_per_token"] == pytest.approx(9.5e-07)
assert model_info["output_cost_per_token"] == pytest.approx(4e-06)
assert model_info["cache_read_input_token_cost"] == pytest.approx(1.6e-07)
def test_kimi_k26_context_window(self):
def test_kimi_k26_context_window(self, model_cost_map):
"""kimi-k2.6 should have a 256K (262144 token) context window."""
model_info = litellm.model_cost["moonshot/kimi-k2.6"]
model_info = model_cost_map["moonshot/kimi-k2.6"]
assert model_info["max_input_tokens"] == 262144
assert model_info["max_output_tokens"] == 262144
assert model_info["max_tokens"] == 262144
def test_kimi_k26_capabilities(self):
def test_kimi_k26_capabilities(self, model_cost_map):
"""kimi-k2.6 should support function calling, vision, video input, tool choice, and reasoning."""
model_info = litellm.model_cost["moonshot/kimi-k2.6"]
model_info = model_cost_map["moonshot/kimi-k2.6"]
assert model_info.get("supports_function_calling") is True
assert model_info.get("supports_tool_choice") is True
assert model_info.get("supports_vision") is True
assert model_info.get("supports_video_input") is True
assert model_info.get("supports_reasoning") is True
def test_kimi_k26_provider(self):
def test_kimi_k26_provider(self, model_cost_map):
"""kimi-k2.6 should be assigned to the moonshot provider."""
model_info = litellm.model_cost["moonshot/kimi-k2.6"]
model_info = model_cost_map["moonshot/kimi-k2.6"]
assert model_info["litellm_provider"] == "moonshot"