resolve pyright failure

Author: aksheyd

tests/shared/test_progress_notifications.py

@@ -7,21 +7,18 @@
 from mcp.server.lowlevel import NotificationOptions
 from mcp.server.models import InitializationOptions
 from mcp.server.session import ServerSession
-from mcp.shared.session import RequestResponder
-from mcp.types import (
-    JSONRPCMessage,
-)
+from mcp.shared.session import RequestResponder, SessionMessage
 
 
 @pytest.mark.anyio
 async def test_bidirectional_progress_notifications():
     """Test that both client and server can send progress notifications."""
     # Create memory streams for client/server
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[
-        JSONRPCMessage
+        SessionMessage
     ](5)
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[
-        JSONRPCMessage
+        SessionMessage
     ](5)
 
     # Run a server session so we can send progress updates in tool