From e8efcc0bd7e926c831c9034d22d45acad34b2f45 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 12 Apr 2025 19:17:24 -0700 Subject: [PATCH] sdk get_run bug fix (#2139) --- skyvern/agent/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/agent/client.py b/skyvern/agent/client.py index 393ce2de..30057f94 100644 --- a/skyvern/agent/client.py +++ b/skyvern/agent/client.py @@ -81,7 +81,7 @@ class SkyvernClient: run_id: str, ) -> RunResponse: run_obj = await self.client.agent.get_run(run_id=run_id) - if run_obj.run_type in [RunType.task_v1, RunType.task_v2]: + if run_obj.run_type in [RunType.task_v1, RunType.task_v2, RunType.openai_cua]: return TaskRunResponse.model_validate(run_obj.dict()) elif run_obj.run_type == RunType.workflow_run: return WorkflowRunResponse.model_validate(run_obj.dict())