diff --git a/common/model-ratio.go b/common/model-ratio.go index 9dcd170..3713c2c 100644 --- 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 {