fix: remove websocket object from session on disconnect

widgetti · Sep 6, 2023 · f50351e · f50351e
1 parent 0fec988
commit f50351e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/solara/server/server.py b/solara/server/server.py
@@ -135,6 +135,10 @@ async def app_loop(ws: websocket.WebsocketWrapper, session_id: str, connection_i
             try:
                 message = await ws.receive()
             except websocket.WebSocketDisconnect:
+                try:
+                    context.kernel.session.websockets.remove(ws)
+                except KeyError:
+                    pass
                 logger.debug("Disconnected")
                 return
             t0 = time.time()