move websocket close to fix memory leak

2023-12-03 14:04:40 -05:00 · 2023-12-03 14:04:40 -05:00 · 0cc2c78f55
commit 0cc2c78f55
parent 3e34c40e73
1 changed files with 2 additions and 2 deletions
--- a/backend/main.py
+++ b/backend/main.py
@ -202,5 +202,5 @@ async def stream_code(websocket: WebSocket):
        await websocket.send_json(
            {"type": "status", "value": "Image generation failed but code is complete."}
        )
-    finally:
-        await websocket.close()
+
+    await websocket.close()