Formatting

mcp_python/shared/memory.py

@@ -15,14 +15,14 @@ from mcp_python.types import JSONRPCMessage
 
 MessageStream = tuple[
     MemoryObjectReceiveStream[JSONRPCMessage | Exception],
-    MemoryObjectSendStream[JSONRPCMessage]
+    MemoryObjectSendStream[JSONRPCMessage],
 ]
 
+
 @asynccontextmanager
-async def create_client_server_memory_streams() -> AsyncGenerator[
-    tuple[MessageStream, MessageStream],
-    None
-]:
+async def create_client_server_memory_streams() -> (
+    AsyncGenerator[tuple[MessageStream, MessageStream], None]
+):
     """
     Creates a pair of bidirectional memory streams for client-server communication.
 

mcp_python/shared/session.py

@@ -154,7 +154,8 @@ class BaseSession(
 
         try:
             with anyio.fail_after(
-                None if self._read_timeout_seconds is None
+                None
+                if self._read_timeout_seconds is None
                 else self._read_timeout_seconds.total_seconds()
             ):
                 response_or_error = await response_stream_reader.receive()
@@ -168,7 +169,6 @@ class BaseSession(
                         f"{self._read_timeout_seconds} seconds."
                     ),
                 )
-
             )
 
         if isinstance(response_or_error, JSONRPCError):