diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index a982e565a..711ffd53d 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -52,6 +52,7 @@ func SaveModelToDb(model *AiModelManage) error { re, err := sess.Insert(model) if err != nil { + log.Info("insert error." + err.Error()) return err } log.Info("success to save db.re=" + fmt.Sprint((re))) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 4f07996fe..a5a1a7d9c 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -89,8 +89,10 @@ func saveModelByParameters(jobId string, versionName string, name string, versio Accuracy: string(accuracyJson), } - models.SaveModelToDb(model) - + err = models.SaveModelToDb(model) + if err != nil { + return err + } if len(lastNewModelId) > 0 { //udpate status and version count models.ModifyModelNewProperty(lastNewModelId, MODEL_NOT_LATEST, 0) @@ -110,6 +112,16 @@ func SaveModel(ctx *context.Context) { label := ctx.Query("Label") description := ctx.Query("Description") + if JobId == "" || VersionName == "" { + ctx.Error(500, fmt.Sprintf("JobId or VersionName is null.")) + return + } + + if name == "" || version == "" { + ctx.Error(500, fmt.Sprintf("name or version is null.")) + return + } + err := saveModelByParameters(JobId, VersionName, name, version, label, description, ctx.User.ID) if err != nil {