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

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

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-15 22:47:52 -07:00
committed by GitHub
parent f6d755da6b
commit 0d24fe3981
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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