From 4999d5874379113d71be51c7f96c84447a872244 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 12 Feb 2025 01:48:52 +0800 Subject: [PATCH] fix cache task run in db client (#1757) --- skyvern/forge/sdk/db/client.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 5ad79a87..957f7220 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -2675,8 +2675,10 @@ class AgentDB: async def cache_task_run(self, run_id: str, organization_id: str | None = None) -> TaskRun: async with self.Session() as session: - task_run = await session.scalars( - select(TaskRunModel).filter_by(organization_id=organization_id).filter_by(run_id=run_id) + task_run = ( + await session.scalars( + select(TaskRunModel).filter_by(organization_id=organization_id).filter_by(run_id=run_id) + ) ).first() if task_run: task_run.cached = True @@ -2697,5 +2699,5 @@ class AgentDB: if organization_id: query = query.filter_by(organization_id=organization_id) query = query.filter_by(cached=True).order_by(TaskRunModel.created_at.desc()) - task_run = await session.scalars(query).first() + task_run = (await session.scalars(query)).first() return TaskRun.model_validate(task_run) if task_run else None