laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-10-05 20:28:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: resolve mypy errors in reasoning effort tests

Browse Source
This commit is contained in:
thefux
2026-09-09 12:58:27 +00:00
parent a5688f05d9
commit c95ef0f188
1 changed files with 19 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_reasoning_effort.py
+19 -16
View File
@@ -60,22 +60,24 @@ SOL_REASONING = {
def test_model_parses_openrouter_reasoning_object() -> None: def test_model_parses_openrouter_reasoning_object() -> None:
model = Model( model = Model.parse_obj(
id="openai/gpt-5.6-sol", {
name="GPT", "id": "openai/gpt-5.6-sol",
created=0, "name": "GPT",
description="", "created": 0,
context_length=1, "description": "",
architecture={ "context_length": 1,
"modality": "text", "architecture": {
"input_modalities": ["text"], "modality": "text",
"output_modalities": ["text"], "input_modalities": ["text"],
"tokenizer": "x", "output_modalities": ["text"],
"instruct_type": None, "tokenizer": "x",
}, "instruct_type": None,
pricing={"prompt": 1e-6, "completion": 1e-6}, },
reasoning=SOL_REASONING, "pricing": {"prompt": 1e-6, "completion": 1e-6},
extra_ignored_field="drop me", "reasoning": SOL_REASONING,
"extra_ignored_field": "drop me",
}
) )
assert model.reasoning is not None assert model.reasoning is not None
assert model.reasoning.supported_efforts == [ assert model.reasoning.supported_efforts == [
@@ -178,6 +180,7 @@ def test_prepare_request_body_rewrites_nested_effort() -> None:
} }
).encode() ).encode()
out = provider.prepare_request_body(body, model) out = provider.prepare_request_body(body, model)
assert out is not None
data = json.loads(out) data = json.loads(out)
assert data["reasoning"]["effort"] == "low" assert data["reasoning"]["effort"] == "low"
assert data["reasoning"]["exclude"] is False assert data["reasoning"]["exclude"] is False