fix(mcp): update server parameters to resolve connection error in mcp_client.py

pull/819/head
Pavan Kumar 3 months ago committed by ascender1729
parent 812115a2a6
commit b567b30409

@ -31,26 +31,26 @@ def main():
try: try:
# Create the server parameters correctly # Create the server parameters correctly
logger.info("Creating MCP server parameters...") logger.info("Creating MCP server parameters...")
mcp_server = { mcp_server = MCPServerSseParams(
"url": "http://0.0.0.0:8000", url="http://0.0.0.0:8000",
"headers": { headers={
"Content-Type": "application/json", "Content-Type": "application/json",
"Accept": "text/event-stream" "Accept": "text/event-stream"
}, },
"timeout": 10.0, timeout=10.0,
"sse_read_timeout": 30.0 sse_read_timeout=30.0
} )
# Log the server params to verify they're correct # Log the server params to verify they're correct
logger.info(f"MCP Server URL: {mcp_server['url']}") logger.info(f"MCP Server URL: {mcp_server.url}")
logger.info(f"MCP Headers: {mcp_server['headers']}") logger.info("MCP Headers configured")
# Create agent with minimal configuration # Create agent with minimal configuration
logger.info("Creating Agent with MCP integration...") logger.info("Creating Agent with MCP integration...")
agent = Agent( agent = Agent(
agent_name="MCP Test Agent", agent_name="MCP Test Agent",
system_prompt=MATH_PROMPT, system_prompt=MATH_PROMPT,
mcp_servers=[mcp_server], # Pass server config as a list of dicts mcp_servers=[mcp_server], # Pass server params object
verbose=True) verbose=True)
print("\nAgent created successfully!") print("\nAgent created successfully!")

Loading…
Cancel
Save