Make buffered_message_sender compatible with v1

Author: blast-hardcheese

src/replit_river/common_session.py

@@ -4,6 +4,7 @@
 from typing import Any, Awaitable, Callable, Coroutine, Protocol
 
 from opentelemetry.trace import Span
+from websockets import WebSocketCommonProtocol
 from websockets.asyncio.client import ClientConnection
 
 from replit_river.messages import (
@@ -54,7 +55,7 @@ class SessionState(enum.Enum):
 async def buffered_message_sender(
     block_until_connected: Callable[[], Awaitable[None]],
     block_until_message_available: Callable[[], Awaitable[None]],
-    get_ws: Callable[[], ClientConnection | None],
+    get_ws: Callable[[], ClientConnection | WebSocketCommonProtocol | None],
     websocket_closed_callback: Callable[[], Coroutine[Any, Any, None]],
     get_next_pending: Callable[[], TransportMessage | None],
     commit: Callable[[TransportMessage], None],