修复scalar_one_or_none相关limit

2026-06-03 14:15:21 +08:00
parent 446450c7d2
commit 8f27794da6
24 changed files with 73 additions and 52 deletions
@@ -41,7 +41,7 @@ async def _reload_task(db, task_id: str) -> ChatGenerationTask | None:
        select(ChatGenerationTask).where(
            ChatGenerationTask.id == task_id,
            ChatGenerationTask.deleted_at.is_(None),
-        )
+        ).limit(1)
    )
    return result.scalar_one_or_none()

@@ -51,7 +51,7 @@ async def _run(task_id: str):
        result = await db.execute(select(ChatGenerationTask).where(
            ChatGenerationTask.id == task_id,
            ChatGenerationTask.deleted_at.is_(None),
-        ))
+        ).limit(1))
        task = result.scalar_one_or_none()
        if not task or task.generation_mode != "chatapi_async":
            return