diff --git a/routers/repo/attachment.go b/routers/repo/attachment.go index b65b6cfce..de1869f1f 100755 --- a/routers/repo/attachment.go +++ b/routers/repo/attachment.go @@ -682,6 +682,7 @@ func GetSuccessChunks(ctx *context.Context) { bucketName := setting.Bucket storage.ObsCopyFile(bucketName, srcObjectName, bucketName, destObjectName) } + UpdateModelSize(modeluuid) ctx.JSON(200, map[string]string{ "uuid": fileChunk.UUID, "uploaded": strconv.Itoa(fileChunk.IsUploaded), @@ -690,6 +691,7 @@ func GetSuccessChunks(ctx *context.Context) { "attachID": strconv.Itoa(int(attachID)), "fileName": attach.Name, }) + return } else { model, err := models.QueryModelById(dbmodeluuid) if err == nil { @@ -705,6 +707,7 @@ func GetSuccessChunks(ctx *context.Context) { "fileName": attach.Name, "modelName": modelname, }) + return } } else { dataset, err := models.GetDatasetByID(attach.DatasetID)