test: enhance MCP integration test with tool discovery and user guidance

.replit

@@ -41,3 +41,20 @@ mode = "sequential"
 [[workflows.workflow.tasks]]
 task = "shell.exec"
 args = "python examples/mcp_example/test_integration.py"
+
+[[workflows.workflow]]
+name = "Run MCP Test"
+author = 13983571
+mode = "sequential"
+
+[[workflows.workflow.tasks]]
+task = "shell.exec"
+args = "python examples/mcp_example/math_server.py & "
+
+[[workflows.workflow.tasks]]
+task = "shell.exec"
+args = "sleep 2"
+
+[[workflows.workflow.tasks]]
+task = "shell.exec"
+args = "python examples/mcp_example/test_integration.py"

examples/mcp_example/test_integration.py

@@ -1,8 +1,10 @@
+
 from swarms import Agent
 from swarms.prompts.finance_agent_sys_prompt import FINANCIAL_AGENT_SYS_PROMPT
 from swarms.tools.mcp_integration import MCPServerSseParams
 import logging
 
+def main():
     # Configure MCP server connection
     server = MCPServerSseParams(
         url="http://0.0.0.0:6274",
@@ -21,23 +23,25 @@ agent = Agent(
         streaming_on=True
     )
 
-def test_mcp_operations():
-    """Test basic MCP operations with error handling"""
     try:
-        # Test addition
-        print("\nTesting addition...")
-        add_result = agent.run("Use the add tool to add 5 and 3")
-        print("Addition result:", add_result)
+        # First get available tools from server
+        print("\nDiscovering available tools from MCP server...")
+        tools = agent.mcp_tool_handling()
+        print("\nAvailable tools:", tools)
+        
+        while True:
+            # Get user input
+            user_input = input("\nEnter a math operation (or 'exit' to quit): ")
             
-        # Test multiplication
-        print("\nTesting multiplication...")
-        mult_result = agent.run("Use the multiply tool to multiply 4 and 6") 
-        print("Multiplication result:", mult_result)
+            if user_input.lower() == 'exit':
+                break
                 
-        # Test error case
-        print("\nTesting error handling...")
-        error_result = agent.run("Use the add tool with invalid inputs")
-        print("Error handling result:", error_result)
+            # Process user input through agent
+            try:
+                result = agent.run(user_input)
+                print("\nResult:", result)
+            except Exception as e:
+                print(f"Error processing request: {e}")
 
     except Exception as e:
         logging.error(f"Test failed: {e}")
@@ -45,4 +49,4 @@ def test_mcp_operations():
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO)
-    test_mcp_operations()
+    main()