zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modify _get_optimal_batch_size

Browse Source
This commit is contained in:
朱潮 2025-11-28 17:50:45 +08:00
parent 13e2ba56a7
commit 498b98c5dc
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/fastapi_utils.py
View File

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