feat: update gemini flash completion ratio #479

077f9807 · CalciumIon · 18a55dd0 · 077f9807
Commit 077f9807 authored Sep 18, 2024 by CalciumIon
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletions

common/model-ratio.go
+4 -1

No files found.
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -375,7 +375,10 @@ func GetCompletionRatio(name string) float64 {
 		return 3
 	}
 	if strings.HasPrefix(name, "gemini-") {
-		return 4
+		if strings.Contains(name, "flash") {
+			return 4
+		}
+		return 3
 	}
 	if strings.HasPrefix(name, "command") {
 		switch name {