From 7be564b435eeeebc4589e7a3617abe9478e5de8c Mon Sep 17 00:00:00 2001
From: JeshuaRen <270813223@qq.com>
Date: Tue, 15 Jul 2025 15:50:13 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=94=B6=E5=8F=91=E9=80=9A?=
 =?UTF-8?q?=E7=9F=A5=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sdks/pcmscheduler/models.go |  3 ++-
 sdks/scheduler/models.go    | 12 ++++++------
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/sdks/pcmscheduler/models.go b/sdks/pcmscheduler/models.go
index 7f2338b..a116ab8 100644
--- a/sdks/pcmscheduler/models.go
+++ b/sdks/pcmscheduler/models.go
@@ -102,7 +102,8 @@ const (
 	Inference = "Inference"
 	Train     = "Train"
 
-	BroadcastWait = "All"
+	BroadcastWait = "BroadcastWait"
+	WaitBroadcast = "WaitBroadcast"
 	Broadcast     = "Broadcast"
 	Wait          = "Wait"
 )
diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go
index 2a7b0ef..1a2c537 100644
--- a/sdks/scheduler/models.go
+++ b/sdks/scheduler/models.go
@@ -188,19 +188,19 @@ type UploadJobInfo struct {
 type BroadcastWaitInfo struct {
 	serder.Metadata `union:"BroadcastWait"`
 	JobInfoBase
-	Type       string    `json:"type"`
-	HandleType string    `json:"handleType"`
-	Broadcast  Broadcast `json:"broadcast"`
-	Wait       Wait      `json:"wait"`
+	Type       string      `json:"type"`
+	HandleType string      `json:"handleType"`
+	Broadcast  []Broadcast `json:"broadcast"`
+	Wait       []Wait      `json:"wait"`
 }
 
 type Broadcast struct {
 	NextJobSetID   JobSetID `json:"nextJobSetID"`
-	NextLocalJobID JobID    `json:"nextLocalJobID"`
+	NextLocalJobID string   `json:"nextLocalJobID"`
 }
 
 type Wait struct {
-	WaitJobIDs []JobID `json:"waitJobIDs"`
+	WaitJobID string `json:"waitJobID"`
 }
 
 type NotifyJobInfo struct {