From 9dcd20584c163d08ef957ef719c8128b262bbbf5 Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 12 Dec 2022 10:14:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=A7=81=E6=9C=89=E5=B1=9E?= =?UTF-8?q?=E6=80=A7=E7=9A=84=E4=BF=AE=E6=94=B9=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- models/ai_model_manage.go | 5 +++-- routers/repo/ai_model_manage.go | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index f6836781c..4fe74e555 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -306,15 +306,16 @@ func ModifyModelPrivate(id string, isPrivate bool) error { return nil } -func ModifyLocalModel(id string, name, label, description string, engine int) error { +func ModifyLocalModel(id string, name, label, description string, engine int, isPrivate bool) error { var sess *xorm.Session sess = x.ID(id) defer sess.Close() - re, err := sess.Cols("name", "label", "description", "engine").Update(&AiModelManage{ + re, err := sess.Cols("name", "label", "description", "engine", "is_private").Update(&AiModelManage{ Description: description, Name: name, Label: label, Engine: int64(engine), + IsPrivate: isPrivate, }) if err != nil { return err diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 7bd286602..4884a815f 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -1182,6 +1182,7 @@ func ModifyModelInfo(ctx *context.Context) { label := ctx.Query("label") description := ctx.Query("description") engine := ctx.QueryInt("engine") + isPrivate := ctx.QueryBool("isPrivate") aimodels := models.QueryModelByName(name, task.RepoId) if aimodels != nil && len(aimodels) > 0 { if len(aimodels) == 1 { @@ -1196,14 +1197,14 @@ func ModifyModelInfo(ctx *context.Context) { return } } - err = models.ModifyLocalModel(id, name, label, description, engine) + err = models.ModifyLocalModel(id, name, label, description, engine, isPrivate) } else { label := ctx.Query("label") description := ctx.Query("description") engine := task.Engine name := task.Name - err = models.ModifyLocalModel(id, name, label, description, int(engine)) + err = models.ModifyLocalModel(id, name, label, description, int(engine), task.IsPrivate) } if err != nil {