Skip to content

Commit 6b8f737

Browse files
committed
fix ruff
1 parent 481f7ea commit 6b8f737

File tree

2 files changed

+11
-3
lines changed

2 files changed

+11
-3
lines changed

src/mcp/shared/memory.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,15 @@
1313
from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
1414

1515
import mcp.types as types
16-
from mcp.client.session import ClientSession, ElicitationFnT, ListRootsFnT, LoggingFnT, MessageHandlerFnT, SamplingFnT
17-
from mcp.client.session import ClientTransportSession
16+
from mcp.client.session import (
17+
ClientSession,
18+
ClientTransportSession,
19+
ElicitationFnT,
20+
ListRootsFnT,
21+
LoggingFnT,
22+
MessageHandlerFnT,
23+
SamplingFnT,
24+
)
1825
from mcp.server import Server
1926
from mcp.server.fastmcp import FastMCP
2027
from mcp.shared.message import SessionMessage

tests/server/test_cancel_handling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66
import pytest
77

88
import mcp.types as types
9+
from mcp.client.session import ClientSession
910
from mcp.server.lowlevel.server import Server
1011
from mcp.shared.exceptions import McpError
1112
from mcp.shared.memory import create_connected_server_and_client_session
12-
from mcp.client.session import ClientSession
1313
from mcp.types import (
1414
CallToolRequest,
1515
CallToolRequestParams,
@@ -58,6 +58,7 @@ async def handle_call_tool(name: str, arguments: dict[str, Any] | None) -> list[
5858
async with create_connected_server_and_client_session(server) as client:
5959
# First request (will be cancelled)
6060
assert isinstance(client, ClientSession)
61+
6162
async def first_request():
6263
try:
6364
await client.send_request(

0 commit comments

Comments
 (0)