From e28f769010982d7027b83ce108e3edeb406f2521 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 16 Apr 2025 11:49:38 -0700 Subject: [PATCH] Revert "Revert "set task_id in skyvern_context when task block starts"" (#2169) --- skyvern/forge/sdk/workflow/models/block.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index 03ace09d..5a49bed4 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -613,6 +613,8 @@ class BaseTaskBlock(Block): raise e try: + current_context = skyvern_context.ensure_context() + current_context.task_id = task.task_id await app.agent.execute_step( organization=organization, task=task, @@ -631,6 +633,8 @@ class BaseTaskBlock(Block): failure_reason=str(e), ) raise e + finally: + current_context.task_id = None # Check task status updated_task = await app.DATABASE.get_task(