feat: support gpt-3.5-turbo-0125 model

coaidev · Feb 2, 2024 · 6670773 · 6670773
1 parent 1f52d56
commit 6670773
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/app/src/admin/channel.ts b/app/src/admin/channel.ts
@@ -48,6 +48,7 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
       "gpt-3.5-turbo-0613",
       "gpt-3.5-turbo-0301",
       "gpt-3.5-turbo-1106",
+      "gpt-3.5-turbo-0125",
       "gpt-3.5-turbo-16k",
       "gpt-3.5-turbo-16k-0613",
       "gpt-3.5-turbo-16k-0301",
@@ -80,6 +81,7 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
       "gpt-3.5-turbo-0613",
       "gpt-3.5-turbo-0301",
       "gpt-3.5-turbo-1106",
+      "gpt-3.5-turbo-0125",
       "gpt-3.5-turbo-16k",
       "gpt-3.5-turbo-16k-0613",
       "gpt-3.5-turbo-16k-0301",

diff --git a/app/src/admin/colors.ts b/app/src/admin/colors.ts
@@ -4,6 +4,7 @@ export const modelColorMapper: Record<string, string> = {
   "gpt-3.5-turbo-0613": "#34bf49",
   "gpt-3.5-turbo-0301": "#34bf49",
   "gpt-3.5-turbo-1106": "#11ba2b",
+  "gpt-3.5-turbo-0125": "#11ba2b",
   dalle: "#e4e5e5",
   "dall-e-2": "#e4e5e5",
   "dall-e-3": "#e4e5e5",

diff --git a/globals/variables.go b/globals/variables.go
@@ -45,6 +45,7 @@ const (
 	GPT3Turbo0613         = "gpt-3.5-turbo-0613"
 	GPT3Turbo0301         = "gpt-3.5-turbo-0301"
 	GPT3Turbo1106         = "gpt-3.5-turbo-1106"
+	GPT3Turbo0125         = "gpt-3.5-turbo-0125"
 	GPT3Turbo16k          = "gpt-3.5-turbo-16k"
 	GPT3Turbo16k0613      = "gpt-3.5-turbo-16k-0613"
 	GPT3Turbo16k0301      = "gpt-3.5-turbo-16k-0301"

diff --git a/utils/tokenizer.go b/utils/tokenizer.go
@@ -16,7 +16,7 @@ func GetWeightByModel(model string) int {
 		globals.Claude1, globals.Claude1100k,
 		globals.Claude2, globals.Claude2100k, globals.Claude2200k:
 		return 2
-	case globals.GPT3Turbo, globals.GPT3Turbo0613, globals.GPT3Turbo1106,
+	case globals.GPT3Turbo, globals.GPT3Turbo0613, globals.GPT3Turbo1106, globals.GPT3Turbo0125,
 		globals.GPT3Turbo16k, globals.GPT3Turbo16k0613,
 		globals.GPT4, globals.GPT40314, globals.GPT40613,
 		globals.GPT41106Preview, globals.GPT4TurboPreview, globals.GPT40125Preview,