|
|
@@ -8,6 +8,7 @@ package repo |
|
|
|
import ( |
|
|
|
"code.gitea.io/gitea/models" |
|
|
|
"code.gitea.io/gitea/modules/context" |
|
|
|
"code.gitea.io/gitea/modules/log" |
|
|
|
"code.gitea.io/gitea/modules/modelarts" |
|
|
|
"net/http" |
|
|
|
) |
|
|
@@ -19,7 +20,7 @@ func GetModelArtsTask(ctx *context.APIContext) { |
|
|
|
|
|
|
|
jobID := ctx.Params(":jobid") |
|
|
|
repoID := ctx.Repo.Repository.ID |
|
|
|
_, err = models.GetRepoCloudBrainByJobID(repoID, jobID) |
|
|
|
job, err := models.GetRepoCloudBrainByJobID(repoID, jobID) |
|
|
|
if err != nil { |
|
|
|
ctx.NotFound(err) |
|
|
|
return |
|
|
@@ -30,6 +31,12 @@ func GetModelArtsTask(ctx *context.APIContext) { |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
job.Status = result.Status |
|
|
|
err = models.UpdateJob(job) |
|
|
|
if err != nil { |
|
|
|
log.Error("UpdateJob failed:", err) |
|
|
|
} |
|
|
|
|
|
|
|
ctx.JSON(http.StatusOK, map[string]interface{}{ |
|
|
|
"JobID": jobID, |
|
|
|
"JobStatus": result.Status, |
|
|
|