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(
447447 ) -> ReceiveResultT | None :
448448 """
449449 Joins a request previously started via start_request.
450-
450+
451451 Returns the result or None if timeout is reached.
452452 """
453453 resume = self ._request_state_manager .resume (request_id )
@@ -483,7 +483,7 @@ async def join_request(
483483
484484 if response_or_error is None :
485485 if done_on_timeout :
486- await self ._request_state_manager .close_request (request_id )
486+ await self ._request_state_manager .close_request (request_id )
487487 return None
488488 elif isinstance (response_or_error , JSONRPCError ):
489489 if response_or_error .error .code == httpx .codes .REQUEST_TIMEOUT .value :
@@ -493,11 +493,10 @@ async def join_request(
493493 else :
494494 await self ._request_state_manager .close_request (request_id )
495495 raise McpError (response_or_error .error )
496- else :
496+ else :
497497 await self ._request_state_manager .close_request (request_id )
498498 return result_type .model_validate (response_or_error .result )
499499
500-
501500 async def cancel_request (self , request_id : RequestId ) -> bool :
502501 """
503502 Cancels a request previously started via start_request
Original file line number Diff line number Diff line change 22from typing import Any
33
44import anyio
5- import httpx
65import pytest
76
87import mcp .types as types
98from mcp .client .session import DEFAULT_CLIENT_INFO , ClientSession
109from mcp .shared .context import RequestContext
11- from mcp .shared .exceptions import McpError
1210from mcp .shared .message import SessionMessage
1311from mcp .shared .session import InMemoryRequestStateManager , RequestResponder
1412from 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():
13961396 captured_request_id = await session .request_call_tool (
13971397 "long_running_with_checkpoints" , arguments = {}
13981398 )
1399-
1399+
14001400 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 ,
14031404 )
14041405
14051406 assert result is None
@@ -1474,7 +1475,7 @@ async def run_tool():
14741475 assert not any (n in captured_notifications_pre for n in captured_notifications ), (
14751476 f"{ captured_notifications_pre } -> { captured_notifications } "
14761477 )
1477-
1478+
14781479 assert len (request_state_manager_1 ._progress_callbacks ) == 0
14791480 assert len (request_state_manager_1 ._response_streams ) == 0
14801481 assert len (request_state_manager_2 ._progress_callbacks ) == 0
You can’t perform that action at this time.
0 commit comments