diff --git a/agent/dashboard/options.py b/agent/dashboard/options.py index efb854fc..78215d83 100644 --- a/agent/dashboard/options.py +++ b/agent/dashboard/options.py @@ -35,6 +35,13 @@ SUPPORTED_MODELS: list[ModelOption] = [ "default_effort": "medium", "supports_images": True, }, + { + "id": "fireworks:accounts/fireworks/models/kimi-k2p7-code", + "label": "Kimi K2.7", + "efforts": ["low", "medium", "high"], + "default_effort": "high", + "supports_images": False, + }, { "id": "fireworks:accounts/fireworks/models/kimi-k2p6", "label": "Kimi K2.6", diff --git a/tests/test_fireworks_model.py b/tests/test_fireworks_model.py index 349a7301..ff68b6cd 100644 --- a/tests/test_fireworks_model.py +++ b/tests/test_fireworks_model.py @@ -25,6 +25,19 @@ def test_provider_model_kwargs_for_fireworks() -> None: assert kwargs["model_kwargs"] == {"reasoning_effort": "high"} +def test_kimi_k2p7_is_supported() -> None: + kimi_k2p7 = next( + (m for m in SUPPORTED_MODELS if m["id"].endswith("kimi-k2p7-code")), + None, + ) + assert kimi_k2p7 is not None + assert kimi_k2p7["efforts"] == ["low", "medium", "high"] + assert "none" not in kimi_k2p7["efforts"] + assert kimi_k2p7["default_effort"] == "high" + kwargs = provider_model_kwargs(kimi_k2p7["id"], "high", max_tokens=16_000) + assert kwargs["model_kwargs"] == {"reasoning_effort": "high"} + + def test_provider_model_kwargs_for_fireworks_none_disables_reasoning() -> None: kwargs = provider_model_kwargs( "fireworks:accounts/fireworks/models/deepseek-v4-pro",