feat: add support for new OpenAI models and update billing ratios

This commit is contained in:
Laisky.Cai
2024-12-19 05:55:21 +00:00
parent 75e7fee400
commit 40f5cd3af0
3 changed files with 6 additions and 7 deletions

View File

@@ -49,7 +49,9 @@ var ModelRatio = map[string]float64{
"gpt-3.5-turbo-instruct": 0.75, // $0.0015 / 1K tokens
"gpt-3.5-turbo-1106": 0.5, // $0.001 / 1K tokens
"gpt-3.5-turbo-0125": 0.25, // $0.0005 / 1K tokens
"o1-preview": 7.5, // $15.00 / 1M input tokens
"o1": 7.5, // $15.00 / 1M input tokens
"o1-2024-12-17": 7.5,
"o1-preview": 7.5, // $15.00 / 1M input tokens
"o1-preview-2024-09-12": 7.5,
"o1-mini": 1.5, // $3.00 / 1M input tokens
"o1-mini-2024-09-12": 1.5,
@@ -397,10 +399,8 @@ func GetCompletionRatio(name string, channelType int) float64 {
}
return 2
}
if strings.HasPrefix(name, "o1-preview") {
return 4
}
if strings.HasPrefix(name, "o1-mini") {
// including o1, o1-preview, o1-mini
if strings.HasPrefix(name, "o1") {
return 4
}
if name == "chatgpt-4o-latest" {