modify _get_optimal_batch_size

2025-11-28 17:50:45 +08:00 · 2025-11-28 17:50:45 +08:00 · 498b98c5dc
commit 498b98c5dc
parent 13e2ba56a7
1 changed files with 3 additions and 5 deletions
--- a/utils/fastapi_utils.py
+++ b/utils/fastapi_utils.py
@ -600,14 +600,12 @@ async def call_guideline_llm(chat_history: str, guidelines_text: str, terms:str,

 def _get_optimal_batch_size(guidelines_count: int) -> int:
    """根据guidelines数量决定最优批次数量（并发数）"""
-    if guidelines_count <= 5:
+    if guidelines_count <= 10:
        return 1
-    elif guidelines_count <= 10:
-        return 2
    elif guidelines_count <= 20:
-        return 3
+        return 2
    elif guidelines_count <= 30:
-        return 4
+        return 3
    else:
        return 5