fixed reasoning agent to not use sonnet

pull/1220/head
Steve-Dusty 1 week ago
parent 8c7670121c
commit c8e4cd6811

@ -90,7 +90,7 @@ class ReasoningAgentRouter:
majority_voting_prompt: Optional[str] = None,
reasoning_model_name: Optional[
str
] = "claude-3-5-sonnet-20240620",
] = "gpt-4o",
):
"""
Initialize the ReasoningAgentRouter with the specified configuration.

@ -37,7 +37,7 @@ class ReasoningDuo:
output_type: OutputType = "dict-all-except-first",
reasoning_model_name: Optional[
str
] = "claude-3-5-sonnet-20240620",
] = "gpt-4o",
max_loops: int = 1,
*args,
**kwargs,

@ -6,6 +6,9 @@ from swarms.agents.reasoning_agents import (
ReasoningAgentInitializationError,
ReasoningAgentRouter,
)
from dotenv import load_dotenv
load_dotenv()
def test_router_initialization():
@ -55,7 +58,7 @@ def test_router_initialization():
eval=True,
random_models_on=True,
majority_voting_prompt="Custom voting prompt",
reasoning_model_name="claude-3-5-sonnet-20240620",
reasoning_model_name="gpt-4o",
)
assert (
custom_router is not None

Loading…
Cancel
Save