diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index 404a76282..310e51019 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -86,6 +86,22 @@ func DeleteModelById(id string) error { } +func ModifyModelDescription(id string, description string) error { + sess := x.NewSession() + defer sess.Close() + + re, err := sess.Update(&AiModelManage{ + ID: id, + Description: description, + }) + if err != nil { + return err + } + log.Info("success to update description from db.re=" + fmt.Sprint((re))) + return nil + +} + func QueryModelByName(name string, uid int64) []*AiModelManage { sess := x.NewSession() defer sess.Close() diff --git a/routers/private/internal.go b/routers/private/internal.go index 8078bedb0..a3ea1c6cd 100755 --- a/routers/private/internal.go +++ b/routers/private/internal.go @@ -47,5 +47,6 @@ func RegisterRoutes(m *macaron.Macaron) { m.Post("/tool/create_model", CreateModel) m.Delete("/tool/delete_model", DeleteModel) m.Get("/tool/show_model", ShowModel) + m.Put("/tool/modify_model", ModifyModel) }, CheckInternalToken) } diff --git a/routers/private/tool.go b/routers/private/tool.go index d751c38a7..bccbc50f3 100755 --- a/routers/private/tool.go +++ b/routers/private/tool.go @@ -76,3 +76,14 @@ func ShowModel(ctx *macaron.Context) { ctx.JSON(500, "query error.") } } + +func ModifyModel(ctx *macaron.Context) { + id := ctx.Query("id") + description := ctx.Query("Description") + err := repo.ModifyModel(id, description) + if err == nil { + ctx.JSON(200, "Success.") + } else { + ctx.JSON(500, "Failed.") + } +} diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 307129911..321b3a023 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -199,3 +199,28 @@ func ShowModelInfo(ctx *context.Context) { ctx.Data["Tasks"] = modelResult ctx.HTML(200, "") } + +func ModifyModel(id string, description string) error { + err := models.ModifyModelDescription(id, description) + if err == nil { + log.Info("modify success.") + } else { + log.Info("Failed to modify.id=" + id + " desc=" + description) + } + return err +} + +func ModifyModelInfo(ctx *context.Context) { + log.Info("delete model start.") + id := ctx.Query("ID") + description := ctx.Query("Description") + + err := ModifyModel(id, description) + + if err == nil { + ctx.HTML(200, "success") + } else { + ctx.HTML(500, "Failed.") + } + +}