From af33f36c7b8f814cbee7a734542293a35716b1cf Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Wed, 18 Sep 2024 20:39:06 +0800
Subject: [PATCH] feat: update gemini flash completion ratio #479

---
 common/model-ratio.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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 {