From 60d51457f1483541f5b79787f84a353e953112a7 Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Thu, 17 Apr 2025 17:38:00 +0000
Subject: [PATCH] fix: correct MCP server port binding and test response logic
- Fixed port configuration in `math_server.py` to ensure proper server startup
- Updated response handling in `multi_server_test.py` to resolve missing outputs during testing
---
examples/mcp_example/math_server.py | 2 +-
examples/mcp_example/multi_server_test.py | 23 ++++++++++++++---------
2 files changed, 15 insertions(+), 10 deletions(-)
diff --git a/examples/mcp_example/math_server.py b/examples/mcp_example/math_server.py
index 66c976c2..a4500cd6 100644
--- a/examples/mcp_example/math_server.py
+++ b/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", host="0.0.0.0", port=6274)
+ mcp.run(transport="sse", port=6274)
diff --git a/examples/mcp_example/multi_server_test.py b/examples/mcp_example/multi_server_test.py
index a44663f7..b6d316de 100644
--- a/examples/mcp_example/multi_server_test.py
+++ b/examples/mcp_example/multi_server_test.py
@@ -88,15 +88,20 @@ def main():
print(f"Response: {response}")
print("-" * 50)
else:
- response = super_agent.run(user_input)
- print("\nSuper Agent Response:")
- print("-" * 50)
- if isinstance(response, dict):
- result = response.get('result', response)
- print(f"Calculation Result: {result}")
- else:
- print(f"Calculation Result: {response}")
- print("-" * 50)
+ 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...")