andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 07:42:40 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix remaining type-checker issues in MCP connector

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-27 16:43:47 +00:00
parent 66d35df858
commit 0328a8aaa4
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/tools/mcp_connector.py
View File

@@ -27,7 +27,7 @@ class MCPToolConnector:
self.timeout = timeout self.timeout = timeout
self.logger = logging.getLogger(__name__) self.logger = logging.getLogger(__name__)
self.token_manager = TokenManager() self.token_manager = TokenManager()
self._sse_client = None self._sse_client: Optional[SSEClient] = None
def connect(self) -> None: def connect(self) -> None:
"""Connect to the MCP SSE server for tools.""" """Connect to the MCP SSE server for tools."""
@@ -60,7 +60,8 @@ class MCPToolConnector:
) )
try: try:
self._sse_client.connect() if self._sse_client is not None:
self._sse_client.connect()
except Exception as e: except Exception as e:
self.logger.error(f"Failed to connect to MCP SSE server: {str(e)}") self.logger.error(f"Failed to connect to MCP SSE server: {str(e)}")
raise raise
@@ -76,7 +77,11 @@ class MCPToolConnector:
} }
try: 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: except Exception as e:
self.logger.error(f"Error listening to MCP SSE events: {str(e)}") self.logger.error(f"Error listening to MCP SSE events: {str(e)}")
raise raise