From 99664298b9cb740acd7b93a3c6cdf3f5d289e982 Mon Sep 17 00:00:00 2001 From: snaily Date: Sat, 19 Jul 2025 22:20:55 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=9B=B4=E6=96=B0=E6=80=9D=E8=80=83?= =?UTF-8?q?=E9=85=8D=E7=BD=AE=EF=BC=8C=E9=92=88=E5=AF=B9gemini-2.5-pro?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=AE=BE=E7=BD=AE=E6=80=9D=E8=80=83=E9=A2=84?= =?UTF-8?q?=E7=AE=97=E4=B8=BA128?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/chat/gemini_chat_service.py | 5 ++++- app/service/chat/openai_chat_service.py | 5 ++++- app/service/chat/vertex_express_chat_service.py | 5 ++++- 3 files changed, 12 insertions(+), 3 deletions(-) diff --git a/app/service/chat/gemini_chat_service.py b/app/service/chat/gemini_chat_service.py index 5af40b7..7348812 100644 --- a/app/service/chat/gemini_chat_service.py +++ b/app/service/chat/gemini_chat_service.py @@ -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"] = { diff --git a/app/service/chat/openai_chat_service.py b/app/service/chat/openai_chat_service.py index a86756b..526e3c2 100644 --- a/app/service/chat/openai_chat_service.py +++ b/app/service/chat/openai_chat_service.py @@ -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: diff --git a/app/service/chat/vertex_express_chat_service.py b/app/service/chat/vertex_express_chat_service.py index 93b4f93..fea7929 100644 --- a/app/service/chat/vertex_express_chat_service.py +++ b/app/service/chat/vertex_express_chat_service.py @@ -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"] = {