feat(prompt): 重构提示词服务并添加模型类型子分类

consts/public/public.go

@@ -2,18 +2,48 @@ package public
 
 // ModelType 模型类型常量
 const (
-	ModelTypeInference = 1 // 推理模型
-	ModelTypeImage     = 2 // 图片模型
-	ModelTypeAudio     = 3 // 音频模型
-	ModelTypeVector    = 4 // 向量化模型
-	ModelTypeOmni      = 5 // 全模态模型
+	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:     "图片模型",
-	ModelTypeAudio:     "音频模型",
-	ModelTypeVector:    "向量化模型",
-	ModelTypeOmni:      "全模态模型",
+
+	ModelTypeImage:             "图片模型",
+	ImageSubTypeTextToImage:    "图片模型-文生图",
+	ImageSubTypeImageToImage:   "图片模型-图生图",
+	ImageSubTypeImageEdit:      "图片模型-图片编辑",
+	ImageSubTypeImageVariation: "图片模型-图片变体",
+
+	ModelTypeAudio:             "音频模型",
+	AudioSubTypeTextToSpeech:   "音频模型-文生音",
+	AudioSubTypeSpeechToText:   "音频模型-音生文",
+	AudioSubTypeSpeechToSpeech: "音频模型-音生音",
+
+	ModelTypeVector:        "向量化模型",
+	VectorSubTypeEmbedding: "向量化模型-文本嵌入",
+	VectorSubTypeRerank:    "向量化模型-重排序",
+
+	ModelTypeOmni:             "全模态模型",
+	OmniSubTypeTextImageAudio: "全模态模型-文图音",
+	OmniSubTypeVision:         "全模态模型-视觉理解",
 }