diff --git a/setting/operation_setting/model-ratio.go b/setting/operation_setting/model-ratio.go index 3c1166d3..1412f614 100644 --- a/setting/operation_setting/model-ratio.go +++ b/setting/operation_setting/model-ratio.go @@ -434,7 +434,7 @@ func getHardcodedCompletionModelRatio(name string) (float64, bool) { return 3, true } if strings.HasPrefix(name, "gemini-") { - if strings.HasPrefix(name, "gemini-1.5-pro") { + if strings.HasPrefix(name, "gemini-1.5") { return 4, true } else if strings.HasPrefix(name, "gemini-2.0") { return 4, true