修复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
@@ -109,7 +109,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":
@@ -236,7 +236,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 task: