调整通知接口参数

sdks/scheduler/models.go

@@ -409,10 +409,11 @@ type DataPreprocessJobInfo struct {
 type DataReturnJobInfo struct {
 	serder.Metadata `union:"DataReturn"`
 	JobInfoBase
-	Type             string          `json:"type"`
-	BucketID         cdssdk.BucketID `json:"bucketID"`
-	TargetLocalJobID string          `json:"targetLocalJobID"`
-	ReportMessage    []ReportMessage `json:"reportMessage"`
+	Type             string               `json:"type"`
+	BucketID         cdssdk.BucketID      `json:"bucketID"`
+	TargetLocalJobID string               `json:"targetLocalJobID"`
+	Report           TrainJobStatusReport `json:"report"`
+	//ReportMessage    []ReportMessage `json:"reportMessage"`
 }
 
 // MultiInstanceJobInfo 多实例(推理任务)
@@ -665,8 +666,9 @@ type BindingJobOutput struct {
 type DataReturnJobOutput struct {
 	serder.Metadata `union:"DataReturn"`
 	JobOutputBase
-	Type          string          `json:"type"`
-	ReportMessage []ReportMessage `json:"reportMessage"`
+	Type   string               `json:"type"`
+	Report TrainJobStatusReport `json:"report"`
+	//ReportMessage []ReportMessage `json:"reportMessage"`
 }
 
 type ReportMessage struct {
@@ -677,3 +679,41 @@ type ReportMessage struct {
 	ClusterID ClusterID `json:"clusterID"`
 	Output    string    `json:"output"`
 }
+
+type JobStatusReport interface {
+	Report()
+}
+
+var JobStatusReportTypeUnion = types.NewTypeUnion[JobStatusReport](
+	(*TrainJobStatusReport)(nil),
+	(*InferenceJobStatusReport)(nil),
+)
+
+var _ = serder.UseTypeUnionInternallyTagged(&JobStatusReportTypeUnion, "type")
+
+type JobStatusReportBase struct{}
+
+func (d *JobStatusReportBase) Report() {}
+
+type TrainJobStatusReport struct {
+	serder.Metadata `union:"Train"`
+	JobStatusReportBase
+	Type      string    `json:"type"`
+	TaskName  string    `json:"taskName"`
+	TaskID    string    `json:"taskID"`
+	Status    bool      `json:"status"`
+	Message   string    `json:"message"`
+	ClusterID ClusterID `json:"clusterID"`
+	Output    string    `json:"output"`
+}
+
+type InferenceJobStatusReport struct {
+	serder.Metadata `union:"Inference"`
+	JobStatusReportBase
+	Type     string `json:"type"`
+	TaskName string `json:"taskName"`
+	TaskID   string `json:"taskID"`
+	Status   bool   `json:"status"`
+	Message  string `json:"message"`
+	URL      string `json:"url"`
+}