diff --git a/src/cai/repl/commands/mcp.py b/src/cai/repl/commands/mcp.py index b0972c1c..5f78e0e8 100644 --- a/src/cai/repl/commands/mcp.py +++ b/src/cai/repl/commands/mcp.py @@ -108,10 +108,10 @@ class GlobalMCPUtil(MCPUtil): # For SSE servers, capture the URL if isinstance(server, MCPServerSse): - server_config["url"] = getattr(server.params, "url") - server_config["headers"] = getattr(server.params, "headers") - server_config["timeout"] = getattr(server.params, "timeout", 5) - server_config["sse_read_timeout"] = getattr(server.params, "sse_read_timeout", 60 * 5) + server_config["url"] = server.params.get("url") + server_config["headers"] = server.params.get("headers") + server_config["timeout"] = server.params.get("timeout", 5) + server_config["sse_read_timeout"] = server.params.get("sse_read_timeout", 60 * 5) # For STDIO servers, capture the command elif isinstance(server, MCPServerStdio): server_config["command"] = server.params.command