NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update test to reflect the change in reasoning effort mapping

Browse source
This commit is contained in:
teknium 2026-01-15 07:48:52 +00:00
parent b46d7effe8
commit 31a8cdc7a7
1 changed files with 10 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

26
atroposlib/tests/test_reasoning_models.py
View file

@ -164,30 +164,24 @@ def test_reasoning_config_full():
}
}
# Test for OpenAI provider (xhigh maps to high)
# Test for OpenAI provider (effort passed through directly)
extra_body = config.build_extra_body("https://api.openai.com/v1")
assert extra_body == {"reasoning_effort": "high"}
assert extra_body == {"reasoning_effort": "xhigh"}
print("✓ ReasoningConfig with full options works correctly")
def test_reasoning_config_effort_mapping():
"""Test that effort levels are correctly mapped for OpenAI."""
mappings = {
"none": "low",
"minimal": "low",
"low": "low",
"medium": "medium",
"high": "high",
"xhigh": "high",
}
"""Test that effort levels are passed through directly for OpenAI."""
# All effort levels are now passed through 1:1
effort_levels = ["none", "minimal", "low", "medium", "high", "xhigh"]
for our_effort, expected_openai in mappings.items():
config = ReasoningConfig(effort=our_effort)
for effort in effort_levels:
config = ReasoningConfig(effort=effort)
extra_body = config.build_extra_body("https://api.openai.com/v1")
assert (
extra_body["reasoning_effort"] == expected_openai
), f"Expected {our_effort} to map to {expected_openai}, got {extra_body}"
print("✓ Effort level mapping for OpenAI works correctly")
extra_body["reasoning_effort"] == effort
), f"Expected {effort} to pass through, got {extra_body}"
print("✓ Effort levels pass through correctly for OpenAI")
def test_reasoning_config_invalid_effort():