From cbe00d54b2b9715083fe1b78ec460392cbbab670 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=87=AA=E7=94=B1=E7=9A=84=E4=B8=96=E7=95=8C=E4=BA=BA?=
 <3196812536@qq.com>
Date: Fri, 6 Jun 2025 03:42:52 +0800
Subject: [PATCH] hotfix: gemini-2.5-pro-preview-06-05 using error (#6870)

---
 src/renderer/src/providers/AiProvider/GeminiProvider.ts | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/renderer/src/providers/AiProvider/GeminiProvider.ts b/src/renderer/src/providers/AiProvider/GeminiProvider.ts
index af1adb375e..1d69285f10 100644
--- a/src/renderer/src/providers/AiProvider/GeminiProvider.ts
+++ b/src/renderer/src/providers/AiProvider/GeminiProvider.ts
@@ -291,8 +291,7 @@ export default class GeminiProvider extends BaseProvider {
       if (reasoningEffort === undefined) {
         return {
           thinkingConfig: {
-            includeThoughts: false,
-            thinkingBudget: 0
+            includeThoughts: false
           } as ThinkingConfig
         }
       }
@@ -308,11 +307,11 @@ export default class GeminiProvider extends BaseProvider {
       }
 
       const { max } = findTokenLimit(model.id) || { max: 0 }
+      const budget = Math.floor(max * effortRatio)
 
-      // 如果thinking_budget是明确设置的值（包括0），使用该值
       return {
         thinkingConfig: {
-          thinkingBudget: Math.floor(max * effortRatio),
+          ...(budget > 0 ? { thinkingBudget: budget } : {}),
           includeThoughts: true
         } as ThinkingConfig
       }