fixes after merge

Author: ihrpr

src/mcp/shared/session.py

@@ -241,7 +241,9 @@ async def send_request(
 
             # TODO: Support progress callbacks
 
-            await self._write_stream.send(SessionMessage(message=JSONRPCMessage(jsonrpc_request)))
+            await self._write_stream.send(
+                SessionMessage(message=JSONRPCMessage(jsonrpc_request))
+            )
 
             # request read timeout takes precedence over session read timeout
             timeout = None

tests/server/test_lowlevel_tool_annotations.py

@@ -8,10 +8,10 @@
 from mcp.server.lowlevel import NotificationOptions
 from mcp.server.models import InitializationOptions
 from mcp.server.session import ServerSession
+from mcp.shared.message import SessionMessage
 from mcp.shared.session import RequestResponder
 from mcp.types import (
     ClientResult,
-    JSONRPCMessage,
     ServerNotification,
     ServerRequest,
     Tool,
@@ -46,10 +46,10 @@ async def list_tools():
         ]
 
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[
-        JSONRPCMessage
+        SessionMessage
     ](10)
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[
-        JSONRPCMessage
+        SessionMessage
     ](10)
 
     # Message handler for client