diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go index d545b78..36804c6 100644 --- a/sdks/scheduler/models.go +++ b/sdks/scheduler/models.go @@ -125,11 +125,23 @@ type PCMInferenceJobInfo struct { type AIJobInfo struct { serder.Metadata `union:"AI"` JobInfoBase - Type string `json:"type"` - Name string `json:"name"` - Description string `json:"description"` - Files JobFilesInfo `json:"files"` - JobResources JobResources `json:"jobResources"` + Type string `json:"type"` + Name string `json:"name"` + Description string `json:"description"` + Files JobFilesInfo `json:"files"` + JobResources JobResources `json:"jobResources"` + ResourceChoice ResourceChoice `json:"resourceChoice"` +} + +type ResourceChoice struct { + Type string `json:"type"` + ResourceScopes []ResourceScope `json:"resourceScopes"` +} + +type ResourceScope struct { + Name string `json:"name"` + Min float64 `json:"min"` + Max float64 `json:"max"` } type BindingJobInfo struct { @@ -443,8 +455,9 @@ func (i *JobFileInfoBase) Noop() {} type BindingJobFileInfo struct { serder.Metadata `union:"Binding"` JobFileInfoBase - Type string `json:"type"` - BindingID int64 `json:"bindingID"` + Type string `json:"type"` + BindingID int64 `json:"bindingID"` + BindingName string `json:"bindingName"` } type PackageJobFileInfo struct { @@ -471,8 +484,9 @@ type DataReturnJobFileInfo struct { type ImageJobFileInfo struct { serder.Metadata `union:"Image"` JobFileInfoBase - Type string `json:"type"` - ImageID ImageID `json:"imageID"` + Type string `json:"type"` + ImageID ImageID `json:"imageID"` + ImageName string `json:"imageName"` } type JobRuntimeInfo struct {