File tree Expand file tree Collapse file tree 3 files changed +6
-12
lines changed
Expand file tree Collapse file tree 3 files changed +6
-12
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments