github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2026-03-22 13:47:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Add timeout on websocket accept (#4169)

Browse Source
This commit is contained in:
tofarr
2024-10-02 10:51:12 -06:00
committed by GitHub
parent e93db80769
commit e0f8a5d508
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
openhands/server/listen.py
View File

@@ -1,3 +1,4 @@
import asyncio
import os
import re
import tempfile
@@ -257,7 +258,7 @@ async def websocket_endpoint(websocket: WebSocket):
{"action": "finish", "args": {}}
```
"""
await websocket.accept()
await asyncio.wait_for(websocket.accept(), 10)
if websocket.query_params.get('token'):
token = websocket.query_params.get('token')