remove asyncio.timeout

Remove asyncio.timeout() (not valid for Py3.10) and replace it by asyncio.wait_for()

src/backend/base/langflow/components/tools/mcp_sse.py

@@ -38,24 +38,26 @@
         if headers is None:
             headers = {}
         url = await self.pre_check_redirect(url)
-
         try:
-            async with asyncio.timeout(timeout_seconds):
-                sse_transport = await self.exit_stack.enter_async_context(
-                    sse_client(url, headers, timeout_seconds, sse_read_timeout_seconds)
-                )
-                self.sse, self.write = sse_transport
-                self.session = await self.exit_stack.enter_async_context(ClientSession(self.sse, self.write))
-
-                await self.session.initialize()
-
-                # List available tools
-                response = await self.session.list_tools()
-                return response.tools
+            await asyncio.wait_for(
+                self._connect_with_timeout(url, headers, timeout_seconds, sse_read_timeout_seconds),
+                timeout=timeout_seconds
+            )
+            # List available tools
+            response = await self.session.list_tools()
+            return response.tools
         except asyncio.TimeoutError as err:
             error_message = f"Connection to {url} timed out after {timeout_seconds} seconds"
             raise TimeoutError(error_message) from err
 
+    async def _connect_with_timeout(self, url: str, headers: dict[str, str] | None, timeout_seconds: int, sse_read_timeout_seconds: int):
+        sse_transport = await self.exit_stack.enter_async_context(
+            sse_client(url, headers, timeout_seconds, sse_read_timeout_seconds)
+        )
+        self.sse, self.write = sse_transport
+        self.session = await self.exit_stack.enter_async_context(ClientSession(self.sse, self.write))
+        await self.session.initialize()
+
 
 class MCPSse(Component):
     client = MCPSseClient()