ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 13:34:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix

Browse source
This commit is contained in:
yihuang 2025-04-21 11:44:55 +08:00
commit 5a57f16989
No known key found for this signature in database
GPG key ID: 5D8B72A08439BE01
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import datetime
from contextlib import AbstractAsyncContextManager
from pathlib import Path
from typing import Any, Literal, cast, overload
from typing import Any, Literal, cast, overload, Tuple
import mcp.types
from mcp import ClientSession
@ -46,7 +46,7 @@ class Client:
self.transport = infer_transport(transport)
# stack to record nested context manager, None is pushed if reuse existing session
self._session_cms: list[
(AbstractAsyncContextManager[ClientSession], ClientSession)
Tuple[AbstractAsyncContextManager[ClientSession] | None, ClientSession]
] = []
self._session_kwargs: SessionKwargs = {
@ -70,7 +70,8 @@ class Client:
raise RuntimeError(
"Client is not connected. Use 'async with client:' context manager first."
)
self._session_cms[-1][1]
_, session = self._session_cms[-1]
return session
def set_roots(self, roots: RootsList | RootsHandler) -> None:
"""Set the roots for the client. This does not automatically call `send_roots_list_changed`."""
@ -88,7 +89,7 @@ class Client:
async def __aenter__(self):
if self._session_cms:
# share the current session, push a None as cms to avoid close it in aexit
# share the current session, push a None as context manager to avoid close it in aexit
_, session = self._session_cms[-1]
self._session_cms.append((None, session))
else: