Skip to content

Commit 14caa21

Browse files
Adding missing call to start_serve_responses
1 parent 6e2bc1a commit 14caa21

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/replit_river/client_session.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,12 @@ async def do_close_websocket() -> None:
8181

8282
self._setup_heartbeats_task(do_close_websocket)
8383

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+
8490
async def start_serve_responses(self) -> None:
8591
self._task_manager.create_task(self.serve())
8692

0 commit comments

Comments
 (0)