Merge pull request #185 from h1xy/main

Fix: CompletionRatio is not working for openrouter.ai
This commit is contained in:
Calcium-Ion 2024-04-08 23:57:37 +08:00 committed by GitHub
commit 89efed48fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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-") {