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