From dad53e1f6a869c874dd845670b33014778c02dcc Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 17 Oct 2024 23:24:12 -0700 Subject: [PATCH] remove task actions (#998) --- skyvern/forge/sdk/db/client.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 57015479..13efc5a5 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -1629,3 +1629,15 @@ class AgentDB: ) actions = (await session.scalars(query)).all() return [Action.model_validate(action) for action in actions] + + async def delete_task_actions(self, organization_id: str, task_id: str) -> None: + async with self.Session() as session: + # delete actions by filtering organization_id and task_id + stmt = delete(ActionModel).where( + and_( + ActionModel.organization_id == organization_id, + ActionModel.task_id == task_id, + ) + ) + await session.execute(stmt) + await session.commit()