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

@ -914,6 +914,7 @@ class HierarchicalSwarm:
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"
)
raise
def agents_no_print(self):
for agent in self.agents:

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

Loading…
Cancel
Save