From 1d4e746c4f493f3ffab8f1404c52c8fcb96b54e4 Mon Sep 17 00:00:00 2001 From: RedwindA Date: Thu, 7 Aug 2025 21:37:08 +0800 Subject: [PATCH] feat: update FormatMatchingModelName to handle gemini-2.5-flash-lite model prefix --- setting/ratio_setting/model_ratio.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/setting/ratio_setting/model_ratio.go b/setting/ratio_setting/model_ratio.go index 647cc1f4..d47b86db 100644 --- a/setting/ratio_setting/model_ratio.go +++ b/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-*" }