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

migrate to wait_for_upload_aiotasks for aiotasks cleanup (#1353)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-12-07 18:13:53 -08:00
committed by GitHub
parent 620b5bfdea
commit e20de1e734
3 changed files with 11 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/agent.py
View File

@@ -1536,7 +1536,7 @@ class ForgeAgent:
await self.cleanup_browser_and_create_artifacts(close_browser_on_completion, last_step, task)
# Wait for all tasks to complete before generating the links for the artifacts
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks_for_task(task.task_id)
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks([task.task_id])
if need_call_webhook:
await self.execute_task_webhook(task=task, last_step=last_step, api_key=api_key)