GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-19 21:10:18 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: variables and VisionSkipModels

Browse Source
This commit is contained in:
XiaomaiTX 2024-05-21 00:56:14 +08:00
parent e60326292a
commit 135def8bb0
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
globals/variables.go
View File

@ -144,12 +144,16 @@ var OpenAIDalleModels = []string{
}
var VisionModels = []string{
GPT4VisionPreview, GPT41106VisionPreview, GPT4O, GPT4O20240513, // openai
GeminiProVision, // gemini
GPT4VisionPreview, GPT41106VisionPreview, GPT4Turbo, GPT4Turbo20240409, GPT4O, GPT4O20240513, // openai
GeminiProVision, Gemini15ProLatest, Gemini15FlashLatest, // gemini
Claude3, // anthropic
ZhiPuChatGLM4Vision, // chatglm
}
var VisionSkipModels = []string{
GPT4TurboPreview,
}
func in(value string, slice []string) bool {
for _, item := range slice {
if item == value || strings.Contains(value, item) {
@ -165,5 +169,6 @@ func IsOpenAIDalleModel(model string) bool {
}
func IsVisionModel(model string) bool {
return in(model, VisionModels)
return in(model, VisionModels) && !in(model, VisionSkipModels)
}