ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-23 14:04:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

Replace asserts with runtime checks

Browse source
This commit is contained in:
Jeremiah Lowin 2025-07-06 10:57:17 -04:00
commit 5f6dfe2d5c
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/fastmcp/client/client.py
View file

@ -295,7 +295,10 @@ class Client(Generic[ClientTransportT]):
async with self._context_lock:
need_to_start = self._session_task is None or self._session_task.done()
if need_to_start:
assert self._nesting_counter == 0
if self._nesting_counter != 0:
raise RuntimeError(
f"Internal error: nesting counter should be 0 when starting new session, got {self._nesting_counter}"
)
self._stop_event = anyio.Event()
self._ready_event = anyio.Event()
self._session_task = asyncio.create_task(self._session_runner())
@ -303,7 +306,10 @@ class Client(Generic[ClientTransportT]):
if self._session_task.done():
exception = self._session_task.exception()
assert exception is not None
if exception is None:
raise RuntimeError(
"Session task completed without exception but connection failed"
)
if isinstance(exception, httpx.HTTPStatusError):
raise exception
raise RuntimeError(