Merge pull request #1136 from aparekh02/multiagentrouter_test

[TEST] MultiAgentRouter after Refactoring

tests/structs/multiagentrouter_models_test.py

@@ -0,0 +1,59 @@
+from swarms.structs.agent import Agent
+from swarms.structs.multi_agent_router import MultiAgentRouter
+
+# Example usage:
+agents = [
+    Agent(
+        agent_name="ResearchAgent",
+        agent_description="Specializes in researching topics and providing detailed, factual information",
+        system_prompt="You are a research specialist. Provide detailed, well-researched information about any topic, citing sources when possible.",
+        max_loops=1,
+    ),
+    Agent(
+        agent_name="CodeExpertAgent",
+        agent_description="Expert in writing, reviewing, and explaining code across multiple programming languages",
+        system_prompt="You are a coding expert. Write, review, and explain code with a focus on best practices and clean code principles.",
+        max_loops=1,
+    ),
+    Agent(
+        agent_name="WritingAgent",
+        agent_description="Skilled in creative and technical writing, content creation, and editing",
+        system_prompt="You are a writing specialist. Create, edit, and improve written content while maintaining appropriate tone and style.",
+        max_loops=1,
+    ),
+]
+
+models_to_test = [
+    "gpt-4.1",
+    "gpt-4o",
+    "gpt-5-mini",
+    "o4-mini", 
+    "o3",            
+    "claude-opus-4-20250514",   
+    "claude-sonnet-4-20250514",
+    "claude-3-7-sonnet-20250219",   
+    "gemini/gemini-2.5-flash",
+    "gemini/gemini-2.5-pro",
+]
+
+task = "Use all the agents available to you to remake the Fibonacci function in Python, providing both an explanation and code."
+
+model_logs = []
+
+for model_name in models_to_test:
+    print(f"\n--- Testing model: {model_name} ---")
+    router_execute = MultiAgentRouter(
+        agents=agents, temperature=0.5, model=model_name,
+    )
+    try:
+        result = router_execute.run(task)
+        print(f"Run completed successfully for {model_name}")
+        model_logs.append({"model": model_name, "status": "✅ Success"})
+    except Exception as e:
+        print(f"An error occurred for {model_name}")
+        model_logs.append({"model": model_name, "status": f"❌ Error: {e}"})
+
+print("\n===== Model Run Summary =====")
+for log in model_logs:
+    print(f"{log['model']}: {log['status']}")
+