services/ai_task_service/cluster/cloudbrain_one.go

@@ -173,10 +173,15 @@ func (c CloudbrainOneClusterAdapter) GetNoteBookOperationProfile(jobId string) (
 		log.Error("ConvertToJobResultPayload failed:%v", err)
 		return nil, err
 	}
-	return parseDiagnosticsToOperationProfile(result.JobStatus.AppExitDiagnostics), nil
+
+	taskRoles := result.TaskRoles
+	taskRes, _ := models.ConvertToTaskPod(taskRoles[cloudbrain.SubTaskName].(map[string]interface{}))
+	ExitDiagnostics := taskRes.TaskStatuses[0].ExitDiagnostics
+
+	return parseDiagnosticsToOperationProfile(result.JobStatus.AppExitDiagnostics,ExitDiagnostics), nil
 }
 
-func parseDiagnosticsToOperationProfile(appExitDiagnostics string) *entity.OperationProfile {
+func parseDiagnosticsToOperationProfile(appExitDiagnostics string, exitDiagnostics string) *entity.OperationProfile {
 	if appExitDiagnostics == "" {
 		return nil
 	}
@@ -211,6 +216,12 @@ func parseDiagnosticsToOperationProfile(appExitDiagnostics string) *entity.Opera
 			Action:  e.Action,
 		})
 	}
+	if exitDiagnostics != ""{
+		events = append(events, entity.ProfileEvent{
+			Message: exitDiagnostics,
+			Reason: "Error",
+		})
+	}
 	return &entity.OperationProfile{Events: events}
 }