We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6e2bc1a commit 14caa21Copy full SHA for 14caa21
src/replit_river/client_session.py
@@ -81,6 +81,12 @@ async def do_close_websocket() -> None:
81
82
self._setup_heartbeats_task(do_close_websocket)
83
84
+ async def replace_with_new_websocket(
85
+ self, new_ws: websockets.WebSocketCommonProtocol
86
+ ) -> None:
87
+ await super().replace_with_new_websocket(new_ws)
88
+ await self.start_serve_responses()
89
+
90
async def start_serve_responses(self) -> None:
91
self._task_manager.create_task(self.serve())
92
0 commit comments