From bdb7a2d145b817143407552b05e98334ec0b99fa Mon Sep 17 00:00:00 2001
From: foxplaying <166147707+foxplaying@users.noreply.github.com>
Date: Wed, 24 Sep 2025 13:21:12 +0800
Subject: [PATCH] fix

---
 src/llm_models/model_client/gemini_client.py | 25 ++++++++++----------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/src/llm_models/model_client/gemini_client.py b/src/llm_models/model_client/gemini_client.py
index 9b34841c..72b7f1ba 100644
--- a/src/llm_models/model_client/gemini_client.py
+++ b/src/llm_models/model_client/gemini_client.py
@@ -339,19 +339,6 @@ def _default_normal_response_parser(
 
     return api_response, _usage_record
 
-def resolve_thinking_budget(extra_params: dict[str, Any] | None, model_id: str) -> int:
-    """
-    统一解析并裁剪 thinking_budget
-    """
-    tb = THINKING_BUDGET_AUTO
-    if extra_params and "thinking_budget" in extra_params:
-        try:
-            tb = int(extra_params["thinking_budget"])
-        except (ValueError, TypeError):
-            logger.warning(
-                f"无效的 thinking_budget 值 {extra_params['thinking_budget']}，将使用模型自动预算模式 {tb}"
-            )
-    return GeminiClient.clamp_thinking_budget(tb, model_id)
 
 @client_registry.register_client_class("gemini")
 class GeminiClient(BaseClient):
@@ -388,6 +375,18 @@ class GeminiClient(BaseClient):
         """
         limits = None
 
+        # 参数传入处理
+        tb = THINKING_BUDGET_AUTO
+        if extra_params and "thinking_budget" in extra_params:
+            try:
+                tb = int(extra_params["thinking_budget"])
+            except (ValueError, TypeError):
+                logger.warning(
+                    f"无效的 thinking_budget 值 {extra_params['thinking_budget']}，"
+                    f"将使用模型自动预算模式 {THINKING_BUDGET_AUTO}"
+                )
+                tb = THINKING_BUDGET_AUTO
+        
         # 优先尝试精确匹配
         if model_id in THINKING_BUDGET_LIMITS:
             limits = THINKING_BUDGET_LIMITS[model_id]