修复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
@@ -416,6 +416,7 @@ async def get_task(
            ChatGenerationTask.generation_mode == "chatapi_async",
            ChatGenerationTask.deleted_at.is_(None),
        )
+        .limit(1)
    )
    task = result.scalar_one_or_none()
    if not task:
@@ -464,6 +465,7 @@ async def delete_task(
            ChatGenerationTask.generation_mode == "chatapi_async",
            ChatGenerationTask.deleted_at.is_(None),
        )
+        .limit(1)
    )
    task = result.scalar_one_or_none()
    if not task:
@@ -536,6 +538,7 @@ async def retry_task(
            ChatGenerationTask.generation_mode == "chatapi_async",
            ChatGenerationTask.deleted_at.is_(None),
        )
+        .limit(1)
    )
    task = result.scalar_one_or_none()
    if not task: