From 9c8ed3a70172f074f7d7e5730ae1c127aa531e2c Mon Sep 17 00:00:00 2001 From: Jonathan Dobson Date: Tue, 5 Aug 2025 15:52:10 -0400 Subject: [PATCH] add duration to WorkflowRunBlock (#3106) --- skyvern/forge/sdk/db/utils.py | 3 +++ skyvern/forge/sdk/schemas/workflow_runs.py | 1 + 2 files changed, 4 insertions(+) diff --git a/skyvern/forge/sdk/db/utils.py b/skyvern/forge/sdk/db/utils.py index 1ad90a11..411c7a27 100644 --- a/skyvern/forge/sdk/db/utils.py +++ b/skyvern/forge/sdk/db/utils.py @@ -489,6 +489,9 @@ def convert_to_workflow_run_block( modified_at=workflow_run_block_model.modified_at, ) if task: + if task.finished_at and task.started_at: + duration = task.finished_at - task.started_at + block.duration = duration.total_seconds() block.url = task.url block.navigation_goal = task.navigation_goal block.navigation_payload = task.navigation_payload diff --git a/skyvern/forge/sdk/schemas/workflow_runs.py b/skyvern/forge/sdk/schemas/workflow_runs.py index 6574ffdd..e5097d60 100644 --- a/skyvern/forge/sdk/schemas/workflow_runs.py +++ b/skyvern/forge/sdk/schemas/workflow_runs.py @@ -38,6 +38,7 @@ class WorkflowRunBlock(BaseModel): created_at: datetime modified_at: datetime include_action_history_in_verification: bool | None = False + duration: float | None = None # for loop block loop_values: list[Any] | None = None