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

fix stuck cron task not found issue (#4627)

Browse Source
This commit is contained in:
LawyZheng
2026-02-04 16:15:39 +08:00
committed by GitHub
parent 112f5ce6f9
commit a7cde692f0
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -378,11 +378,10 @@ class AgentDB(BaseAlchemyDB):
async def get_task(self, task_id: str, organization_id: str | None = None) -> Task | None: async def get_task(self, task_id: str, organization_id: str | None = None) -> Task | None:
"""Get a task by its id""" """Get a task by its id"""
async with self.Session() as session: async with self.Session() as session:
if task_obj := ( query = select(TaskModel).filter_by(task_id=task_id)
await session.scalars( if organization_id is not None:
select(TaskModel).filter_by(task_id=task_id).filter_by(organization_id=organization_id) query = query.filter_by(organization_id=organization_id)
) if task_obj := (await session.scalars(query)).first():
).first():
return convert_to_task(task_obj, self.debug_enabled) return convert_to_task(task_obj, self.debug_enabled)
else: else:
LOG.info( LOG.info(
@@ -395,7 +394,7 @@ class AgentDB(BaseAlchemyDB):
async def get_tasks_by_ids( async def get_tasks_by_ids(
self, self,
task_ids: list[str], task_ids: list[str],
organization_id: str | None = None, organization_id: str,
) -> list[Task]: ) -> list[Task]:
try: try:
async with self.Session() as session: async with self.Session() as session: