fix: update MCP server config and test script for single-agent multi-server setup

- Fixed `math_server.py` to support clean server initialization
- Updated `multi_server_test.py` to validate single agent interactions with multiple MCP servers

examples/mcp_example/math_server.py

@@ -29,4 +29,4 @@ def divide(a: float, b: float) -> float:
 
 if __name__ == "__main__":
     print("Starting Math Server on port 6274...")
-    mcp.run(transport="sse", port=6274)
+    mcp.run(transport="sse", host="0.0.0.0", port=6274)

examples/mcp_example/multi_server_test.py

@@ -1,8 +1,7 @@
 from swarms import Agent
 from swarms.tools.mcp_integration import MCPServerSseParams
-from swarms.prompts.finance_agent_sys_prompt import FINANCIAL_AGENT_SYS_PROMPT
 
-# Configure multiple MCP servers
+# Configure servers
 math_server = MCPServerSseParams(
     url="http://0.0.0.0:6274",
     headers={"Content-Type": "application/json"},
@@ -17,34 +16,11 @@ calc_server = MCPServerSseParams(
     sse_read_timeout=300.0
 )
 
-# Create specialized agents with different server access
+# Create agent with access to both servers
-math_agent = Agent(
+calculator = Agent(
-    agent_name="Math-Specialist",
+    agent_name="Calculator",
-    agent_description="Advanced mathematics expert",
+    agent_description="Math and calculation expert",
-    system_prompt="You are a mathematics expert. Use available math operations.",
+    system_prompt="You are a math expert. Use the available calculation tools to solve problems.",
-    max_loops=1,
-    mcp_servers=[math_server],
-    interactive=True,
-    streaming_on=True,
-    model_name="gpt-4o-mini"
-)
-
-finance_agent = Agent(
-    agent_name="Finance-Specialist",
-    agent_description="Financial calculations expert", 
-    system_prompt=FINANCIAL_AGENT_SYS_PROMPT,
-    max_loops=1,
-    mcp_servers=[calc_server],
-    interactive=True,
-    streaming_on=True,
-    model_name="gpt-4o-mini"
-)
-
-# Multi-server agent with access to all operations
-super_agent = Agent(
-    agent_name="Super-Calculator",
-    agent_description="Multi-capable calculation expert",
-    system_prompt="You have access to multiple calculation servers. Use them appropriately.",
     max_loops=1,
     mcp_servers=[math_server, calc_server],
     interactive=True,
@@ -52,62 +28,28 @@ super_agent = Agent(
     model_name="gpt-4o-mini"
 )
 
-def format_agent_output(agent_name: str, output: str) -> str:
-    return f"""
-╭─── {agent_name} Response ───╮
-{output}
-╰───────────────────────────╯
-"""
-
 def main():
-    print("\nMulti-Agent MCP Test Environment")
+    print("\nMulti-Server Calculator Test")
     print("Type 'exit' to quit\n")
-    print("Available commands:")
+    print("Example commands:")
-    print("- math: <calculation> (e.g., 'math: add 5 and 3')")
+    print("- add 5 and 3")
-    print("- finance: <calculation> (e.g., 'finance: calculate compound interest on 1000 at 5% for 3 years')")
+    print("- multiply 4 by 7\n")
-    print("- Any other calculation will use the super agent\n")
 
     while True:
         try:
-            user_input = input("\nEnter your calculation request: ")
+            user_input = input("\nEnter calculation: ")
 
             if user_input.lower() == 'exit':
                 break
 
-            # Route request to appropriate agent based on keywords
+            response = calculator.run(user_input)
-            if 'finance' in user_input.lower():
+            print(f"\nCalculation result: {response}")
-                response = finance_agent.run(user_input)
-                print("\nFinance Agent Response:")
-                print("-" * 50)
-                print(f"Response: {response}")
-                print("-" * 50)
-            elif 'math' in user_input.lower():
-                response = math_agent.run(user_input)
-                print("\nMath Agent Response:")
-                print("-" * 50) 
-                print(f"Response: {response}")
-                print("-" * 50)
-            else:
-                try:
-                    response = super_agent.run(user_input)
-                    print("\nSuper Agent Response:")
-                    print("-" * 50)
-                    if isinstance(response, dict):
-                        result = response.get('result', response)
-                        print(f"Result: {result}")
-                    elif hasattr(response, 'content'):
-                        print(f"Result: {response.content}")
-                    else:
-                        print(f"Result: {response}")
-                    print("-" * 50)
-                except Exception as e:
-                    print(f"Error processing calculation: {str(e)}")
 
         except KeyboardInterrupt:
             print("\nExiting gracefully...")
             break
         except Exception as e:
-            print(f"Error processing request: {e}")
+            print(f"Error: {e}")
 
 if __name__ == "__main__":
     main()