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