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"`