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, majority_voting_prompt: Optional[str] = None,
reasoning_model_name: Optional[ reasoning_model_name: Optional[
str str
] = "claude-3-5-sonnet-20240620", ] = "gpt-4o",
): ):
""" """
Initialize the ReasoningAgentRouter with the specified configuration. Initialize the ReasoningAgentRouter with the specified configuration.

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

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

Loading…
Cancel
Save