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

mark task v2 timed_out (#2113)
Some checks failed
Run tests and pre-commit / test (push) Failing after 33s
Details
Run tests and pre-commit / fe-lint-build (push) Successful in 1m33s
Details
Publish Fern Docs / run (push) Failing after 46s
Details
Sync to skyvern-cloud / sync (push) Failing after 3s
Details

Browse source
This commit is contained in:
Shuchang Zheng 2025-04-07 11:54:39 -04:00 • committed by GitHub
parent f7ca61c4bd
commit 5145bc8a1b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 29 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
skyvern/forge/sdk/workflow/service.py
View file

@ -770,6 +770,18 @@ class WorkflowService:
status=WorkflowRunStatus.canceled, status=WorkflowRunStatus.canceled,
) )
async def mark_workflow_run_as_timed_out(self, workflow_run_id: str, failure_reason: str | None = None) -> None:
LOG.info(
f"Marking workflow run {workflow_run_id} as timed out",
workflow_run_id=workflow_run_id,
workflow_status="timed_out",
)
await app.DATABASE.update_workflow_run(
workflow_run_id=workflow_run_id,
status=WorkflowRunStatus.timed_out,
failure_reason=failure_reason,
)
async def get_workflow_run(self, workflow_run_id: str, organization_id: str | None = None) -> WorkflowRun: async def get_workflow_run(self, workflow_run_id: str, organization_id: str | None = None) -> WorkflowRun:
workflow_run = await app.DATABASE.get_workflow_run( workflow_run = await app.DATABASE.get_workflow_run(
workflow_run_id=workflow_run_id, workflow_run_id=workflow_run_id,

17
skyvern/services/task_v2_service.py
View file

@ -1320,6 +1320,23 @@ async def mark_task_v2_as_terminated(
return task_v2 return task_v2
async def mark_task_v2_as_timed_out(
task_v2_id: str,
workflow_run_id: str | None = None,
organization_id: str | None = None,
failure_reason: str | None = None,
) -> TaskV2:
task_v2 = await app.DATABASE.update_task_v2(
task_v2_id,
organization_id=organization_id,
status=TaskV2Status.timed_out,
)
if workflow_run_id:
await app.WORKFLOW_SERVICE.mark_workflow_run_as_timed_out(workflow_run_id, failure_reason)
await send_task_v2_webhook(task_v2)
return task_v2
def _get_extracted_data_from_block_result( def _get_extracted_data_from_block_result(
block_result: BlockResult, block_result: BlockResult,
task_type: str, task_type: str,