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

This commit is contained in:
Shuchang Zheng 2025-04-15 22:47:52 -07:00 committed by GitHub
parent f6d755da6b
commit 0d24fe3981
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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.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
@ -613,8 +612,6 @@ 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,
@ -633,8 +630,6 @@ 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(