From 3607032d00b272c6fb07de1092d670ce99350d4d Mon Sep 17 00:00:00 2001
From: LawyZheng <lawyzheng1106@gmail.com>
Date: Thu, 31 Jul 2025 02:23:28 +0800
Subject: [PATCH] fix task v2 cancel (#3062)

---
 skyvern/services/task_v2_service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/skyvern/services/task_v2_service.py b/skyvern/services/task_v2_service.py
index b725403a..6ca5c7d0 100644
--- a/skyvern/services/task_v2_service.py
+++ b/skyvern/services/task_v2_service.py
@@ -818,6 +818,7 @@ async def run_task_v2_helper(
             task_v2 = await update_task_v2_status_to_workflow_run_status(
                 task_v2_id=task_v2_id,
                 workflow_run_status=workflow_run.status,
+                organization_id=organization_id,
             )
             break
         if block_result.success is True:
@@ -1503,7 +1504,7 @@ async def mark_task_v2_as_timed_out(
 async def update_task_v2_status_to_workflow_run_status(
     task_v2_id: str,
     workflow_run_status: WorkflowRunStatus,
-    organization_id: str | None = None,
+    organization_id: str,
 ) -> TaskV2:
     task_v2 = await app.DATABASE.update_task_v2(
         task_v2_id,