GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-20 05:20:15 +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{ var VisionModels = []string{
GPT4VisionPreview, GPT41106VisionPreview, GPT4O, GPT4O20240513, // openai GPT4VisionPreview, GPT41106VisionPreview, GPT4Turbo, GPT4Turbo20240409, GPT4O, GPT4O20240513, // openai
GeminiProVision, // gemini GeminiProVision, Gemini15ProLatest, Gemini15FlashLatest, // gemini
Claude3, // anthropic Claude3, // anthropic
ZhiPuChatGLM4Vision, // chatglm ZhiPuChatGLM4Vision, // chatglm
} }
var VisionSkipModels = []string{
GPT4TurboPreview,
}
func in(value string, slice []string) bool { func in(value string, slice []string) bool {
for _, item := range slice { for _, item := range slice {
if item == value || strings.Contains(value, item) { if item == value || strings.Contains(value, item) {
@ -165,5 +169,6 @@ func IsOpenAIDalleModel(model string) bool {
} }
func IsVisionModel(model string) bool { func IsVisionModel(model string) bool {
return in(model, VisionModels) return in(model, VisionModels) && !in(model, VisionSkipModels)
} }