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

reset skyvern_context after task v2 block is done (#2168)

Browse source
This commit is contained in:
Shuchang Zheng 2025-04-16 11:48:51 -07:00 • committed by GitHub
parent a8b8dfee5b
commit 9aa9e3f55c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 35 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

10
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
@ -2457,6 +2458,7 @@ class TaskV2Block(Block):
workflow_run = await app.DATABASE.get_workflow_run(workflow_run_id, organization_id)
if not workflow_run:
raise ValueError(f"WorkflowRun not found {workflow_run_id} when running TaskV2Block")
try:
task_v2 = await task_v2_service.initialize_task_v2(
organization=organization,
user_prompt=self.prompt,
@ -2485,6 +2487,14 @@ class TaskV2Block(Block):
max_steps_override=self.max_steps,
browser_session_id=browser_session_id,
)
finally:
skyvern_context.set(
skyvern_context.SkyvernContext(
organization_id=organization_id,
workflow_run_id=workflow_run_id,
browser_session_id=browser_session_id,
)
)
result_dict = None
if task_v2:
result_dict = task_v2.output