From 8c08edfb787c45c982787e74973e051b62a88e05 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Tue, 31 Dec 2024 12:27:35 -0800 Subject: [PATCH] fix successful steps count (#1457) --- skyvern/forge/sdk/workflow/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index fdb822eb..82b7f8ae 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -836,7 +836,7 @@ class WorkflowService: total_steps = len(workflow_run_steps) # TODO: This is a temporary cost calculation. We need to implement a more accurate cost calculation. # successful steps are the ones that have a status of completed and the total count of unique step.order - successful_steps = set(step.order for step in workflow_run_steps if step.status == StepStatus.completed) + successful_steps = [step for step in workflow_run_steps if step.status == StepStatus.completed] total_cost = 0.1 * len(successful_steps) return WorkflowRunStatusResponse( workflow_id=workflow.workflow_permanent_id,