From 149902bd8a5ef6f6a91861974137c172ecb042ae Mon Sep 17 00:00:00 2001 From: wozulong <> Date: Wed, 10 Apr 2024 11:21:12 +0800 Subject: [PATCH] add gpt-4-turbo Signed-off-by: wozulong <> --- common/model-ratio.go | 4 +++- relay/channel/openai/constant.go | 2 ++ web/src/helpers/render.js | 2 ++ 3 files changed, 7 insertions(+), 1 deletion(-) diff --git a/common/model-ratio.go b/common/model-ratio.go index dba84fc..6df0311 100644 --- a/common/model-ratio.go +++ b/common/model-ratio.go @@ -21,6 +21,8 @@ var DefaultModelRatio = map[string]float64{ "gpt-4-32k": 30, "gpt-4-32k-0314": 30, "gpt-4-32k-0613": 30, + "gpt-4-turbo": 5, // $0.01 / 1K tokens + "gpt-4-turbo-2024-04-09": 5, // $0.01 / 1K tokens "gpt-4-1106-preview": 5, // $0.01 / 1K tokens "gpt-4-0125-preview": 5, // $0.01 / 1K tokens "gpt-4-turbo-preview": 5, // $0.01 / 1K tokens @@ -203,7 +205,7 @@ func GetCompletionRatio(name string) float64 { return 1.333333 } if strings.HasPrefix(name, "gpt-4") { - if strings.HasSuffix(name, "preview") { + if strings.HasSuffix(name, "preview") || strings.HasPrefix(name, "gpt-4-turbo") { return 3 } return 2 diff --git a/relay/channel/openai/constant.go b/relay/channel/openai/constant.go index 91f4e51..b80410c 100644 --- a/relay/channel/openai/constant.go +++ b/relay/channel/openai/constant.go @@ -6,6 +6,8 @@ var ModelList = []string{ "gpt-3.5-turbo-instruct", "gpt-4", "gpt-4-0314", "gpt-4-0613", "gpt-4-1106-preview", "gpt-4-0125-preview", "gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613", + "gpt-4-turbo", + "gpt-4-turbo-2024-04-09", "gpt-4-turbo-preview", "gpt-4-vision-preview", "text-embedding-ada-002", "text-embedding-3-small", "text-embedding-3-large", diff --git a/web/src/helpers/render.js b/web/src/helpers/render.js index b76b6c8..7183e57 100644 --- a/web/src/helpers/render.js +++ b/web/src/helpers/render.js @@ -180,6 +180,8 @@ export const modelColorMap = { 'gpt-4-1106-preview': 'rgb(30,144,255)', // 道奇蓝 'gpt-4-0125-preview': 'rgb(2,177,236)', // 深天蓝 'gpt-4-turbo-preview': 'rgb(2,177,255)', // 深天蓝 + 'gpt-4-turbo': 'rgb(2,190,255)', // 深天蓝 + 'gpt-4-turbo-2024-04-09': 'rgb(2,200,255)', // 深天蓝 'gpt-4-32k': 'rgb(104,111,238)', // 中紫色 'gpt-4-32k-0314': 'rgb(90,105,205)', // 暗灰蓝色 'gpt-4-32k-0613': 'rgb(61,71,139)', // 暗蓝灰色