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

handle unexpected llm error in task v2 runner and clean up browser (#2100)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-03 21:51:38 -04:00
committed by GitHub
parent b0b4862c2a
commit 93371cd0cd
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/services/task_v2_service.py
View File

@@ -278,6 +278,12 @@ async def run_task_v2(
organization_id=organization_id, organization_id=organization_id,
) )
finally: finally:
if task_v2.workflow_id and not workflow:
workflow = await app.WORKFLOW_SERVICE.get_workflow(task_v2.workflow_id, organization_id=organization_id)
if task_v2.workflow_run_id and not workflow_run:
workflow_run = await app.WORKFLOW_SERVICE.get_workflow_run(
task_v2.workflow_run_id, organization_id=organization_id
)
if workflow and workflow_run and workflow_run.parent_workflow_run_id is None: if workflow and workflow_run and workflow_run.parent_workflow_run_id is None:
await app.WORKFLOW_SERVICE.clean_up_workflow( await app.WORKFLOW_SERVICE.clean_up_workflow(
workflow=workflow, workflow=workflow,