Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bugfix] fix GetCompletionRatio for gpt-3.5-turbo-1106

Browse Source
This commit is contained in:
Jiachen 2023-11-08 15:51:30 +08:00
parent 7822af3edc
commit d790d3f83f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
common/model-ratio.go
View File

@ -88,9 +88,6 @@ func GetModelRatio(name string) float64 {
} }
func GetCompletionRatio(name string) float64 { func GetCompletionRatio(name string) float64 {
if strings.HasPrefix(name, "gpt-3.5") {
return 1.333333
}
if strings.HasPrefix(name, "gpt-4") { if strings.HasPrefix(name, "gpt-4") {
return 2 return 2
} }
@ -103,5 +100,8 @@ func GetCompletionRatio(name string) float64 {
if strings.HasPrefix(name, "gpt-3.5-turbo-1106") { if strings.HasPrefix(name, "gpt-3.5-turbo-1106") {
return 2 // Input: $0.0010 / 1K tokens Output: $0.0020 / 1K tokens return 2 // Input: $0.0010 / 1K tokens Output: $0.0020 / 1K tokens
} }
if strings.HasPrefix(name, "gpt-3.5") {
return 1.333333
}
return 1 return 1
} }