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-*" }