mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-11 03:31:55 +00:00
fix(openrouter): treat xiaomi models as reasoning-capable
This commit is contained in:
parent
96514de472
commit
19eebf6e0d
2 changed files with 23 additions and 0 deletions
|
|
@ -8582,6 +8582,7 @@ class AIAgent:
|
||||||
"google/gemini-2",
|
"google/gemini-2",
|
||||||
"qwen/qwen3",
|
"qwen/qwen3",
|
||||||
"tencent/hy3-preview",
|
"tencent/hy3-preview",
|
||||||
|
"xiaomi/",
|
||||||
)
|
)
|
||||||
return any(model.startswith(prefix) for prefix in reasoning_model_prefixes)
|
return any(model.startswith(prefix) for prefix in reasoning_model_prefixes)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -4990,6 +4990,28 @@ class TestDeadRetryCode:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class TestSupportsReasoningExtraBody:
|
||||||
|
def _make_agent(self):
|
||||||
|
agent = object.__new__(AIAgent)
|
||||||
|
agent.provider = "openrouter"
|
||||||
|
agent.base_url = "https://openrouter.ai/api/v1"
|
||||||
|
agent._base_url_lower = agent.base_url.lower()
|
||||||
|
agent.model = ""
|
||||||
|
return agent
|
||||||
|
|
||||||
|
def test_xiaomi_models_are_treated_as_reasoning_capable(self):
|
||||||
|
agent = self._make_agent()
|
||||||
|
for model in (
|
||||||
|
"xiaomi/mimo-v2.5-pro",
|
||||||
|
"xiaomi/mimo-v2.5",
|
||||||
|
"xiaomi/mimo-v2-omni",
|
||||||
|
"xiaomi/mimo-v2-pro",
|
||||||
|
"xiaomi/mimo-v2-flash",
|
||||||
|
):
|
||||||
|
agent.model = model
|
||||||
|
assert agent._supports_reasoning_extra_body() is True, model
|
||||||
|
|
||||||
|
|
||||||
class TestMemoryContextSanitization:
|
class TestMemoryContextSanitization:
|
||||||
"""sanitize_context() helper correctness — used at provider boundaries."""
|
"""sanitize_context() helper correctness — used at provider boundaries."""
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue