Merge pull request #1220 from Steve-Dusty/reasoning

fixed reasoning agent to not use sonnet

Author: kyegomez

swarms/agents/reasoning_agents.py

@@ -90,7 +90,7 @@ def __init__(
         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.

swarms/agents/reasoning_duo.py

@@ -37,7 +37,7 @@ def __init__(
         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,

tests/structs/test_reasoning_agent_router.py

@@ -6,6 +6,9 @@
     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