Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions atomic-agents/tests/agents/test_minimax_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
)


def _make_minimax_agent(model="MiniMax-M2.7", **agent_kwargs):
def _make_minimax_agent(model="MiniMax-M3", **agent_kwargs):
"""Helper to create a MiniMax-backed agent."""
from openai import OpenAI

Expand Down Expand Up @@ -71,7 +71,7 @@ class AnalysisOutput(BaseIOSchema):
api_key=os.environ["MINIMAX_API_KEY"],
)
client = instructor.from_openai(raw, mode=instructor.Mode.JSON)
config = AgentConfig(client=client, model="MiniMax-M2.7", mode=instructor.Mode.JSON)
config = AgentConfig(client=client, model="MiniMax-M3", mode=instructor.Mode.JSON)
agent = AtomicAgent[BasicChatInputSchema, AnalysisOutput](config)

response = agent.run(BasicChatInputSchema(chat_message="I love this product, it's amazing!"))
Expand Down
34 changes: 12 additions & 22 deletions atomic-agents/tests/agents/test_minimax_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,41 +40,31 @@ def test_minimax_agent_config(self):
client = _create_minimax_client()
config = AgentConfig(
client=client,
model="MiniMax-M2.7",
model="MiniMax-M3",
)
assert config.model == "MiniMax-M2.7"
assert config.model == "MiniMax-M3"
assert config.assistant_role == "assistant"

def test_minimax_agent_initialization(self):
"""Test that AtomicAgent can be initialized with MiniMax config."""
client = _create_minimax_client()
config = AgentConfig(
client=client,
model="MiniMax-M2.7",
model="MiniMax-M3",
)
agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
assert agent.model == "MiniMax-M2.7"
assert agent.model == "MiniMax-M3"
assert agent.assistant_role == "assistant"

def test_minimax_m25_model(self):
"""Test that M2.5 model variant works."""
client = _create_minimax_client()
config = AgentConfig(
client=client,
model="MiniMax-M2.5",
)
agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
assert agent.model == "MiniMax-M2.5"

def test_minimax_m25_highspeed_model(self):
"""Test that M2.5-highspeed model variant works."""
def test_minimax_m27_legacy_model(self):
"""Test that the legacy M2.7 model variant still works."""
client = _create_minimax_client()
config = AgentConfig(
client=client,
model="MiniMax-M2.5-highspeed",
model="MiniMax-M2.7",
)
agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
assert agent.model == "MiniMax-M2.5-highspeed"
assert agent.model == "MiniMax-M2.7"


class TestMiniMaxAgentBehavior:
Expand All @@ -96,7 +86,7 @@ def mock_minimax_instructor(self):
def minimax_agent(self, mock_minimax_instructor):
config = AgentConfig(
client=mock_minimax_instructor,
model="MiniMax-M2.7",
model="MiniMax-M3",
)
return AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)

Expand All @@ -112,7 +102,7 @@ def test_run_passes_correct_model(self, minimax_agent, mock_minimax_instructor):
user_input = BasicChatInputSchema(chat_message="Test")
minimax_agent.run(user_input)
call_kwargs = mock_minimax_instructor.chat.completions.create.call_args
assert call_kwargs.kwargs["model"] == "MiniMax-M2.7"
assert call_kwargs.kwargs["model"] == "MiniMax-M3"

def test_run_stream_with_minimax(self, minimax_agent):
"""Test that streaming works with MiniMax mock client."""
Expand All @@ -135,7 +125,7 @@ def test_system_prompt_with_minimax(self, mock_minimax_instructor):
)
config = AgentConfig(
client=mock_minimax_instructor,
model="MiniMax-M2.7",
model="MiniMax-M3",
system_prompt_generator=spg,
)
agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
Expand All @@ -146,7 +136,7 @@ def test_model_api_parameters_with_minimax(self, mock_minimax_instructor):
"""Test that custom API parameters are passed through."""
config = AgentConfig(
client=mock_minimax_instructor,
model="MiniMax-M2.7",
model="MiniMax-M3",
model_api_parameters={"temperature": 0.7, "max_tokens": 1024},
)
agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def setup_client(provider):
MiniMaxClient(base_url="https://api.minimax.io/v1", api_key=api_key),
mode=instructor.Mode.JSON,
)
model = "MiniMax-M2.7"
model = "MiniMax-M3"
model_api_parameters = {"max_tokens": 2048}
assistant_role = "assistant"
else:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ client = instructor.from_openai(
OpenAI(base_url="https://api.minimax.io/v1", api_key=os.environ["MINIMAX_API_KEY"]),
mode=Mode.JSON,
)
model = "MiniMax-M2.7"
model = "MiniMax-M3" # current default; "MiniMax-M2.7" remains available as a legacy option
api_params = {"max_tokens": 2048}
```

Expand Down
Loading