vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

make task failure log more consistent with 'marking task as failed' (#782)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-09-07 16:48:50 -07:00
committed by GitHub
parent bfce9f2279
commit 433e30e84f
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
skyvern/forge/sdk/routes/streaming.py
View File

@@ -104,12 +104,12 @@ async def task_stream(
except ValidationError as e:
await websocket.send_text(f"Invalid data: {e}")
except WebSocketDisconnect:
LOG.info("WebSocket connection closed")
LOG.info("WebSocket connection closed", task_id=task_id, organization_id=organization_id)
except ConnectionClosedOK:
LOG.info("ConnectionClosedOK error while streaming", exc_info=True)
LOG.info("ConnectionClosedOK error while streaming", task_id=task_id, organization_id=organization_id)
return
except Exception:
LOG.warning("Error while streaming", exc_info=True)
LOG.warning("Error while streaming", task_id=task_id, organization_id=organization_id)
return
LOG.info("WebSocket connection closed successfully")
LOG.info("WebSocket connection closed successfully", task_id=task_id, organization_id=organization_id)
return