diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 58baf1a0c..091ae022f 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -95,6 +95,9 @@ func GetAiSafetyTask(ctx *context.Context) { job, err = models.GetCloudbrainByIDWithDeleted(ID) job.BenchmarkType = "安全评测" job.BenchmarkTypeName = "Image Classification" + job.CanModify = cloudbrain.CanModifyJob(ctx, job) + job.CanDel = cloudbrain.CanDeleteJob(ctx, job) + ctx.JSON(200, job) }