From 55d88691fc92e01b50d9295662dbfa14464fa55f Mon Sep 17 00:00:00 2001 From: zouap Date: Tue, 8 Nov 2022 15:14:36 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=88=A0=E9=99=A4=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E6=96=87=E4=BB=B6=E6=8E=A5=E5=8F=A3=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/ai_model_manage.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 7ef9045f5..a71203b40 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -447,13 +447,14 @@ func DeleteModelFile(ctx *context.Context) { err := storage.Attachments.DeleteDir(model.Path[len(bucketName)+1:] + fileName) if err != nil { log.Info("Failed to delete model. id=" + id) - } else { re := map[string]string{ "code": "-1", } re["msg"] = err.Error() ctx.JSON(200, re) return + } else { + UpdateModelSize(id) } } } 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) if err != nil { log.Info("Failed to delete model. id=" + id) - } else { re := map[string]string{ "code": "-1", } re["msg"] = err.Error() ctx.JSON(200, re) return + } else { + UpdateModelSize(id) } } }