Skip to content

Commit 0ce5b79

Browse files
committed
Fixed pre-commit errors
1 parent 9327cff commit 0ce5b79

File tree

3 files changed

+6
-12
lines changed

3 files changed

+6
-12
lines changed

src/mcp/client/streamable_http.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -442,9 +442,7 @@ async def streamablehttp_client(
442442

443443
async with create_mcp_http_client(
444444
headers=transport.request_headers,
445-
timeout=httpx.Timeout(
446-
transport.timeout, read=transport.sse_read_timeout
447-
),
445+
timeout=httpx.Timeout(transport.timeout, read=transport.sse_read_timeout),
448446
) as client:
449447
async with anyio.create_task_group() as tg:
450448
# Define callbacks that need access to tg

src/mcp/server/session.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,12 +97,10 @@ def __init__(
9797

9898
async def __aenter__(self) -> Self:
9999
await super().__aenter__()
100-
self._incoming_message_stream_writer, self._incoming_message_stream_reader = (
101-
anyio.create_memory_object_stream[ServerRequestResponder](0)
102-
)
103-
self._exit_stack.push_async_callback(
104-
self._incoming_message_stream_reader.aclose
105-
)
100+
self._incoming_message_stream_writer, self._incoming_message_stream_reader = anyio.create_memory_object_stream[
101+
ServerRequestResponder
102+
](0)
103+
self._exit_stack.push_async_callback(self._incoming_message_stream_reader.aclose)
106104
return self
107105

108106
@property

src/mcp/shared/session.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -202,9 +202,7 @@ def __init__(
202202

203203
async def __aenter__(self) -> Self:
204204
async with AsyncExitStack() as exit_stack:
205-
self._task_group = await exit_stack.enter_async_context(
206-
anyio.create_task_group()
207-
)
205+
self._task_group = await exit_stack.enter_async_context(anyio.create_task_group())
208206
self._task_group.start_soon(self._receive_loop)
209207
# Using BaseSession as a context manager should not block on exit (this
210208
# would be very surprising behavior), so make sure to cancel the tasks

0 commit comments

Comments
 (0)