77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package common
|
|
|
|
import "strings"
|
|
|
|
var (
|
|
// OpenAIResponseOnlyModels is a list of models that are only available for OpenAI responses.
|
|
OpenAIResponseOnlyModels = []string{
|
|
"o3-pro",
|
|
"o3-deep-research",
|
|
"o4-mini-deep-research",
|
|
}
|
|
ImageGenerationModels = []string{
|
|
"dall-e-3",
|
|
"dall-e-2",
|
|
"gpt-image-1",
|
|
"prefix:imagen-",
|
|
"flux-",
|
|
"flux.1-",
|
|
"stable-diffusion",
|
|
"sd3",
|
|
"sdxl",
|
|
"midjourney",
|
|
"mj-",
|
|
"cogview",
|
|
"wanx",
|
|
"kolors",
|
|
"minimax-image",
|
|
"jimeng",
|
|
"seedream",
|
|
"ideogram",
|
|
"playground-v",
|
|
"recraft",
|
|
"stability.",
|
|
"image-",
|
|
"sd-",
|
|
}
|
|
OpenAITextModels = []string{
|
|
"gpt-",
|
|
"o1",
|
|
"o3",
|
|
"o4",
|
|
"chatgpt",
|
|
}
|
|
)
|
|
|
|
func IsOpenAIResponseOnlyModel(modelName string) bool {
|
|
for _, m := range OpenAIResponseOnlyModels {
|
|
if strings.Contains(modelName, m) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func IsImageGenerationModel(modelName string) bool {
|
|
modelName = strings.ToLower(modelName)
|
|
for _, m := range ImageGenerationModels {
|
|
if strings.Contains(modelName, m) {
|
|
return true
|
|
}
|
|
if strings.HasPrefix(m, "prefix:") && strings.HasPrefix(modelName, strings.TrimPrefix(m, "prefix:")) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func IsOpenAITextModel(modelName string) bool {
|
|
modelName = strings.ToLower(modelName)
|
|
for _, m := range OpenAITextModels {
|
|
if strings.Contains(modelName, m) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|