From a936cffa134eece2e32229f4ff2e39f592777d1b Mon Sep 17 00:00:00 2001
From: JeshuaRen <270813223@qq.com>
Date: Thu, 13 Mar 2025 16:59:03 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=88=9B=E5=BB=BApackage?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sdks/scheduler/models.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go
index 6ba148c..7393c28 100644
--- a/sdks/scheduler/models.go
+++ b/sdks/scheduler/models.go
@@ -126,8 +126,9 @@ var JobResourceTypeUnion = types.NewTypeUnion[JobResource](
 	(*NPU)(nil),
 	(*MLU)(nil),
 	(*DCU)(nil),
-	(*Memory)(nil),
+	(*MEMORY)(nil),
 	(*PRICE)(nil),
+	(*STORAGE)(nil),
 )
 
 var _ = serder.UseTypeUnionInternallyTagged(&JobResourceTypeUnion, "type")
@@ -144,6 +145,14 @@ type CPU struct {
 	Number int64  `json:"number"`
 }
 
+type STORAGE struct {
+	serder.Metadata `union:"STORAGE"`
+	JobResourceBase
+	Type   string `json:"type"`
+	Name   string `json:"name"`
+	Number int64  `json:"number"`
+}
+
 type GPU struct {
 	serder.Metadata `union:"GPU"`
 	JobResourceBase
@@ -160,8 +169,8 @@ type NPU struct {
 	Number int64  `json:"number"`
 }
 
-type Memory struct {
-	serder.Metadata `union:"Memory"`
+type MEMORY struct {
+	serder.Metadata `union:"MEMORY"`
 	JobResourceBase
 	Type   string `json:"type"`
 	Name   string `json:"name"`