fix task v2 cancel (#3062)

This commit is contained in:
LawyZheng
2025-07-31 02:23:28 +08:00
committed by GitHub
parent 815fb28141
commit 3607032d00

View File

@@ -818,6 +818,7 @@ async def run_task_v2_helper(
task_v2 = await update_task_v2_status_to_workflow_run_status( task_v2 = await update_task_v2_status_to_workflow_run_status(
task_v2_id=task_v2_id, task_v2_id=task_v2_id,
workflow_run_status=workflow_run.status, workflow_run_status=workflow_run.status,
organization_id=organization_id,
) )
break break
if block_result.success is True: if block_result.success is True:
@@ -1503,7 +1504,7 @@ async def mark_task_v2_as_timed_out(
async def update_task_v2_status_to_workflow_run_status( async def update_task_v2_status_to_workflow_run_status(
task_v2_id: str, task_v2_id: str,
workflow_run_status: WorkflowRunStatus, workflow_run_status: WorkflowRunStatus,
organization_id: str | None = None, organization_id: str,
) -> TaskV2: ) -> TaskV2:
task_v2 = await app.DATABASE.update_task_v2( task_v2 = await app.DATABASE.update_task_v2(
task_v2_id, task_v2_id,