mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-15 01:39:41 +00:00
Revert "Revert "set task_id in skyvern_context when task block starts"" (#2169)
This commit is contained in:
parent
9aa9e3f55c
commit
e28f769010
1 changed files with 4 additions and 0 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue