From ea64f1b2eda4a4869018e153bdc01b5b56562acd Mon Sep 17 00:00:00 2001
From: foxplaying <166147707+foxplaying@users.noreply.github.com>
Date: Thu, 21 Aug 2025 05:53:03 +0800
Subject: [PATCH] =?UTF-8?q?fix(gemini):=20=E5=AF=B9=20thinking=5Fbudget=20?=
 =?UTF-8?q?=E4=B8=8D=E5=90=8C=E6=A8=A1=E5=9E=8B=E7=9A=84=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/llm_models/model_client/gemini_client.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/llm_models/model_client/gemini_client.py b/src/llm_models/model_client/gemini_client.py
index 8afddee0..c927060f 100644
--- a/src/llm_models/model_client/gemini_client.py
+++ b/src/llm_models/model_client/gemini_client.py
@@ -345,21 +345,19 @@ class GeminiClient(BaseClient):
         """
         按模型限制思考预算范围，仅支持指定的模型（支持带数字后缀的新版本）
         """
-        # 精确匹配或更精确的包含匹配
         limits = None
         matched_key = None
 
-        # 首先尝试精确匹配
+        # 优先尝试精确匹配
         if model_id in THINKING_BUDGET_LIMITS:
             limits = THINKING_BUDGET_LIMITS[model_id]
             matched_key = model_id
         else:
-            # 如果没有精确匹配，尝试更精确的包含匹配
-            # 按键的长度降序排序，优先匹配更长的键
+            # 按 key 长度倒序，保证更长的（更具体的，如 -lite）优先
             sorted_keys = sorted(THINKING_BUDGET_LIMITS.keys(), key=len, reverse=True)
             for key in sorted_keys:
-                # 使用更精确的匹配逻辑：键必须是模型ID的一部分，但不能是部分匹配
-                if key in model_id and (model_id == key or model_id.startswith(key + "-")):
+                # 必须满足：完全等于 或者 前缀匹配（带 "-" 边界）
+                if model_id == key or model_id.startswith(key + "-"):
                     limits = THINKING_BUDGET_LIMITS[key]
                     matched_key = key
                     break