40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package consts
|
|
|
|
type AutoModeDefaultModel string
|
|
|
|
const (
|
|
AutoModeDefaultChatModel AutoModeDefaultModel = "deepseek-chat"
|
|
AutoModeDefaultEmbeddingModel AutoModeDefaultModel = "bge-m3"
|
|
AutoModeDefaultRerankModel AutoModeDefaultModel = "bge-reranker-v2-m3"
|
|
AutoModeDefaultAnalysisModel AutoModeDefaultModel = "qwen2.5-3b-instruct"
|
|
AutoModeDefaultAnalysisVLModel AutoModeDefaultModel = "qwen-vl-max-latest"
|
|
)
|
|
|
|
func GetAutoModeDefaultModel(modelType string) string {
|
|
switch modelType {
|
|
case "chat":
|
|
return string(AutoModeDefaultChatModel)
|
|
case "embedding":
|
|
return string(AutoModeDefaultEmbeddingModel)
|
|
case "rerank":
|
|
return string(AutoModeDefaultRerankModel)
|
|
case "analysis":
|
|
return string(AutoModeDefaultAnalysisModel)
|
|
case "analysis-vl":
|
|
return string(AutoModeDefaultAnalysisVLModel)
|
|
default:
|
|
return string(AutoModeDefaultChatModel)
|
|
}
|
|
}
|
|
|
|
type ModelSettingMode string
|
|
|
|
const (
|
|
ModelSettingModeManual ModelSettingMode = "manual"
|
|
ModelSettingModeAuto ModelSettingMode = "auto"
|
|
)
|
|
|
|
const (
|
|
AutoModeBaseURL = "https://model-square.app.baizhi.cloud/v1"
|
|
)
|