From a8e7e2db6d6aef75f50d9599817621d50629dbd3 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:49:18 +0000 Subject: [PATCH] Fix type-checker issues with proper type annotations Co-Authored-By: Joe Moura --- src/crewai/tools/mcp_connector.py | 2 +- src/crewai/utilities/sse_client.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/crewai/tools/mcp_connector.py b/src/crewai/tools/mcp_connector.py index 97a5d6a34..c0a33e946 100644 --- a/src/crewai/tools/mcp_connector.py +++ b/src/crewai/tools/mcp_connector.py @@ -1,5 +1,5 @@ import logging -from typing import Any, Callable, Dict, List, Optional, Union +from typing import Any, Callable, Dict, List, Mapping, Optional, Union from crewai.cli.authentication.utils import TokenManager from crewai.tools import BaseTool diff --git a/src/crewai/utilities/sse_client.py b/src/crewai/utilities/sse_client.py index 8ebc68c2a..5440fac22 100644 --- a/src/crewai/utilities/sse_client.py +++ b/src/crewai/utilities/sse_client.py @@ -1,6 +1,6 @@ import json import logging -from typing import Any, Callable, Dict, Optional, Union +from typing import Any, Callable, Dict, Mapping, Optional, Union from urllib.parse import urljoin import requests @@ -55,8 +55,8 @@ class SSEClient: self.headers = headers or {} self.timeout = timeout self.logger = logging.getLogger(__name__) - self._client = None - self._response = None + self._client: Optional[sseclient.SSEClient] = None + self._response: Optional[requests.Response] = None def connect(self) -> None: """Establish a connection to the SSE server.""" @@ -93,7 +93,7 @@ class SSEClient: ) raise - def listen(self, event_handlers: Optional[Dict[str, Callable[[Any], None]]] = None) -> None: + def listen(self, event_handlers: Optional[Mapping[str, Callable[[Any], None]]] = None) -> None: """Listen for SSE events and process them with registered handlers. Args: