vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Revert "Revert "set task_id in skyvern_context when task block starts"" (#2169)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-16 11:49:38 -07:00
committed by GitHub
parent 9aa9e3f55c
commit e28f769010
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -613,6 +613,8 @@ class BaseTaskBlock(Block):
raise e raise e
try: try:
current_context = skyvern_context.ensure_context()
current_context.task_id = task.task_id
await app.agent.execute_step( await app.agent.execute_step(
organization=organization, organization=organization,
task=task, task=task,
@@ -631,6 +633,8 @@ class BaseTaskBlock(Block):
failure_reason=str(e), failure_reason=str(e),
) )
raise e raise e
finally:
current_context.task_id = None
# Check task status # Check task status
updated_task = await app.DATABASE.get_task( updated_task = await app.DATABASE.get_task(