diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 1ee6ae6cd..5102a6722 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -79,13 +79,13 @@ func GetAiSafetyTaskByJob(job *models.Cloudbrain) { } func GetAiSafetyTaskTmpl(ctx *context.Context) { - ctx.Data["id"] = ctx.Params(":jobid") + ctx.Data["id"] = ctx.Params(":id") ctx.Data["PageIsCloudBrain"] = true ctx.HTML(200, tplModelSafetyTestShow) } func GetAiSafetyTask(ctx *context.Context) { - var ID = ctx.Params(":jobid") + var ID = ctx.Params(":id") job, err := models.GetCloudbrainByIDWithDeleted(ID) if err != nil { log.Error("GetCloudbrainByJobID failed:" + err.Error()) @@ -117,7 +117,7 @@ func GetAiSafetyTask(ctx *context.Context) { func StopAiSafetyTask(ctx *context.Context) { log.Info("start to stop the task.") - var ID = ctx.Params(":jobid") + var ID = ctx.Params(":id") task, err := models.GetCloudbrainByIDWithDeleted(ID) result := make(map[string]interface{}) result["result_code"] = "-1" @@ -201,7 +201,7 @@ func StopAiSafetyTask(ctx *context.Context) { } func DelAiSafetyTask(ctx *context.Context) { - var ID = ctx.Params(":jobid") + var ID = ctx.Params(":id") task, err := models.GetCloudbrainByIDWithDeleted(ID) if err != nil { diff --git a/routers/routes/routes.go b/routers/routes/routes.go index 5fbe8cd84..53566ad71 100755 --- a/routers/routes/routes.go +++ b/routers/routes/routes.go @@ -1227,7 +1227,7 @@ func RegisterRoutes(m *macaron.Macaron) { }, context.RepoRef()) m.Group("/modelsafety", func() { - m.Group("/:jobid", func() { + m.Group("/:id", func() { m.Get("/show", reqRepoCloudBrainWriter, repo.GetAiSafetyTaskTmpl) m.Get("", reqRepoCloudBrainWriter, repo.GetAiSafetyTask) m.Post("/stop", cloudbrain.AdminOrOwnerOrJobCreaterRight, repo.StopAiSafetyTask)