GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-06-01 13:49:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 更新思考配置，针对gemini-2.5-pro模型设置思考预算为128

Browse Source
This commit is contained in:
snaily
2025-07-19 22:20:55 +08:00
parent a6fe5a7022
commit 99664298b9
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/service/chat/gemini_chat_service.py
View File

@@ -206,7 +206,10 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
else:
# 客户端没有提供思考配置，使用默认配置
if model.endswith("-non-thinking"):
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
if "gemini-2.5-pro" in model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
elif model in settings.THINKING_BUDGET_MAP:
if settings.SHOW_THINKING_PROCESS:
payload["generationConfig"]["thinkingConfig"] = {

5
app/service/chat/openai_chat_service.py
View File

@@ -184,7 +184,10 @@ def _build_payload(
payload["generationConfig"]["responseModalities"] = ["Text", "Image"]
if request.model.endswith("-non-thinking"):
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
if "gemini-2.5-pro" in request.model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
if request.model in settings.THINKING_BUDGET_MAP:
if settings.SHOW_THINKING_PROCESS:

5
app/service/chat/vertex_express_chat_service.py
View File

@@ -144,7 +144,10 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
else:
# 客户端没有提供思考配置，使用默认配置
if model.endswith("-non-thinking"):
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
if "gemini-2.5-pro" in model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
elif model in settings.THINKING_BUDGET_MAP:
if settings.SHOW_THINKING_PROCESS:
payload["generationConfig"]["thinkingConfig"] = {