mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 21:10:18 +09:00
139 lines
4.1 KiB
Go
139 lines
4.1 KiB
Go
package globals
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
const ChatMaxThread = 5
|
|
const AnonymousMaxThread = 1
|
|
|
|
var AllowedOrigins []string
|
|
|
|
var NotifyUrl = ""
|
|
var ArticlePermissionGroup []string
|
|
var GenerationPermissionGroup []string
|
|
var CacheAcceptedModels []string
|
|
var CacheAcceptedExpire int64
|
|
var CacheAcceptedSize int64
|
|
var AcceptImageStore bool
|
|
|
|
func OriginIsAllowed(uri string) bool {
|
|
if len(AllowedOrigins) == 0 {
|
|
// if allowed origins is empty, allow all origins
|
|
return true
|
|
}
|
|
|
|
instance, _ := url.Parse(uri)
|
|
if instance == nil {
|
|
return false
|
|
}
|
|
|
|
if instance.Hostname() == "localhost" || instance.Scheme == "file" {
|
|
return true
|
|
}
|
|
|
|
if strings.HasPrefix(instance.Host, "www.") {
|
|
instance.Host = instance.Host[4:]
|
|
}
|
|
|
|
return in(instance.Host, AllowedOrigins)
|
|
}
|
|
|
|
func OriginIsOpen(c *gin.Context) bool {
|
|
return strings.HasPrefix(c.Request.URL.Path, "/v1") || strings.HasPrefix(c.Request.URL.Path, "/dashboard")
|
|
}
|
|
|
|
const (
|
|
GPT3Turbo = "gpt-3.5-turbo"
|
|
GPT3TurboInstruct = "gpt-3.5-turbo-instruct"
|
|
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"
|
|
GPT4 = "gpt-4"
|
|
GPT4All = "gpt-4-all"
|
|
GPT4Vision = "gpt-4-v"
|
|
GPT4Dalle = "gpt-4-dalle"
|
|
GPT40314 = "gpt-4-0314"
|
|
GPT40613 = "gpt-4-0613"
|
|
GPT41106Preview = "gpt-4-1106-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"
|
|
GPT432k0314 = "gpt-4-32k-0314"
|
|
GPT432k0613 = "gpt-4-32k-0613"
|
|
Dalle = "dalle"
|
|
Dalle2 = "dall-e-2"
|
|
Dalle3 = "dall-e-3"
|
|
Claude1 = "claude-1"
|
|
Claude1100k = "claude-1.3"
|
|
Claude2 = "claude-1-100k"
|
|
Claude2100k = "claude-2"
|
|
Claude2200k = "claude-2.1"
|
|
Claude3 = "claude-3"
|
|
ClaudeSlack = "claude-slack"
|
|
SparkDesk = "spark-desk-v1.5"
|
|
SparkDeskV2 = "spark-desk-v2"
|
|
SparkDeskV3 = "spark-desk-v3"
|
|
SparkDeskV35 = "spark-desk-v3.5"
|
|
ChatBison001 = "chat-bison-001"
|
|
GeminiPro = "gemini-pro"
|
|
GeminiProVision = "gemini-pro-vision"
|
|
BingCreative = "bing-creative"
|
|
BingBalanced = "bing-balanced"
|
|
BingPrecise = "bing-precise"
|
|
ZhiPuChatGLMTurbo = "zhipu-chatglm-turbo"
|
|
ZhiPuChatGLMPro = "zhipu-chatglm-pro"
|
|
ZhiPuChatGLMStd = "zhipu-chatglm-std"
|
|
ZhiPuChatGLMLite = "zhipu-chatglm-lite"
|
|
QwenTurbo = "qwen-turbo"
|
|
QwenPlus = "qwen-plus"
|
|
QwenTurboNet = "qwen-turbo-net"
|
|
QwenPlusNet = "qwen-plus-net"
|
|
Midjourney = "midjourney"
|
|
MidjourneyFast = "midjourney-fast"
|
|
MidjourneyTurbo = "midjourney-turbo"
|
|
Hunyuan = "hunyuan"
|
|
GPT360V9 = "360-gpt-v9"
|
|
Baichuan53B = "baichuan-53b"
|
|
SkylarkLite = "skylark-lite-public"
|
|
SkylarkPlus = "skylark-plus-public"
|
|
SkylarkPro = "skylark-pro-public"
|
|
SkylarkChat = "skylark-chat"
|
|
)
|
|
|
|
var OpenAIDalleModels = []string{
|
|
Dalle, Dalle2, Dalle3,
|
|
}
|
|
|
|
var VisionModels = []string{
|
|
GPT4VisionPreview, GPT41106VisionPreview, // openai
|
|
GeminiProVision, // gemini
|
|
Claude3, // anthropic
|
|
}
|
|
|
|
func in(value string, slice []string) bool {
|
|
for _, item := range slice {
|
|
if item == value || strings.Contains(value, item) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func IsOpenAIDalleModel(model string) bool {
|
|
// using image generation api if model is in dalle models
|
|
return in(model, OpenAIDalleModels) && !strings.Contains(model, "gpt-4-dalle")
|
|
}
|
|
|
|
func IsVisionModel(model string) bool {
|
|
return in(model, VisionModels)
|
|
}
|