GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-08 01:06:37 +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
app/service/chat/gemini_chat_service.py
+3
View File
@@ -206,6 +206,9 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
else: else:
# 客户端没有提供思考配置，使用默认配置 # 客户端没有提供思考配置，使用默认配置
if model.endswith("-non-thinking"): if model.endswith("-non-thinking"):
if "gemini-2.5-pro" in model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0} payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
elif model in settings.THINKING_BUDGET_MAP: elif model in settings.THINKING_BUDGET_MAP:
if settings.SHOW_THINKING_PROCESS: if settings.SHOW_THINKING_PROCESS:
app/service/chat/openai_chat_service.py
+3
View File
@@ -184,6 +184,9 @@ def _build_payload(
payload["generationConfig"]["responseModalities"] = ["Text", "Image"] payload["generationConfig"]["responseModalities"] = ["Text", "Image"]
if request.model.endswith("-non-thinking"): if request.model.endswith("-non-thinking"):
if "gemini-2.5-pro" in request.model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0} payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
if request.model in settings.THINKING_BUDGET_MAP: if request.model in settings.THINKING_BUDGET_MAP:
app/service/chat/vertex_express_chat_service.py
+3
View File
@@ -144,6 +144,9 @@ def _build_payload(model: str, request: GeminiRequest) -> Dict[str, Any]:
else: else:
# 客户端没有提供思考配置，使用默认配置 # 客户端没有提供思考配置，使用默认配置
if model.endswith("-non-thinking"): if model.endswith("-non-thinking"):
if "gemini-2.5-pro" in model:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 128}
else:
payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0} payload["generationConfig"]["thinkingConfig"] = {"thinkingBudget": 0}
elif model in settings.THINKING_BUDGET_MAP: elif model in settings.THINKING_BUDGET_MAP:
if settings.SHOW_THINKING_PROCESS: if settings.SHOW_THINKING_PROCESS: