vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

WorkflowRunStatusResponse -> WorkflowRunResponse (#1959)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-17 15:19:18 -07:00
committed by GitHub
parent 8955c0498c
commit 09fb6ae60a
12 changed files with 44 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/agent/client.py
View File

@@ -7,7 +7,7 @@ from skyvern.config import settings
from skyvern.exceptions import SkyvernClientException
from skyvern.forge.sdk.schemas.task_runs import TaskRunResponse
from skyvern.forge.sdk.schemas.tasks import ProxyLocation
from skyvern.forge.sdk.workflow.models.workflow import RunWorkflowResponse, WorkflowRunStatusResponse
from skyvern.forge.sdk.workflow.models.workflow import RunWorkflowResponse, WorkflowRunResponse
class RunEngine(StrEnum):
@@ -82,7 +82,7 @@ class SkyvernClient:
async def get_workflow_run(
self,
workflow_run_id: str,
) -> WorkflowRunStatusResponse:
) -> WorkflowRunResponse:
async with httpx.AsyncClient() as client:
response = await client.get(
f"{self.base_url}/api/v1/workflows/runs/{workflow_run_id}",
@@ -94,4 +94,4 @@ class SkyvernClient:
f"Failed to get workflow run: {response.text}",
status_code=response.status_code,
)
return WorkflowRunStatusResponse.model_validate(response.json())
return WorkflowRunResponse.model_validate(response.json())