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

This commit is contained in:
2026-05-22 09:49:46 +08:00
parent b6cdb8ff1d
commit 4946220185
3 changed files with 43 additions and 9 deletions

View File

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

View File

@@ -51,6 +51,7 @@ func UploadByTask(ctx context.Context, _ *entity.AsynchTask, data []byte, fileEx
headers := util.ForwardHeaders(ctx) headers := util.ForwardHeaders(ctx)
headers["Content-Type"] = contentType headers["Content-Type"] = contentType
//fullURL := "oss/file/uploadFile"
fullURL := "oss/file/uploadFile" fullURL := "oss/file/uploadFile"
g.Log().Infof(ctx, "[OSS] upload start url=%s filename=%s size=%d", fullURL, filename, len(data)) g.Log().Infof(ctx, "[OSS] upload start url=%s filename=%s size=%d", fullURL, filename, len(data))

View File

@@ -231,6 +231,7 @@ func (s *modelService) Get(ctx context.Context, req *dto.GetModelReq) (*dto.GetM
model.RequestMapping = util.ParseJSONField(model.RequestMapping) model.RequestMapping = util.ParseJSONField(model.RequestMapping)
model.ResponseMapping = util.ParseJSONField(model.ResponseMapping) model.ResponseMapping = util.ParseJSONField(model.ResponseMapping)
model.ResponseBody = util.ParseJSONField(model.ResponseBody) model.ResponseBody = util.ParseJSONField(model.ResponseBody)
model.TokenConfig = util.ParseJSONField(model.TokenConfig)
return &dto.GetModelRes{ return &dto.GetModelRes{
Model: model, Model: model,
}, nil }, nil
@@ -266,6 +267,7 @@ func (s *modelService) List(ctx context.Context, req *dto.ListModelReq) (res *dt
m.RequestMapping = util.ParseJSONField(m.RequestMapping) m.RequestMapping = util.ParseJSONField(m.RequestMapping)
m.ResponseMapping = util.ParseJSONField(m.ResponseMapping) m.ResponseMapping = util.ParseJSONField(m.ResponseMapping)
m.ResponseBody = util.ParseJSONField(m.ResponseBody) m.ResponseBody = util.ParseJSONField(m.ResponseBody)
m.TokenConfig = util.ParseJSONField(m.TokenConfig)
} }
return &dto.ListModelRes{ return &dto.ListModelRes{
List: models, List: models,
@@ -345,6 +347,7 @@ func (s *modelService) GetIsChatModel(ctx context.Context) (*dto.GetIsChatModelR
model.RequestMapping = util.ParseJSONField(model.RequestMapping) model.RequestMapping = util.ParseJSONField(model.RequestMapping)
model.ResponseMapping = util.ParseJSONField(model.ResponseMapping) model.ResponseMapping = util.ParseJSONField(model.ResponseMapping)
model.ResponseBody = util.ParseJSONField(model.ResponseBody) model.ResponseBody = util.ParseJSONField(model.ResponseBody)
model.TokenConfig = util.ParseJSONField(model.TokenConfig)
return &dto.GetIsChatModelRes{ return &dto.GetIsChatModelRes{
Model: model, Model: model,
}, nil }, nil