Permitting graceful termination when we close intentionally

Author: blast-hardcheese

src/replit_river/v2/session.py

@@ -20,6 +20,7 @@
 )
 
 import nanoid
+from websockets import ConnectionClosedOK
 import websockets.asyncio.client
 from aiochannel import Channel, ChannelEmpty, ChannelFull
 from aiochannel.errors import ChannelClosed
@@ -1184,6 +1185,9 @@ async def websocket_closed_callback() -> None:
 
             try:
                 data = await ws.recv(decode=False)
+            except ConnectionClosedOK as e:
+                close_session(e)
+                continue
             except ConnectionClosed as e:
                 logger.debug(
                     "_do_ensure_connected: Connection closed during waiting "