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 // 全模态模型-视觉理解
)

// ModelTypeName 模型类型名称映射
var ModelTypeName = map[int]string{
	ModelTypeInference: "推理模型",

	ModelTypeImage:             "图片模型",
	ImageSubTypeTextToImage:    "图片模型-文生图",
	ImageSubTypeImageToImage:   "图片模型-图生图",
	ImageSubTypeImageEdit:      "图片模型-图片编辑",
	ImageSubTypeImageVariation: "图片模型-图片变体",

	ModelTypeAudio:             "音频模型",
	AudioSubTypeTextToSpeech:   "音频模型-文生音",
	AudioSubTypeSpeechToText:   "音频模型-音生文",
	AudioSubTypeSpeechToSpeech: "音频模型-音生音",

	ModelTypeVector:        "向量化模型",
	VectorSubTypeEmbedding: "向量化模型-文本嵌入",
	VectorSubTypeRerank:    "向量化模型-重排序",

	ModelTypeOmni:             "全模态模型",
	OmniSubTypeTextImageAudio: "全模态模型-文图音",
	OmniSubTypeVision:         "全模态模型-视觉理解",
}