From 60e051eeb5036883bf39acebe2f5a0ea69a6f053 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 3 Jan 2025 14:22:42 -0800 Subject: [PATCH] fix restart workflow run (#1481) --- skyvern/forge/sdk/db/client.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index ab1bdcda..76624711 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -2132,6 +2132,17 @@ class AgentDB: task = await self.get_task(task_id, organization_id=organization_id) return convert_to_workflow_run_block(new_workflow_run_block, task=task) + async def delete_workflow_run_blocks(self, workflow_run_id: str, organization_id: str | None = None) -> None: + async with self.Session() as session: + stmt = delete(WorkflowRunBlockModel).where( + and_( + WorkflowRunBlockModel.workflow_run_id == workflow_run_id, + WorkflowRunBlockModel.organization_id == organization_id, + ) + ) + await session.execute(stmt) + await session.commit() + async def update_workflow_run_block( self, workflow_run_block_id: str,