feat: support gpt-4o (merge pr #196 @XiaomaiTX)

feat: support model `gpt-4o` and `gpt-4o-2024-05-13`
This commit is contained in:
Minghan Zhang 2024-05-18 12:54:52 +08:00 committed by GitHub
commit a03fa7d63b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 13 additions and 1 deletions

View File

@ -112,6 +112,8 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
"gpt-4-32k", "gpt-4-32k",
"gpt-4-32k-0314", "gpt-4-32k-0314",
"gpt-4-32k-0613", "gpt-4-32k-0613",
"gpt-4o",
"gpt-4o-2024-05-13",
"dalle", "dalle",
"dall-e-2", "dall-e-2",
"dall-e-3", "dall-e-3",

View File

@ -31,6 +31,8 @@ export const modelColorMapper: Record<string, string> = {
"gpt-4-32k": "purple-600", "gpt-4-32k": "purple-600",
"gpt-4-32k-0613": "purple-600", "gpt-4-32k-0613": "purple-600",
"gpt-4-32k-0314": "purple-600", "gpt-4-32k-0314": "purple-600",
"gpt-4o": "purple-600",
"gpt-4o-2024-05-13": "purple-600",
"dall-e-3": "purple-700", "dall-e-3": "purple-700",

View File

@ -66,6 +66,12 @@ export const pricing: PricingDataset = [
input: 0.01, input: 0.01,
output: 0.03, output: 0.03,
}, },
{
models: ["gpt-4o", "gpt-4o-2024-05-13"],
input: 0.005,
output: 0.015,
},
{ {
models: ["gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613"], models: ["gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613"],
input: 0.06, input: 0.06,

View File

@ -90,6 +90,8 @@ const (
GPT432k = "gpt-4-32k" GPT432k = "gpt-4-32k"
GPT432k0314 = "gpt-4-32k-0314" GPT432k0314 = "gpt-4-32k-0314"
GPT432k0613 = "gpt-4-32k-0613" GPT432k0613 = "gpt-4-32k-0613"
GPT4O = "gpt-4o"
GPT4O20240513 = "gpt-4o-2024-05-13"
Dalle = "dalle" Dalle = "dalle"
Dalle2 = "dall-e-2" Dalle2 = "dall-e-2"
Dalle3 = "dall-e-3" Dalle3 = "dall-e-3"
@ -138,7 +140,7 @@ var OpenAIDalleModels = []string{
} }
var VisionModels = []string{ var VisionModels = []string{
GPT4VisionPreview, GPT41106VisionPreview, // openai GPT4VisionPreview, GPT41106VisionPreview, GPT4O, GPT4O20240513, // openai
GeminiProVision, // gemini GeminiProVision, // gemini
Claude3, // anthropic Claude3, // anthropic
ZhiPuChatGLM4Vision, // chatglm ZhiPuChatGLM4Vision, // chatglm