feat: enhance model ratio lookup with case-insensitive and direct matching

fix: update DeepSeek reasoner model ratio check
2025-01-26 16:07:41 +08:00 · 2025-01-25 23:09:14 +08:00
1 changed files with 8 additions and 2 deletions
@@ -342,6 +342,12 @@ func UpdateCompletionRatioByJSONString(jsonStr string) error {
 }

 func GetCompletionRatio(name string) float64 {
+	if strings.Contains(name, "/") {
+		if ratio, ok := CompletionRatio[name]; ok {
+			return ratio
+		}
+	}
+	lowercaseName := strings.ToLower(name)
 	if strings.HasPrefix(name, "gpt-4-gizmo") {
 		name = "gpt-4-gizmo-*"
 	}
@@ -404,8 +410,8 @@ func GetCompletionRatio(name string) float64 {
 			return 4
 		}
 	}
-	if strings.HasPrefix(name, "deepseek") {
-		if name == "deepseek-reasoner" {
+	if strings.HasPrefix(lowercaseName, "deepseek") {
+		if strings.HasSuffix(lowercaseName, "reasoner") || strings.HasSuffix(lowercaseName, "r1") {
 			return 4
 		}
 		return 2
Author	SHA1	Message	Date
1808837298@qq.com	b6bae9dc3a	feat: enhance model ratio lookup with case-insensitive and direct matching	2025-01-26 16:07:41 +08:00
1808837298@qq.com	2993ce37a9	fix: update DeepSeek reasoner model ratio check	2025-01-25 23:09:14 +08:00