It turns out max_size was causing session interruptions/deadlocking

Author: blast-hardcheese

src/replit_river/client_transport.py

@@ -170,7 +170,7 @@ async def _establish_new_connection(
 
             try:
                 uri_and_metadata = await self._uri_and_metadata_factory()
-                ws = await websockets.connect(uri_and_metadata["uri"])
+                ws = await websockets.connect(uri_and_metadata["uri"], max_size=None)
                 session_id = (
                     self.generate_nanoid()
                     if not old_session

src/replit_river/v2/session.py

@@ -1084,7 +1084,10 @@ async def _do_ensure_connected[HandshakeMetadata](
         ws: ClientConnection | None = None
         try:
             uri_and_metadata = await uri_and_metadata_factory()
-            ws = await websockets.asyncio.client.connect(uri_and_metadata["uri"])
+            ws = await websockets.asyncio.client.connect(
+                uri_and_metadata["uri"],
+                max_size=None,
+            )
             transition_connecting(ws)
 
             try: