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

fix: improve model ratio handling for reserved models in getHardcodedCompletionModelRatio

CaIon 6 месяцев назад
Родитель
Сommit
c9f5b1de1a
1 измененных файлов с 7 добавлено и 1 удалено
  1. 7 1
      setting/ratio_setting/model_ratio.go

+ 7 - 1
setting/ratio_setting/model_ratio.go

@@ -468,7 +468,13 @@ func GetCompletionRatio(name string) float64 {
 
 
 func getHardcodedCompletionModelRatio(name string) (float64, bool) {
 func getHardcodedCompletionModelRatio(name string) (float64, bool) {
 	lowercaseName := strings.ToLower(name)
 	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 strings.HasPrefix(name, "gpt-4o") {
 			if name == "gpt-4o-2024-05-13" {
 			if name == "gpt-4o-2024-05-13" {
 				return 3, true
 				return 3, true