diff --git a/src/fastmcp/client/client.py b/src/fastmcp/client/client.py index dcb4ff013..139ce0333 100644 --- a/src/fastmcp/client/client.py +++ b/src/fastmcp/client/client.py @@ -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