Просмотр исходного кода

feat: update FormatMatchingModelName to handle gemini-2.5-flash-lite model prefix

RedwindA 4 месяцев назад
Родитель
Сommit
1d4e746c4f
1 измененных файлов с 9 добавлено и 2 удалено
  1. 9 2
      setting/ratio_setting/model_ratio.go

+ 9 - 2
setting/ratio_setting/model_ratio.go

@@ -657,8 +657,15 @@ func GetCompletionRatioCopy() map[string]float64 {
 
 // 转换模型名,减少渠道必须配置各种带参数模型
 func FormatMatchingModelName(name string) string {
-	name = handleThinkingBudgetModel(name, "gemini-2.5-flash", "gemini-2.5-flash-thinking-*")
-	name = handleThinkingBudgetModel(name, "gemini-2.5-pro", "gemini-2.5-pro-thinking-*")
+
+	if strings.HasPrefix(name, "gemini-2.5-flash-lite") {
+		name = handleThinkingBudgetModel(name, "gemini-2.5-flash-lite", "gemini-2.5-flash-lite-thinking-*")
+	} else if strings.HasPrefix(name, "gemini-2.5-flash") {
+		name = handleThinkingBudgetModel(name, "gemini-2.5-flash", "gemini-2.5-flash-thinking-*")
+	} else if strings.HasPrefix(name, "gemini-2.5-pro") {
+		name = handleThinkingBudgetModel(name, "gemini-2.5-pro", "gemini-2.5-pro-thinking-*")
+	}
+
 	if strings.HasPrefix(name, "gpt-4-gizmo") {
 		name = "gpt-4-gizmo-*"
 	}