mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 17:23:15 +00:00
38dcc37040
* feat: add Kimi K2.7 model option
Surface Moonshot's newly released Kimi K2.7 in the model picker, routed
through Fireworks following the existing kimi-k2pN convention, with the
same none/low/medium/high reasoning efforts (default high) as K2.6.
Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
* fix: correct Kimi K2.7 Fireworks model id to kimi-k2p7-code
Fireworks now hosts the K2.7 release as
accounts/fireworks/models/kimi-k2p7-code (the `kimi-k2p7` slug 404s).
Point the model option and its test at the live model id.
Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
* fix: drop unsupported "none" effort for Kimi K2.7
Kimi K2.7 Code is a thinking-only model — Moonshot documents that it does
not support non-thinking mode (passing thinking={"type":"disabled"} or
reasoning_effort="none" errors). Removing "none" from the advertised
efforts so the picker can't send an invalid Fireworks request.
Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
---------
Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
import pytest
|
|
|
|
from agent.dashboard.options import SUPPORTED_MODELS
|
|
from agent.utils.model import (
|
|
fallback_model_id_for,
|
|
fireworks_reasoning_effort_for,
|
|
provider_model_kwargs,
|
|
)
|
|
|
|
|
|
def test_fireworks_reasoning_effort_maps_effort() -> None:
|
|
for effort in ("none", "low", "medium", "high", "xhigh", "max"):
|
|
assert fireworks_reasoning_effort_for(effort) == effort
|
|
assert fireworks_reasoning_effort_for("bogus") is None
|
|
assert fireworks_reasoning_effort_for(None) is None
|
|
|
|
|
|
def test_provider_model_kwargs_for_fireworks() -> None:
|
|
kwargs = provider_model_kwargs(
|
|
"fireworks:accounts/fireworks/models/kimi-k2p6",
|
|
"high",
|
|
max_tokens=16_000,
|
|
)
|
|
assert kwargs["max_tokens"] == 16_000
|
|
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",
|
|
"none",
|
|
max_tokens=16_000,
|
|
)
|
|
assert kwargs["model_kwargs"] == {"reasoning_effort": "none"}
|
|
|
|
|
|
def test_provider_model_kwargs_for_fireworks_unknown_effort_omits_reasoning() -> None:
|
|
kwargs = provider_model_kwargs(
|
|
"fireworks:accounts/fireworks/models/glm-5p1",
|
|
"bogus",
|
|
max_tokens=16_000,
|
|
)
|
|
assert "model_kwargs" not in kwargs
|
|
|
|
|
|
def test_fireworks_has_no_cross_provider_fallback() -> None:
|
|
assert fallback_model_id_for("fireworks:accounts/fireworks/models/deepseek-v4-pro") is None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("model_id", "effort"),
|
|
[(m["id"], effort) for m in SUPPORTED_MODELS for effort in m["efforts"]],
|
|
)
|
|
def test_every_supported_effort_translates_to_a_reasoning_kwarg(model_id: str, effort: str) -> None:
|
|
"""Each effort surfaced in the UI must map to a provider reasoning param."""
|
|
kwargs = provider_model_kwargs(model_id, effort, max_tokens=16_000)
|
|
assert set(kwargs) - {"max_tokens"}, (
|
|
f"{model_id} effort {effort!r} did not produce a reasoning kwarg"
|
|
)
|