reformat

Author: wreed4

src/mcp/shared/session.py

@@ -353,6 +353,7 @@ async def _receive_loop(self) -> None:
                                     on_complete=lambda r: self._in_flight.pop(r.request_id, None),
                                     message_metadata=message.metadata,
                                 )
+
                                 async def _handle_received_request() -> None:
                                     await self._received_request(responder)
                                     if not responder._completed:  # type: ignore[reportPrivateUsage]
@@ -361,7 +362,6 @@ async def _handle_received_request() -> None:
                                 self._in_flight[responder.request_id] = responder
                                 tg.start_soon(_handle_received_request)
 
-
                                 if not responder._completed:  # type: ignore[reportPrivateUsage]
                                     await self._handle_incoming(responder)
                             except Exception as e:

tests/client/test_sampling_callback.py

@@ -56,10 +56,7 @@ async def test_sampling_tool(message: str):
         result = await client_session.call_tool("test_sampling", {"message": "Test message for sampling"})
         assert result.isError is True
         assert isinstance(result.content[0], TextContent)
-        assert (
-            result.content[0].text
-            == "Error executing tool test_sampling: Sampling not supported"
-        )
+        assert result.content[0].text == "Error executing tool test_sampling: Sampling not supported"
 
 
 @pytest.mark.anyio
@@ -122,9 +119,7 @@ async def make_sampling_call(call_id: str, delay: float):
             # Start operations with out-of-order timing
             tg.start_soon(make_sampling_call, "slow_call", 0.6)  # Should finish last
             tg.start_soon(make_sampling_call, "fast_call", 0.2)  # Should finish first
-            tg.start_soon(
-                make_sampling_call, "medium_call", 0.4
-            )  # Should finish middle
+            tg.start_soon(make_sampling_call, "medium_call", 0.4)  # Should finish middle
 
         # Combine results to show all completed
         combined_response = " | ".join(
@@ -138,19 +133,15 @@ async def make_sampling_call(call_id: str, delay: float):
         return combined_response
 
     # Test concurrent sampling calls with time-sort verification
-    async with create_session(
-        server._mcp_server, sampling_callback=sampling_callback
-    ) as client_session:
+    async with create_session(server._mcp_server, sampling_callback=sampling_callback) as client_session:
         # Make a request that triggers multiple concurrent sampling calls
         result = await client_session.call_tool("concurrent_sampling_tool", {})
 
         assert result.isError is False
         assert isinstance(result.content[0], TextContent)
 
         # Verify all sampling calls completed with expected responses
-        expected_result = (
-            "Response after 0.6s | Response after 0.2s | Response after 0.4s"
-        )
+        expected_result = "Response after 0.6s | Response after 0.2s | Response after 0.4s"
         assert result.content[0].text == expected_result
 
         # Key test: verify concurrent execution using time-sort

tests/shared/test_session.py

@@ -229,17 +229,13 @@ async def handle_list_tools() -> list[types.Tool]:
 
         return server
 
-    async with create_connected_server_and_client_session(
-        make_server()
-    ) as client_session:
+    async with create_connected_server_and_client_session(make_server()) as client_session:
         # Test basic async handling with a single request
         result = await client_session.send_request(
             ClientRequest(
                 types.CallToolRequest(
                     method="tools/call",
-                    params=types.CallToolRequestParams(
-                        name="timed_tool_0.1", arguments={}
-                    ),
+                    params=types.CallToolRequestParams(name="timed_tool_0.1", arguments={}),
                 )
             ),
             types.CallToolResult,