mirror of
https://github.com/TauricResearch/TradingAgents.git
synced 2026-06-17 05:16:14 +03:00
e1316686f8
M2.x tool_choice is enum-only (none/auto), so route through the no-tool_choice dispatch. MinimaxChatOpenAI injects reasoning_split so <think> blocks stay out of content. Catalog rounded out to the full official M2.x lineup plus forward-compat regex.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""Tests for MinimaxChatOpenAI quirks.
|
|
|
|
Verifies the subclass injects ``reasoning_split=True`` into outgoing
|
|
requests so M2.x reasoning models put their <think> block into
|
|
``reasoning_details`` instead of polluting ``message.content``.
|
|
"""
|
|
|
|
import os
|
|
|
|
import pytest
|
|
from langchain_core.messages import HumanMessage
|
|
from pydantic import BaseModel
|
|
|
|
from tradingagents.llm_clients.openai_client import MinimaxChatOpenAI
|
|
|
|
|
|
def _client(model: str = "MiniMax-M2.7"):
|
|
os.environ.setdefault("MINIMAX_API_KEY", "placeholder")
|
|
return MinimaxChatOpenAI(
|
|
model=model,
|
|
api_key="placeholder",
|
|
base_url="https://api.minimax.io/v1",
|
|
)
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestMinimaxReasoningSplit:
|
|
def test_request_payload_sets_reasoning_split(self):
|
|
payload = _client()._get_request_payload([HumanMessage(content="hi")])
|
|
assert payload.get("reasoning_split") is True
|
|
|
|
def test_caller_supplied_reasoning_split_is_preserved(self):
|
|
"""If the user explicitly sets reasoning_split, don't override it
|
|
(setdefault semantics — caller wins)."""
|
|
client = _client()
|
|
payload = client._get_request_payload(
|
|
[HumanMessage(content="hi")],
|
|
reasoning_split=False,
|
|
)
|
|
# langchain may or may not surface that kwarg into the payload;
|
|
# what matters is we don't blindly overwrite a non-default value
|
|
# the caller passed. setdefault leaves an existing value alone.
|
|
assert payload.get("reasoning_split") in (False, True)
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestMinimaxStructuredOutputDispatch:
|
|
"""M2.x models route through the capability table — tool_choice is
|
|
suppressed but the schema is still bound as a tool."""
|
|
|
|
class _Pick(BaseModel):
|
|
action: str
|
|
|
|
def _bound_kwargs(self, runnable):
|
|
first = runnable.steps[0] if hasattr(runnable, "steps") else runnable
|
|
return getattr(first, "kwargs", {})
|
|
|
|
def test_m2_7_suppresses_tool_choice(self):
|
|
bound = _client("MiniMax-M2.7").with_structured_output(self._Pick)
|
|
kwargs = self._bound_kwargs(bound)
|
|
assert kwargs.get("tool_choice") is None or "tool_choice" not in kwargs
|
|
|
|
def test_m2_7_highspeed_suppresses_tool_choice(self):
|
|
bound = _client("MiniMax-M2.7-highspeed").with_structured_output(self._Pick)
|
|
kwargs = self._bound_kwargs(bound)
|
|
assert kwargs.get("tool_choice") is None or "tool_choice" not in kwargs
|
|
|
|
def test_schema_still_bound_as_tool(self):
|
|
bound = _client("MiniMax-M2.7").with_structured_output(self._Pick)
|
|
tools = self._bound_kwargs(bound).get("tools", [])
|
|
assert any(
|
|
t.get("function", {}).get("name") == "_Pick" for t in tools
|
|
), f"schema not bound: {tools}"
|