46 lines
2.1 KiB
Python
46 lines
2.1 KiB
Python
"""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
|