104 lines
3.5 KiB
Go
104 lines
3.5 KiB
Go
package public
|
|
|
|
// ModelType 模型类型常量
|
|
const (
|
|
ModelTypeInference = 100 // 推理模型
|
|
|
|
ModelTypeImage = 200 // 图片模型
|
|
ImageSubTypeTextToImage = 201 // 图片模型-文生图
|
|
ImageSubTypeImageToImage = 202 // 图片模型-图生图
|
|
ImageSubTypeImageEdit = 203 // 图片模型-图片编辑
|
|
ImageSubTypeImageVariation = 204 // 图片模型-图片变体
|
|
|
|
ModelTypeAudio = 300 // 音频模型
|
|
AudioSubTypeTextToSpeech = 301 // 音频模型-文生音
|
|
AudioSubTypeSpeechToText = 302 // 音频模型-音生文
|
|
AudioSubTypeSpeechToSpeech = 303 // 音频模型-音生音
|
|
|
|
ModelTypeVector = 400 // 向量化模型
|
|
VectorSubTypeEmbedding = 401 // 向量化模型-文本嵌入
|
|
VectorSubTypeRerank = 402 // 向量化模型-重排序
|
|
|
|
ModelTypeOmni = 500 // 全模态模型
|
|
OmniSubTypeTextImageAudio = 501 // 全模态模型-文图音
|
|
OmniSubTypeVision = 502 // 全模态模型-视觉理解
|
|
|
|
ModelTypeVideo = 600 // 视频模型
|
|
VideoSubTypeTextToVideo = 601 // 视频模型-文生视频
|
|
VideoSubTypeImageToVideo = 602 // 视频模型-图生视频
|
|
VideoSubTypeImageTextToVideo = 603 // 视频模型-图文生视频
|
|
VideoSubTypeVideoToVideo = 604 // 视频模型-视频生视频
|
|
VideoSubTypeVideoEdit = 605 // 视频模型-视频编辑
|
|
)
|
|
|
|
// ModelTypeName 模型类型名称映射
|
|
var ModelTypeName = map[int]string{
|
|
ModelTypeInference: "推理模型",
|
|
|
|
ModelTypeImage: "图片模型",
|
|
ImageSubTypeTextToImage: "图片模型-文生图",
|
|
ImageSubTypeImageToImage: "图片模型-图生图",
|
|
ImageSubTypeImageEdit: "图片模型-图片编辑",
|
|
ImageSubTypeImageVariation: "图片模型-图片变体",
|
|
|
|
ModelTypeAudio: "音频模型",
|
|
AudioSubTypeTextToSpeech: "音频模型-文生音",
|
|
AudioSubTypeSpeechToText: "音频模型-音生文",
|
|
AudioSubTypeSpeechToSpeech: "音频模型-音生音",
|
|
|
|
ModelTypeVector: "向量化模型",
|
|
VectorSubTypeEmbedding: "向量化模型-文本嵌入",
|
|
VectorSubTypeRerank: "向量化模型-重排序",
|
|
|
|
ModelTypeOmni: "全模态模型",
|
|
OmniSubTypeTextImageAudio: "全模态模型-文图音",
|
|
OmniSubTypeVision: "全模态模型-视觉理解",
|
|
|
|
ModelTypeVideo: "视频模型",
|
|
VideoSubTypeTextToVideo: "视频模型-文生视频",
|
|
VideoSubTypeImageToVideo: "视频模型-图生视频",
|
|
VideoSubTypeImageTextToVideo: "视频模型-图文生视频",
|
|
VideoSubTypeVideoToVideo: "视频模型-视频生视频",
|
|
VideoSubTypeVideoEdit: "视频模型-视频编辑",
|
|
}
|
|
|
|
// 运营商常量
|
|
const (
|
|
OperatorAliyun = "阿里云百炼"
|
|
OperatorVolcengine = "火山引擎"
|
|
OperatorTencent = "腾讯云"
|
|
OperatorHuawei = "华为云"
|
|
OperatorBaidu = "百度智能云"
|
|
OperatorOpenAI = "OpenAI"
|
|
OperatorAzure = "Azure OpenAI"
|
|
OperatorAWS = "AWS Bedrock"
|
|
OperatorGoogle = "Google Cloud"
|
|
OperatorDeepSeek = "DeepSeek"
|
|
OperatorMoonshot = "Moonshot"
|
|
OperatorZhipu = "智谱AI"
|
|
OperatorBaichuan = "百川智能"
|
|
OperatorMinimax = "MiniMax"
|
|
OperatorXunfei = "科大讯飞"
|
|
OperatorOthers = "其他"
|
|
)
|
|
|
|
// OperatorList 运营商列表（供前端下拉框使用）
|
|
var OperatorList = []string{
|
|
OperatorAliyun,
|
|
OperatorVolcengine,
|
|
OperatorTencent,
|
|
OperatorHuawei,
|
|
OperatorBaidu,
|
|
OperatorOpenAI,
|
|
OperatorAzure,
|
|
OperatorAWS,
|
|
OperatorGoogle,
|
|
OperatorDeepSeek,
|
|
OperatorMoonshot,
|
|
OperatorZhipu,
|
|
OperatorBaichuan,
|
|
OperatorMinimax,
|
|
OperatorXunfei,
|
|
OperatorOthers,
|
|
}
|