From f81d4bd5d84b93c417d044285a3e66c75762b8a2 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 24 Oct 2025 12:28:06 -0700 Subject: [PATCH] fix script gen asyncio create_task bug (#3798) --- skyvern/forge/sdk/workflow/service.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index 4d581002..2ccfb412 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -3007,6 +3007,17 @@ class WorkflowService: script=created_script, rendered_cache_key_value=rendered_cache_key_value, ) + aio_task_primary_key = f"{created_script.script_id}_{created_script.version}" + if aio_task_primary_key in app.ARTIFACT_MANAGER.upload_aiotasks_map: + aio_tasks = app.ARTIFACT_MANAGER.upload_aiotasks_map[aio_task_primary_key] + if aio_tasks: + await asyncio.gather(*aio_tasks) + else: + LOG.warning( + "No upload aio tasks found for script", + script_id=created_script.script_id, + version=created_script.version, + ) def should_run_script( self,