File tree Expand file tree Collapse file tree 3 files changed +8
-10
lines changed Expand file tree Collapse file tree 3 files changed +8
-10
lines changed Original file line number Diff line number Diff line change @@ -447,7 +447,7 @@ async def join_request(
447
447
) -> ReceiveResultT | None :
448
448
"""
449
449
Joins a request previously started via start_request.
450
-
450
+
451
451
Returns the result or None if timeout is reached.
452
452
"""
453
453
resume = self ._request_state_manager .resume (request_id )
@@ -483,7 +483,7 @@ async def join_request(
483
483
484
484
if response_or_error is None :
485
485
if done_on_timeout :
486
- await self ._request_state_manager .close_request (request_id )
486
+ await self ._request_state_manager .close_request (request_id )
487
487
return None
488
488
elif isinstance (response_or_error , JSONRPCError ):
489
489
if response_or_error .error .code == httpx .codes .REQUEST_TIMEOUT .value :
@@ -493,11 +493,10 @@ async def join_request(
493
493
else :
494
494
await self ._request_state_manager .close_request (request_id )
495
495
raise McpError (response_or_error .error )
496
- else :
496
+ else :
497
497
await self ._request_state_manager .close_request (request_id )
498
498
return result_type .model_validate (response_or_error .result )
499
499
500
-
501
500
async def cancel_request (self , request_id : RequestId ) -> bool :
502
501
"""
503
502
Cancels a request previously started via start_request
Original file line number Diff line number Diff line change 2
2
from typing import Any
3
3
4
4
import anyio
5
- import httpx
6
5
import pytest
7
6
8
7
import mcp .types as types
9
8
from mcp .client .session import DEFAULT_CLIENT_INFO , ClientSession
10
9
from mcp .shared .context import RequestContext
11
- from mcp .shared .exceptions import McpError
12
10
from mcp .shared .message import SessionMessage
13
11
from mcp .shared .session import InMemoryRequestStateManager , RequestResponder
14
12
from mcp .shared .version import SUPPORTED_PROTOCOL_VERSIONS
Original file line number Diff line number Diff line change @@ -1396,10 +1396,11 @@ async def run_tool():
1396
1396
captured_request_id = await session .request_call_tool (
1397
1397
"long_running_with_checkpoints" , arguments = {}
1398
1398
)
1399
-
1399
+
1400
1400
result = await session .join_call_tool (
1401
- captured_request_id , request_read_timeout_seconds = timedelta (seconds = 0.01 ),
1402
- done_on_timeout = False
1401
+ captured_request_id ,
1402
+ request_read_timeout_seconds = timedelta (seconds = 0.01 ),
1403
+ done_on_timeout = False ,
1403
1404
)
1404
1405
1405
1406
assert result is None
@@ -1474,7 +1475,7 @@ async def run_tool():
1474
1475
assert not any (n in captured_notifications_pre for n in captured_notifications ), (
1475
1476
f"{ captured_notifications_pre } -> { captured_notifications } "
1476
1477
)
1477
-
1478
+
1478
1479
assert len (request_state_manager_1 ._progress_callbacks ) == 0
1479
1480
assert len (request_state_manager_1 ._response_streams ) == 0
1480
1481
assert len (request_state_manager_2 ._progress_callbacks ) == 0
You can’t perform that action at this time.
0 commit comments