GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-06-07 14:20:20 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: support gpt-4-0125-preview, gpt-4-turbo-preview, gpt-4-1106-vision-preview models

Browse Source
This commit is contained in:
Zhang Minghan 2024-01-30 23:32:27 +08:00
parent 31e1eaa684
commit 59e822264b
4 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/src/admin/channel.ts
View File

@ -55,7 +55,10 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
"gpt-4-0314", "gpt-4-0314",
"gpt-4-0613", "gpt-4-0613",
"gpt-4-1106-preview", "gpt-4-1106-preview",
"gpt-4-0125-preview",
"gpt-4-turbo-preview",
"gpt-4-vision-preview", "gpt-4-vision-preview",
"gpt-4-1106-vision-preview",
"gpt-4-32k", "gpt-4-32k",
"gpt-4-32k-0314", "gpt-4-32k-0314",
"gpt-4-32k-0613", "gpt-4-32k-0613",
@ -84,7 +87,10 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
"gpt-4-0314", "gpt-4-0314",
"gpt-4-0613", "gpt-4-0613",
"gpt-4-1106-preview", "gpt-4-1106-preview",
"gpt-4-0125-preview",
"gpt-4-turbo-preview",
"gpt-4-vision-preview", "gpt-4-vision-preview",
"gpt-4-1106-vision-preview",
"gpt-4-32k", "gpt-4-32k",
"gpt-4-32k-0314", "gpt-4-32k-0314",
"gpt-4-32k-0613", "gpt-4-32k-0613",

3
app/src/admin/colors.ts
View File

@ -19,6 +19,9 @@ export const modelColorMapper: Record<string, string> = {
"gpt-4": "#8e43e7", "gpt-4": "#8e43e7",
"gpt-4-1106-preview": "#8e43e7", "gpt-4-1106-preview": "#8e43e7",
"gpt-4-0125-preview": "#8e43e7",
"gpt-4-turbo-preview": "#8e43e7",
"gpt-4-1106-vision-preview": "#8e43e7",
"gpt-4-vision-preview": "#8e43e7", "gpt-4-vision-preview": "#8e43e7",
"gpt-4-0613": "#8e43e7", "gpt-4-0613": "#8e43e7",
"gpt-4-0314": "#8e43e7", "gpt-4-0314": "#8e43e7",

12
globals/variables.go
View File

@ -1,6 +1,7 @@
package globals package globals
import ( import (
"chat/utils"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"net/url" "net/url"
"strings" "strings"
@ -55,7 +56,10 @@ const (
GPT40314 = "gpt-4-0314" GPT40314 = "gpt-4-0314"
GPT40613 = "gpt-4-0613" GPT40613 = "gpt-4-0613"
GPT41106Preview = "gpt-4-1106-preview" GPT41106Preview = "gpt-4-1106-preview"
GPT41106VisionPreview = "gpt-4-vision-preview" GPT40125Preview = "gpt-4-0125-preview"
GPT4TurboPreview = "gpt-4-turbo-preview"
GPT4VisionPreview = "gpt-4-vision-preview"
GPT41106VisionPreview = "gpt-4-1106-vision-preview"
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"
@ -117,6 +121,8 @@ func IsDalleModel(model string) bool {
func IsGPT41106VisionPreview(model string) bool { func IsGPT41106VisionPreview(model string) bool {
// enable openai image format for gpt-4-vision-preview model // enable openai image format for gpt-4-vision-preview model
return model == GPT41106VisionPreview || return utils.Any(
strings.Contains(model, GPT41106VisionPreview) model == GPT41106VisionPreview || strings.Contains(model, GPT41106VisionPreview),
model == GPT4VisionPreview || strings.Contains(model, GPT4VisionPreview),
)
} }

4
utils/tokenizer.go
View File

@ -18,7 +18,9 @@ func GetWeightByModel(model string) int {
return 2 return 2
case globals.GPT3Turbo, globals.GPT3Turbo0613, globals.GPT3Turbo1106, case globals.GPT3Turbo, globals.GPT3Turbo0613, globals.GPT3Turbo1106,
globals.GPT3Turbo16k, globals.GPT3Turbo16k0613, globals.GPT3Turbo16k, globals.GPT3Turbo16k0613,
globals.GPT4, globals.GPT40314, globals.GPT40613, globals.GPT41106Preview, globals.GPT41106VisionPreview, globals.GPT4, globals.GPT40314, globals.GPT40613,
globals.GPT41106Preview, globals.GPT4TurboPreview, globals.GPT40125Preview,
globals.GPT4VisionPreview, globals.GPT41106VisionPreview,
globals.GPT432k, globals.GPT432k0613, globals.GPT432k0314: globals.GPT432k, globals.GPT432k0613, globals.GPT432k0314:
return 3 return 3
case globals.GPT3Turbo0301, globals.GPT3Turbo16k0301: case globals.GPT3Turbo0301, globals.GPT3Turbo16k0301: