WorkflowRunStatusResponse -> WorkflowRunResponse (#1959)

skyvern/forge/sdk/workflow/models/workflow.py

@@ -130,7 +130,7 @@ class WorkflowRunOutputParameter(BaseModel):
     created_at: datetime
 
 
-class WorkflowRunStatusResponse(BaseModel):
+class WorkflowRunResponse(BaseModel):
     workflow_id: str
     workflow_run_id: str
     status: WorkflowRunStatus

skyvern/forge/sdk/workflow/service.py

@@ -79,8 +79,8 @@ from skyvern.forge.sdk.workflow.models.workflow import (
     WorkflowRun,
     WorkflowRunOutputParameter,
     WorkflowRunParameter,
+    WorkflowRunResponse,
     WorkflowRunStatus,
-    WorkflowRunStatusResponse,
     WorkflowStatus,
 )
 from skyvern.forge.sdk.workflow.models.yaml import (
@@ -949,7 +949,7 @@ class WorkflowService:
         workflow_run_id: str,
         organization_id: str,
         include_cost: bool = False,
-    ) -> WorkflowRunStatusResponse:
+    ) -> WorkflowRunResponse:
         workflow_run = await self.get_workflow_run(workflow_run_id=workflow_run_id, organization_id=organization_id)
         if workflow_run is None:
             LOG.error(f"Workflow run {workflow_run_id} not found")
@@ -968,7 +968,7 @@ class WorkflowService:
         workflow_run_id: str,
         organization_id: str,
         include_cost: bool = False,
-    ) -> WorkflowRunStatusResponse:
+    ) -> WorkflowRunResponse:
         workflow = await self.get_workflow_by_permanent_id(workflow_permanent_id)
         if workflow is None:
             LOG.error(f"Workflow {workflow_permanent_id} not found")
@@ -1064,7 +1064,7 @@ class WorkflowService:
             # successful steps are the ones that have a status of completed and the total count of unique step.order
             successful_steps = [step for step in workflow_run_steps if step.status == StepStatus.completed]
             total_cost = 0.1 * (len(successful_steps) + len(text_prompt_blocks))
-        return WorkflowRunStatusResponse(
+        return WorkflowRunResponse(
             workflow_id=workflow.workflow_permanent_id,
             workflow_run_id=workflow_run_id,
             status=workflow_run.status,