Spaces:

yuntian-group
/

neural-os

Runtime error

yuntian-deng commited on Oct 10, 2024

Commit

a37042d

1 Parent(s): c54f817

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -84,7 +84,6 @@ async def websocket_endpoint(websocket: WebSocket):
             except asyncio.TimeoutError:
                 print("WebSocket connection timed out")
-                await websocket.close(code=1000)
                 break
             except WebSocketDisconnect:
@@ -96,4 +95,4 @@ async def websocket_endpoint(websocket: WebSocket):
     finally:
         print("WebSocket connection closed")
-        await websocket.close()

             except asyncio.TimeoutError:
                 print("WebSocket connection timed out")
                 break
             except WebSocketDisconnect:
     finally:
         print("WebSocket connection closed")
+        # Remove the explicit websocket.close() call here