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/sdk/workflow/service.py
View File

@@ -859,7 +859,7 @@ class WorkflowService:
)
LOG.info("Persisted browser session for workflow run", workflow_run_id=workflow_run.workflow_run_id)
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks_for_tasks(all_workflow_task_ids)
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks(all_workflow_task_ids)
try:
async with asyncio.timeout(SAVE_DOWNLOADED_FILES_TIMEOUT):