@@ -1559,7 +1559,8 @@ type CreateGrampusJobResponse struct { | |||||
type GetGrampusJobResponse struct { | type GetGrampusJobResponse struct { | ||||
GrampusResult | GrampusResult | ||||
JobInfo GrampusJobInfo `json:"otJob"` | |||||
JobInfo GrampusJobInfo `json:"otJob"` | |||||
ExitDiagnostics string `json:"exitDiagnostics"` | |||||
} | } | ||||
type GrampusNotebookResponse struct { | type GrampusNotebookResponse struct { | ||||
@@ -198,7 +198,6 @@ sendjob: | |||||
SetAuthToken(TOKEN). | SetAuthToken(TOKEN). | ||||
SetResult(&result). | SetResult(&result). | ||||
Get(HOST + urlTrainJob + "/" + jobID) | Get(HOST + urlTrainJob + "/" + jobID) | ||||
if err != nil { | if err != nil { | ||||
return nil, fmt.Errorf("resty GetJob: %v", err) | return nil, fmt.Errorf("resty GetJob: %v", err) | ||||
} | } | ||||
@@ -26,9 +26,9 @@ import ( | |||||
const ( | const ( | ||||
//notebook | //notebook | ||||
storageTypeOBS = "obs" | |||||
autoStopDuration = 4 * 60 * 60 | |||||
AutoStopDurationMs = 4 * 60 * 60 * 1000 | |||||
storageTypeOBS = "obs" | |||||
autoStopDuration = 4 * 60 * 60 | |||||
AutoStopDurationMs = 4 * 60 * 60 * 1000 | |||||
CodePath = "/code/" | CodePath = "/code/" | ||||
OutputPath = "/output/" | OutputPath = "/output/" | ||||
@@ -172,7 +172,6 @@ type OrgMultiNode struct { | |||||
Node []int `json:"node"` | Node []int `json:"node"` | ||||
} | } | ||||
type Parameters struct { | type Parameters struct { | ||||
Parameter []struct { | Parameter []struct { | ||||
Label string `json:"label"` | Label string `json:"label"` | ||||
@@ -1249,6 +1249,7 @@ func GrampusTrainJobShow(ctx *context.Context) { | |||||
} | } | ||||
if result != nil { | if result != nil { | ||||
log.Info("resultliuzx:", result.JobInfo) | |||||
if len(result.JobInfo.Tasks[0].CenterID) == 1 && len(result.JobInfo.Tasks[0].CenterName) == 1 { | if len(result.JobInfo.Tasks[0].CenterID) == 1 && len(result.JobInfo.Tasks[0].CenterName) == 1 { | ||||
task.AiCenter = result.JobInfo.Tasks[0].CenterID[0] + "+" + result.JobInfo.Tasks[0].CenterName[0] | task.AiCenter = result.JobInfo.Tasks[0].CenterID[0] + "+" + result.JobInfo.Tasks[0].CenterName[0] | ||||
} | } | ||||
@@ -1358,6 +1359,23 @@ func GrampusGetLog(ctx *context.Context) { | |||||
}) | }) | ||||
return | return | ||||
} | } | ||||
result, err := grampus.GetJob(jobID) | |||||
if err != nil { | |||||
log.Error("GetJob(%s) failed:%v", job.JobName, err) | |||||
ctx.JSON(http.StatusOK, map[string]interface{}{ | |||||
"JobName": job.JobName, | |||||
"Content": content, | |||||
"CanLogDownload": false, | |||||
}) | |||||
return | |||||
} | |||||
if result != nil { | |||||
job.Status = grampus.TransTrainJobStatus(result.JobInfo.Status) | |||||
if job.Status == models.GrampusStatusFailed { | |||||
content = content + "\n" + result.ExitDiagnostics | |||||
} | |||||
} | |||||
canLogDownload := err == nil && job.IsUserHasRight(ctx.User) | canLogDownload := err == nil && job.IsUserHasRight(ctx.User) | ||||
ctx.JSON(http.StatusOK, map[string]interface{}{ | ctx.JSON(http.StatusOK, map[string]interface{}{ | ||||
"JobName": job.JobName, | "JobName": job.JobName, | ||||