vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 09:49:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

set task_id in skyvern_context when task block starts (#2114)

Browse source
This commit is contained in:
Shuchang Zheng 2025-04-14 09:41:59 -07:00 • committed by GitHub
parent 1ed5f22200
commit 9b32a77c59
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
skyvern/forge/sdk/workflow/models/block.py
View file

@ -48,6 +48,7 @@ from skyvern.forge.sdk.api.files import (
)
from skyvern.forge.sdk.api.llm.api_handler_factory import LLMAPIHandlerFactory
from skyvern.forge.sdk.artifact.models import ArtifactType
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.db.enums import TaskType
from skyvern.forge.sdk.schemas.files import FileInfo
from skyvern.forge.sdk.schemas.task_v2 import TaskV2Status
@ -612,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,
@ -630,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(