From 3308faae610f9d2df8895255de0d795ac5b1c617 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 24 Jan 2026 23:20:15 -0800 Subject: [PATCH] fix task v2 script generation (#4546) --- skyvern/services/task_v2_service.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/skyvern/services/task_v2_service.py b/skyvern/services/task_v2_service.py index 043f7649..af844af3 100644 --- a/skyvern/services/task_v2_service.py +++ b/skyvern/services/task_v2_service.py @@ -755,6 +755,10 @@ async def run_task_v2_helper( context=context, screenshots=scraped_page.screenshots, ) + await app.WORKFLOW_SERVICE.generate_script_if_needed( + workflow=workflow, + workflow_run=workflow_run, + ) break if not plan: @@ -963,11 +967,10 @@ async def run_task_v2_helper( context=context, screenshots=completion_screenshots, ) - if task_v2.run_with == "code": - await app.WORKFLOW_SERVICE.generate_script_if_needed( - workflow=workflow, - workflow_run=workflow_run, - ) + await app.WORKFLOW_SERVICE.generate_script_if_needed( + workflow=workflow, + workflow_run=workflow_run, + ) break # total step number validation