Fix reasoning agent model and HierarchicalSwarm error handling

pull/1213/head
Steve-Dusty 1 month ago
parent b697d92952
commit b246d4e892

@ -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,

@ -914,6 +914,7 @@ class HierarchicalSwarm:
logger.error( logger.error(
f"{error_msg}\n[TRACE] Traceback: {traceback.format_exc()}\n[BUG] If this issue persists, please report it at: https://github.com/kyegomez/swarms/issues" f"{error_msg}\n[TRACE] Traceback: {traceback.format_exc()}\n[BUG] If this issue persists, please report it at: https://github.com/kyegomez/swarms/issues"
) )
raise
def agents_no_print(self): def agents_no_print(self):
for agent in self.agents: for agent in self.agents:

@ -7,6 +7,10 @@ from swarms.agents.reasoning_agents import (
ReasoningAgentRouter, ReasoningAgentRouter,
) )
from dotenv import load_dotenv
load_dotenv()
def test_router_initialization(): def test_router_initialization():
""" """
@ -55,7 +59,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
@ -250,7 +254,7 @@ def test_agent_factories():
"eval": False, "eval": False,
"random_models_on": False, "random_models_on": False,
"majority_voting_prompt": None, "majority_voting_prompt": None,
"reasoning_model_name": "claude-3-5-sonnet-20240620", "reasoning_model_name": "gpt-4o",
} }
# Test 1: Reasoning Duo factory # Test 1: Reasoning Duo factory

Loading…
Cancel
Save