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

fix cache task run in db client (#1757)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-02-12 01:48:52 +08:00
committed by GitHub
parent 9cb2a7dbaf
commit 4999d58743
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2675,8 +2675,10 @@ class AgentDB:
async def cache_task_run(self, run_id: str, organization_id: str | None = None) -> TaskRun: async def cache_task_run(self, run_id: str, organization_id: str | None = None) -> TaskRun:
async with self.Session() as session: async with self.Session() as session:
task_run = await session.scalars( task_run = (
select(TaskRunModel).filter_by(organization_id=organization_id).filter_by(run_id=run_id) await session.scalars(
select(TaskRunModel).filter_by(organization_id=organization_id).filter_by(run_id=run_id)
)
).first() ).first()
if task_run: if task_run:
task_run.cached = True task_run.cached = True
@@ -2697,5 +2699,5 @@ class AgentDB:
if organization_id: if organization_id:
query = query.filter_by(organization_id=organization_id) query = query.filter_by(organization_id=organization_id)
query = query.filter_by(cached=True).order_by(TaskRunModel.created_at.desc()) 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 return TaskRun.model_validate(task_run) if task_run else None