From c9f5b1de1a08d364b89c360fe2e34129e9e75521 Mon Sep 17 00:00:00 2001 From: CaIon Date: Mon, 25 Aug 2025 11:59:55 +0800 Subject: [PATCH] fix: improve model ratio handling for reserved models in getHardcodedCompletionModelRatio --- setting/ratio_setting/model_ratio.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/setting/ratio_setting/model_ratio.go b/setting/ratio_setting/model_ratio.go index 5382b8f1f..d295b0b21 100644 --- a/setting/ratio_setting/model_ratio.go +++ b/setting/ratio_setting/model_ratio.go @@ -468,7 +468,13 @@ func GetCompletionRatio(name string) float64 { func getHardcodedCompletionModelRatio(name string) (float64, bool) { lowercaseName := strings.ToLower(name) - if strings.HasPrefix(name, "gpt-") && !strings.HasSuffix(name, "-all") && !strings.HasSuffix(name, "-gizmo-*") { + + isReservedModel := strings.HasSuffix(name, "-all") || strings.HasSuffix(name, "-gizmo-*") + if isReservedModel { + return 2, false + } + + if strings.HasPrefix(name, "gpt-") { if strings.HasPrefix(name, "gpt-4o") { if name == "gpt-4o-2024-05-13" { return 3, true