File size: 2,191 Bytes
108e227 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | """Thinking-toggle / reasoning_effort wire invariants shared by the Moonshot- and
DeepSeek-style chat_completions profiles (all route through
``agent.reasoning_effort.thinking_toggle_extras``)."""
import pytest
from agent.reasoning_effort import DEEPSEEK_V4_EFFORTS
from providers import get_provider_profile
REASONING_MATRIX = (
None,
{"enabled": False},
{"enabled": True},
*({"enabled": True, "effort": e} for e in ("low", "medium", "high", "xhigh", "max", "none")),
)
@pytest.mark.parametrize("reasoning_config", REASONING_MATRIX, ids=str)
def test_thinking_toggle_and_effort_never_both_on_moonshot_wire(reasoning_config):
for provider, model in (("kimi-coding", "kimi-k3"), ("opencode-go", "kimi-k2.6"), ("opencode-go", "deepseek-v4-pro")):
extra_body, top_level = get_provider_profile(provider).build_api_kwargs_extras(
reasoning_config=reasoning_config, model=model
)
assert not ("thinking" in extra_body and "reasoning_effort" in top_level), (provider, model, reasoning_config)
# DeepSeek's own API wants the toggle on every request (omitting it defaults thinking on
# and then demands reasoning_content echoes); effort rides alongside only when supported.
extra_body, top_level = get_provider_profile("deepseek").build_api_kwargs_extras(
reasoning_config=reasoning_config, model="deepseek-v4-pro"
)
assert extra_body["thinking"]["type"] in ("enabled", "disabled")
assert top_level.get("reasoning_effort", DEEPSEEK_V4_EFFORTS[0]) in DEEPSEEK_V4_EFFORTS
if isinstance(reasoning_config, dict) and reasoning_config.get("enabled") is False:
assert (extra_body, top_level) == ({"thinking": {"type": "disabled"}}, {})
@pytest.mark.parametrize("reasoning_config", REASONING_MATRIX, ids=str)
def test_ox_alpha_translation_identical_on_zen_and_free(reasoning_config):
zen = get_provider_profile("opencode-zen").build_api_kwargs_extras(
reasoning_config=reasoning_config, model="x-preview-f-free"
)
free = get_provider_profile("opencode-free").build_api_kwargs_extras(
reasoning_config=reasoning_config, model="x-preview-f-free"
)
assert zen == free
|