andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 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.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