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"] = {