From 0328a8aaa41069879900442564c5aa86896bb25f Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 27 Apr 2025 16:43:47 +0000 Subject: [PATCH] Fix remaining type-checker issues in MCP connector Co-Authored-By: Joe Moura --- src/crewai/tools/mcp_connector.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/crewai/tools/mcp_connector.py b/src/crewai/tools/mcp_connector.py index 29accfb91..97a5d6a34 100644 --- a/src/crewai/tools/mcp_connector.py +++ b/src/crewai/tools/mcp_connector.py @@ -27,7 +27,7 @@ class MCPToolConnector: self.timeout = timeout self.logger = logging.getLogger(__name__) self.token_manager = TokenManager() - self._sse_client = None + self._sse_client: Optional[SSEClient] = None def connect(self) -> None: """Connect to the MCP SSE server for tools.""" @@ -60,7 +60,8 @@ class MCPToolConnector: ) try: - self._sse_client.connect() + if self._sse_client is not None: + self._sse_client.connect() except Exception as e: self.logger.error(f"Failed to connect to MCP SSE server: {str(e)}") raise @@ -76,7 +77,11 @@ class MCPToolConnector: } try: - self._sse_client.listen(event_handlers) + if self._sse_client is not None: + self._sse_client.listen(event_handlers) + else: + self.logger.error("SSE client is not initialized") + raise ValueError("SSE client is not initialized") except Exception as e: self.logger.error(f"Error listening to MCP SSE events: {str(e)}") raise