fix(mcp): resolve indentation error in mcp_client.py

examples/mcp_example/mcp_client.py

@@ -1,83 +1,84 @@
-        from swarms import Agent
+
-        from swarms.tools.mcp_integration import MCPServerSseParams
+from swarms import Agent
-        from loguru import logger
+from swarms.tools.mcp_integration import MCPServerSseParams
-
+from loguru import logger
-        # Comprehensive math prompt that encourages proper JSON formatting
+
-        MATH_AGENT_PROMPT = """
+# Comprehensive math prompt that encourages proper JSON formatting
-        You are a helpful math calculator assistant.
+MATH_AGENT_PROMPT = """
-
+You are a helpful math calculator assistant.
-        Your role is to understand natural language math requests and perform calculations.
+
-        When asked to perform calculations:
+Your role is to understand natural language math requests and perform calculations.
-
+When asked to perform calculations:
-        1. Determine the operation (add, multiply, or divide)
+
-        2. Extract the numbers from the request
+1. Determine the operation (add, multiply, or divide)
-        3. Use the appropriate math operation tool
+2. Extract the numbers from the request
-
+3. Use the appropriate math operation tool
-        FORMAT YOUR TOOL CALLS AS JSON with this format:
+
-        {"tool_name": "add", "a": <first_number>, "b": <second_number>}
+FORMAT YOUR TOOL CALLS AS JSON with this format:
-        or
+{"tool_name": "add", "a": <first_number>, "b": <second_number>}
-        {"tool_name": "multiply", "a": <first_number>, "b": <second_number>}
+or
-        or
+{"tool_name": "multiply", "a": <first_number>, "b": <second_number>}
-        {"tool_name": "divide", "a": <first_number>, "b": <second_number>}
+or
-
+{"tool_name": "divide", "a": <first_number>, "b": <second_number>}
-        Always respond with a tool call in JSON format first, followed by a brief explanation.
+
-        """
+Always respond with a tool call in JSON format first, followed by a brief explanation.
-
+"""
-        def initialize_math_system():
+
-            """Initialize the math agent with MCP server configuration."""
+def initialize_math_system():
-            # Configure the MCP server connection
+    """Initialize the math agent with MCP server configuration."""
-            math_server = MCPServerSseParams(
+    # Configure the MCP server connection
-                url="http://0.0.0.0:8000",
+    math_server = MCPServerSseParams(
-                headers={"Content-Type": "application/json"},
+        url="http://0.0.0.0:8000",
-                timeout=5.0,
+        headers={"Content-Type": "application/json"},
-                sse_read_timeout=30.0
+        timeout=5.0,
-            )
+        sse_read_timeout=30.0
-
+    )
-            # Create the agent with the MCP server configuration
+
-            math_agent = Agent(
+    # Create the agent with the MCP server configuration
-                agent_name="Math Assistant",
+    math_agent = Agent(
-                agent_description="Friendly math calculator",
+        agent_name="Math Assistant",
-                system_prompt=MATH_AGENT_PROMPT,
+        agent_description="Friendly math calculator",
-                max_loops=1,
+        system_prompt=MATH_AGENT_PROMPT,
-                mcp_servers=[math_server],  # Pass MCP server config as a list
+        max_loops=1,
-                model_name="gpt-3.5-turbo",
+        mcp_servers=[math_server],  # Pass MCP server config as a list
-                verbose=True  # Enable verbose mode to see more details
+        model_name="gpt-3.5-turbo",
-            )
+        verbose=True  # Enable verbose mode to see more details
-
+    )
-            return math_agent
+
-
+    return math_agent
-        def main():
+
+def main():
+    try:
+        logger.info("Initializing math system...")
+        math_agent = initialize_math_system()
+
+        print("\nMath Calculator Ready!")
+        print("Ask me any math question!")
+        print("Examples: 'what is 5 plus 3?' or 'can you multiply 4 and 6?'")
+        print("Type 'exit' to quit\n")
+
+        while True:
             try:
-                logger.info("Initializing math system...")
+                query = input("What would you like to calculate? ").strip()
-                math_agent = initialize_math_system()
+                if not query:
-
+                    continue
-                print("\nMath Calculator Ready!")
+                if query.lower() == 'exit':
-                print("Ask me any math question!")
+                    break
-                print("Examples: 'what is 5 plus 3?' or 'can you multiply 4 and 6?'")
+
-                print("Type 'exit' to quit\n")
+                logger.info(f"Processing query: {query}")
-
+                result = math_agent.run(query)
-                while True:
+                print(f"\nResult: {result}\n")
-                    try:
-                        query = input("What would you like to calculate? ").strip()
-                        if not query:
-                            continue
-                        if query.lower() == 'exit':
-                            break
-
-                        logger.info(f"Processing query: {query}")
-                        result = math_agent.run(query)
-                        print(f"\nResult: {result}\n")
-
-                    except KeyboardInterrupt:
-                        print("\nGoodbye!")
-                        break
-                    except Exception as e:
-                        logger.error(f"Error processing query: {e}")
-                        print(f"Sorry, there was an error: {str(e)}")
 
+            except KeyboardInterrupt:
+                print("\nGoodbye!")
+                break
             except Exception as e:
-                logger.error(f"System initialization error: {e}")
+                logger.error(f"Error processing query: {e}")
-                print(f"Failed to start the math system: {str(e)}")
+                print(f"Sorry, there was an error: {str(e)}")
+
+    except Exception as e:
+        logger.error(f"System initialization error: {e}")
+        print(f"Failed to start the math system: {str(e)}")
 
-        if __name__ == "__main__":
+if __name__ == "__main__":
-            main()
+    main()