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

fix restart workflow run (#1481)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-03 14:22:42 -08:00
committed by GitHub
parent d05b39f0fc
commit 60e051eeb5
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
skyvern/forge/sdk/db/client.py
View File

@@ -2132,6 +2132,17 @@ class AgentDB:
task = await self.get_task(task_id, organization_id=organization_id) task = await self.get_task(task_id, organization_id=organization_id)
return convert_to_workflow_run_block(new_workflow_run_block, task=task) 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( async def update_workflow_run_block(
self, self,
workflow_run_block_id: str, workflow_run_block_id: str,