Avoid hanging on initialize

This commit is contained in:
Jeremiah Lowin 2025-05-20 18:39:32 -04:00
commit be95110596

View file

@ -1,8 +1,10 @@
import asyncio
import datetime
from contextlib import AsyncExitStack, asynccontextmanager
from pathlib import Path
from typing import Any, cast
import anyio
import mcp.types
from exceptiongroup import catch
from mcp import ClientSession
@ -161,7 +163,11 @@ class Client:
self._initialize_result = await self._session.initialize()
try:
with anyio.fail_after(1):
self._initialize_result = await self._session.initialize()
yield
except asyncio.TimeoutError:
raise RuntimeError("Failed to initialize server session")
finally:
self._exit_stack = None
self._session = None