根据任务名称查询模型及模型转换任务

Signed-off-by: zouap <zouap@pcl.ac.cn>

models/ai_model_manage.go

@@ -221,10 +221,11 @@ func SaveModelToDb(model *AiModelManage) error {
 	return nil
 }
 
-func QueryModelConvertByName(name string) ([]*AiModelConvert, error) {
+func QueryModelConvertByName(name string, repoId int64) ([]*AiModelConvert, error) {
 	sess := x.NewSession()
 	defer sess.Close()
-	sess.Select("*").Table(new(AiModelConvert)).Where("name='" + name + "'")
+	sess.Select("*").Table(new(AiModelConvert)).
+		Where("name='" + name + "' and repo_id=" + fmt.Sprint(repoId)).OrderBy("created_unix desc")
 	aiModelManageConvertList := make([]*AiModelConvert, 0)
 	err := sess.Find(&aiModelManageConvertList)
 	if err == nil {

routers/repo/ai_model_convert.go

@@ -758,6 +758,11 @@ func GetModelConvertById(ctx *context.Context) (*models.AiModelConvert, error) {
 	return models.QueryModelConvertById(id)
 }
 
+func GetModelConvertByName(ctx *context.Context) ([]*models.AiModelConvert, error) {
+	name := ctx.Query("name")
+	return models.QueryModelConvertByName(name, ctx.Repo.Repository.ID)
+}
+
 func GetModelConvertPageData(ctx *context.Context) ([]*models.AiModelConvert, int64, error) {
 	page := ctx.QueryInt("page")
 	if page <= 0 {