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