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

Merge pull request #607 from jfouret/main

Browse source 
add init_timeout for mcp client
This commit is contained in:
Jeremiah Lowin 2025-05-27 08:44:48 -04:00 • committed by GitHub
commit e80db43456
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -62,6 +62,8 @@ class Client:
message_handler: Optional handler for protocol messages
progress_handler: Optional handler for progress notifications
timeout: Optional timeout for requests (seconds or timedelta)
init_timeout: Optional timeout for initial connection (seconds or
timedelta)
Examples:
```python
@ -93,6 +95,7 @@ class Client:
message_handler: MessageHandler | None = None,
progress_handler: ProgressHandler | None = None,
timeout: datetime.timedelta | float | int | None = None,
init_timeout: datetime.timedelta | float | int = 1,
):
self.transport = infer_transport(transport)
self._session: ClientSession | None = None
@ -111,6 +114,15 @@ class Client:
if isinstance(timeout, int | float):
timeout = datetime.timedelta(seconds=timeout)
if isinstance(init_timeout, int):
self._init_timeout = float(init_timeout)
elif isinstance(init_timeout, datetime.timedelta):
self._init_timeout = float(init_timeout.total_seconds())
elif isinstance(init_timeout, float):
self._init_timeout = init_timeout
else:
raise ValueError("init_timeout must be int, float or datetime.timedelta")
self._session_kwargs: SessionKwargs = {
"sampling_callback": None,
"list_roots_callback": None,
@ -168,7 +180,7 @@ class Client:
self._session = session
# Initialize the session
try:
with anyio.fail_after(1):
with anyio.fail_after(self._init_timeout):
self._initialize_result = await self._session.initialize()
yield
except TimeoutError: