Browse Source

日志及下载日志接口统一。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.10.1^2
zouap 3 years ago
parent
commit
0eebea40be
2 changed files with 13 additions and 4 deletions
  1. +4
    -4
      routers/api/v1/api.go
  2. +9
    -0
      routers/api/v1/repo/cloudbrain.go

+ 4
- 4
routers/api/v1/api.go View File

@@ -952,10 +952,10 @@ func RegisterRoutes(m *macaron.Macaron) {
m.Get("/:id/modelartlog", repo.TrainJobForModelConvertGetLog)
m.Get("/:id/model_list", repo.CloudBrainModelConvertList)
}, reqRepoReader(models.UnitTypeModelManage))
m.Group("/modelsafety", func() {
m.Get("/:id/log", repo.ModelSafetyGetLog)
m.Get("/:id/down_log_file", repo.ModelSafetyDownloadLogFile)
}, reqRepoReader(models.UnitTypeModelManage))
// m.Group("/modelsafety", func() {
// m.Get("/:id/log", repo.ModelSafetyGetLog)
// m.Get("/:id/down_log_file", repo.ModelSafetyDownloadLogFile)
// }, reqRepoReader(models.UnitTypeModelManage))
m.Group("/modelarts", func() {
m.Group("/notebook", func() {
//m.Get("/:jobid", repo.GetModelArtsNotebook)


+ 9
- 0
routers/api/v1/repo/cloudbrain.go View File

@@ -494,6 +494,10 @@ func CloudbrainDownloadLogFile(ctx *context.Context) {
ctx.ServerError(err.Error(), err)
return
}
if job.JobType == string(models.JobTypeModelSafety) {
ModelSafetyDownloadLogFile(ctx)
return
}

logDir := "/model"
if job.JobType == string(models.JobTypeInference) || job.JobType == string(models.JobTypeModelSafety) {
@@ -535,6 +539,11 @@ func CloudbrainGetLog(ctx *context.APIContext) {
ctx.ServerError(err.Error(), err)
return
}
if job.JobType == string(models.JobTypeModelSafety) {
ModelSafetyGetLog(ctx)
return
}

lines := ctx.QueryInt("lines")
baseLine := ctx.Query("base_line")
order := ctx.Query("order")


Loading…
Cancel
Save