From d11d756b61111f66fdac0096533f0d11d7a0e0dd Mon Sep 17 00:00:00 2001 From: ikeshav26 Date: Tue, 9 Dec 2025 15:44:05 +0530 Subject: [PATCH] fix: issues in error handling --- demo/web/app.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/demo/web/app.py b/demo/web/app.py index dc54f4a..65a5d66 100644 --- a/demo/web/app.py +++ b/demo/web/app.py @@ -467,6 +467,11 @@ async def websocket_stream(ws: WebSocket) -> None: print("Client disconnected (WebSocketDisconnect)") enqueue_log("client_disconnected") stop_signal.set() + except Exception as e: + print(f"Error in websocket stream: {e}") + traceback.print_exc() + enqueue_log("backend_error", message=str(e)) + stop_signal.set() finally: stop_signal.set() enqueue_log("backend_stream_complete") @@ -483,8 +488,11 @@ async def websocket_stream(ws: WebSocket) -> None: log_queue.get_nowait() except Empty: break - if ws.client_state == WebSocketState.CONNECTED: - await ws.close() + try: + if ws.client_state == WebSocketState.CONNECTED: + await ws.close() + except Exception as e: + print(f"Error closing websocket: {e}") print("WS handler exit") finally: if acquired: