diff --git a/examples/mcp_example/mock_math_server.py b/examples/mcp_example/mock_math_server.py index 42187e2e..1f8df921 100644 --- a/examples/mcp_example/mock_math_server.py +++ b/examples/mcp_example/mock_math_server.py @@ -28,4 +28,4 @@ def divide(a: int, b: int) -> float: if __name__ == "__main__": print("Starting Mock Math Server on port 8000...") - mcp.run(transport="sse", port=8000, host="0.0.0.0") + mcp.run(transport="sse", host="0.0.0.0", port=8000) diff --git a/examples/mcp_example/mock_multi_agent.py b/examples/mcp_example/mock_multi_agent.py index 9216c127..cdcd52b8 100644 --- a/examples/mcp_example/mock_multi_agent.py +++ b/examples/mcp_example/mock_multi_agent.py @@ -17,7 +17,10 @@ class MathAgent: system_prompt=f"You are {name}, a math processing agent. Use the provided tools to solve math problems.", max_loops=1, mcp_servers=[self.server], - streaming_on=True + streaming_on=False, + model_name="gpt-4o-mini", + temperature=0.1, + max_tokens=1000 ) async def process(self, task: str): @@ -26,7 +29,7 @@ class MathAgent: return { "agent": self.agent.agent_name, "task": task, - "response": response + "response": str(response) } except Exception as e: logging.error(f"Error in {self.agent.agent_name}: {str(e)}")