Pārlūkot izejas kodu

Fix: CompletionRatio is not working for openrouter.ai

https://openrouter.ai/docs#models
Model name of openrouter is prefix with company name, e.g. "model": "anthropic/claude-3-opus:beta", therefore, CompletionRatio will not working for it which is only work for prefix with claude-xxx
h1xy 1 gadu atpakaļ
vecāks
revīzija
c5f6d0e063
1 mainītis faili ar 3 papildinājumiem un 3 dzēšanām
  1. 3 3
      common/model-ratio.go

+ 3 - 3
common/model-ratio.go

@@ -206,11 +206,11 @@ func GetCompletionRatio(name string) float64 {
 		}
 		return 2
 	}
-	if strings.HasPrefix(name, "claude-instant-1") {
+	if strings.Contains(name, "claude-instant-1") {
 		return 3
-	} else if strings.HasPrefix(name, "claude-2") {
+	} else if strings.Contains(name, "claude-2") {
 		return 3
-	} else if strings.HasPrefix(name, "claude-3") {
+	} else if strings.Contains(name, "claude-3") {
 		return 5
 	}
 	if strings.HasPrefix(name, "mistral-") {