Ver Fonte

feat: update gemini flash completion ratio #479

CalciumIon há 1 ano atrás
pai
commit
af33f36c7b
1 ficheiros alterados com 4 adições e 1 exclusões
  1. 4 1
      common/model-ratio.go

+ 4 - 1
common/model-ratio.go

@@ -375,7 +375,10 @@ func GetCompletionRatio(name string) float64 {
 		return 3
 		return 3
 	}
 	}
 	if strings.HasPrefix(name, "gemini-") {
 	if strings.HasPrefix(name, "gemini-") {
-		return 4
+		if strings.Contains(name, "flash") {
+			return 4
+		}
+		return 3
 	}
 	}
 	if strings.HasPrefix(name, "command") {
 	if strings.HasPrefix(name, "command") {
 		switch name {
 		switch name {