Browse Source

增加删除模型文件接口。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.11.2^2
zouap 3 years ago
parent
commit
55d88691fc
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      routers/repo/ai_model_manage.go

+ 4
- 2
routers/repo/ai_model_manage.go View File

@@ -447,13 +447,14 @@ func DeleteModelFile(ctx *context.Context) {
err := storage.Attachments.DeleteDir(model.Path[len(bucketName)+1:] + fileName) err := storage.Attachments.DeleteDir(model.Path[len(bucketName)+1:] + fileName)
if err != nil { if err != nil {
log.Info("Failed to delete model. id=" + id) log.Info("Failed to delete model. id=" + id)
} else {
re := map[string]string{ re := map[string]string{
"code": "-1", "code": "-1",
} }
re["msg"] = err.Error() re["msg"] = err.Error()
ctx.JSON(200, re) ctx.JSON(200, re)
return return
} else {
UpdateModelSize(id)
} }
} }
} else if model.Type == models.TypeCloudBrainTwo { } else if model.Type == models.TypeCloudBrainTwo {
@@ -462,13 +463,14 @@ func DeleteModelFile(ctx *context.Context) {
err := storage.ObsRemoveObject(setting.Bucket, model.Path[len(setting.Bucket)+1:]+fileName) err := storage.ObsRemoveObject(setting.Bucket, model.Path[len(setting.Bucket)+1:]+fileName)
if err != nil { if err != nil {
log.Info("Failed to delete model. id=" + id) log.Info("Failed to delete model. id=" + id)
} else {
re := map[string]string{ re := map[string]string{
"code": "-1", "code": "-1",
} }
re["msg"] = err.Error() re["msg"] = err.Error()
ctx.JSON(200, re) ctx.JSON(200, re)
return return
} else {
UpdateModelSize(id)
} }
} }
} }


Loading…
Cancel
Save