Browse Source

Merge branch 'inference-job' of https://git.openi.org.cn/OpenI/aiforge into inference-job

pull/1406/head
zhoupzh 3 years ago
parent
commit
7749757f34
2 changed files with 14 additions and 9 deletions
  1. +12
    -0
      routers/repo/modelarts.go
  2. +2
    -9
      routers/repo/setting.go

+ 12
- 0
routers/repo/modelarts.go View File

@@ -2102,3 +2102,15 @@ func DownloadMultiResultFile(ctx *context.Context) {
ctx.ServerError("no file to download.", err)
}
}

func SetJobCount(ctx *context.Context) {
repoId := ctx.Repo.Repository.ID
_, jobCount, err := models.Cloudbrains(&models.CloudbrainsOptions{
RepoID: repoId,
})
if err != nil {
ctx.ServerError("Get job faild:", err)
return
}
ctx.Data["jobCount"] = jobCount
}

+ 2
- 9
routers/repo/setting.go View File

@@ -51,6 +51,7 @@ func Settings(ctx *context.Context) {
ctx.Data["PageIsSettingsOptions"] = true
ctx.Data["ForcePrivate"] = setting.Repository.ForcePrivate
SetModelCount(ctx)
SetJobCount(ctx)
ctx.HTML(200, tplSettingsOptions)
}

@@ -59,17 +60,9 @@ func SettingsPost(ctx *context.Context, form auth.RepoSettingForm) {
ctx.Data["Title"] = ctx.Tr("repo.settings")
ctx.Data["PageIsSettingsOptions"] = true
SetModelCount(ctx)
SetJobCount(ctx)
repo := ctx.Repo.Repository

_, jobCount, err := models.Cloudbrains(&models.CloudbrainsOptions{
RepoID: repo.ID,
})
if err != nil {
ctx.ServerError("Get job faild:", err)
return
}
ctx.Data["jobCount"] = jobCount

switch ctx.Query("action") {
case "update":
if ctx.HasError() {


Loading…
Cancel
Save