fix fmt

Author: freeznet

src/agents/mcp/server.py

@@ -11,9 +11,9 @@
 
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from mcp import ClientSession, StdioServerParameters, Tool as MCPTool, stdio_client
+from mcp.client.session import MessageHandlerFnT
 from mcp.client.sse import sse_client
 from mcp.client.streamable_http import GetSessionIdCallback, streamablehttp_client
-from mcp.client.session import MessageHandlerFnT
 from mcp.shared.message import SessionMessage
 from mcp.types import CallToolResult, GetPromptResult, InitializeResult, ListPromptsResult
 from typing_extensions import NotRequired, TypedDict

tests/mcp/test_message_handler.py

@@ -2,16 +2,16 @@
 
 import anyio
 import pytest
+from mcp.client.session import MessageHandlerFnT
 from mcp.shared.message import SessionMessage
 from mcp.types import InitializeResult
 
 from agents.mcp.server import (
     MCPServerSse,
-    MCPServerStreamableHttp,
     MCPServerStdio,
+    MCPServerStreamableHttp,
     _MCPServerWithClientSession,
 )
-from mcp.client.session import MessageHandlerFnT
 
 
 class _StubClientSession:
@@ -53,9 +53,8 @@ def __init__(self, handler: MessageHandlerFnT | None):
     def create_streams(self):
         @contextlib.asynccontextmanager
         async def _streams():
-            send_stream, recv_stream = anyio.create_memory_object_stream[SessionMessage | Exception](
-                1
-            )
+            send_stream, recv_stream = (
+                anyio.create_memory_object_stream[SessionMessage | Exception](1))
             try:
                 yield recv_stream, send_stream, None
             finally: