From f084f3e3db88d00d00428e68ec85548b9182301d Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Tue, 20 May 2025 19:05:38 -0400 Subject: [PATCH] Trap timeouterror --- src/fastmcp/client/client.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/fastmcp/client/client.py b/src/fastmcp/client/client.py index bafdf3385..98c3298be 100644 --- a/src/fastmcp/client/client.py +++ b/src/fastmcp/client/client.py @@ -1,4 +1,3 @@ -import asyncio import datetime from contextlib import AsyncExitStack, asynccontextmanager from pathlib import Path @@ -164,7 +163,7 @@ class Client: with anyio.fail_after(1): self._initialize_result = await self._session.initialize() yield - except asyncio.TimeoutError: + except TimeoutError: raise RuntimeError("Failed to initialize server session") finally: self._exit_stack = None