Skip to content

Commit f262bb6

Browse files
committed
uv ruff fixes
1 parent 6c47890 commit f262bb6

File tree

3 files changed

+8
-10
lines changed

3 files changed

+8
-10
lines changed

src/mcp/shared/session.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff 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

tests/client/test_session.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,11 @@
22
from typing import Any
33

44
import anyio
5-
import httpx
65
import pytest
76

87
import mcp.types as types
98
from mcp.client.session import DEFAULT_CLIENT_INFO, ClientSession
109
from mcp.shared.context import RequestContext
11-
from mcp.shared.exceptions import McpError
1210
from mcp.shared.message import SessionMessage
1311
from mcp.shared.session import InMemoryRequestStateManager, RequestResponder
1412
from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS

tests/shared/test_streamable_http.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)