@daocloud-proto/baize 0.118.0 → 0.119.0

package/common/common.pb.ts

@@ -138,6 +138,6 @@ export type Node = {
   name?: string
   arch?: string
   labels?: {[key: string]: string}
-  gpuModel?: string
+  gpuModels?: string[]
   status?: NodeStatus
 }

package/common/k8s.pb.ts

@@ -68,6 +68,7 @@ export type PodConfig = {
   shmSize?: number
   livenessProbe?: Probe
   readinessProbe?: Probe
+  startupProbe?: Probe
 }
 
 

package/management-api/queue/v1alpha1/queue.pb.ts

@@ -65,7 +65,13 @@ export type QueueStatus = {
 export type FlavorQuota = {
   name?: string
   resources?: QueueResource[]
+}
+
+export type QueueFlavor = {
+  name?: string
+  resources?: QueueResource[]
   resourcesAvailable?: QueueResource[]
+  nodes?: BaizeCommonCommon.Node[]
 }
 
 export type Queue = {
@@ -82,7 +88,7 @@ export type Queue = {
   strategy?: QueueStrategy
   workspaceAlias?: string
   preemptionWithinClusterQueue?: QueuePreemptionWithinClusterQueuePolicy
-  flavors?: FlavorQuota[]
+  flavors?: QueueFlavor[]
   tasEnabled?: boolean
 }
 
@@ -209,14 +215,15 @@ export type ResourceFlavor = {
   nodeTaints?: BaizeCommonCommon.Taint[]
   tolerations?: BaizeCommonCommon.Toleration[]
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  nodeDetails?: BaizeCommonCommon.Node[]
 }
 
-export type ListKueueResourceFlavorsRequest = {
+export type ListResourceFlavorsRequest = {
   cluster?: string
   page?: BaizeCommonCommon.Pagination
 }
 
-export type ListKueueResourceFlavorsResponse = {
+export type ListResourceFlavorsResponse = {
   items?: ResourceFlavor[]
   page?: BaizeCommonCommon.Pagination
 }
@@ -230,12 +237,12 @@ export type Topology = {
   levels?: TopologyLevel[]
 }
 
-export type ListKueueTopologiesRequest = {
+export type ListTopologiesRequest = {
   cluster?: string
   page?: BaizeCommonCommon.Pagination
 }
 
-export type ListKueueTopologiesResponse = {
+export type ListTopologiesResponse = {
   items?: Topology[]
   page?: BaizeCommonCommon.Pagination
 }
@@ -296,8 +303,8 @@ export class QueueManagement {
   static ListQueueWorkloads(req: ListQueueWorkloadRequest, initReq?: fm.InitReq): Promise<ListQueueWorkloadResponse> {
     return fm.fetchReq<ListQueueWorkloadRequest, ListQueueWorkloadResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues/${req["queueName"]}/workloads?${fm.renderURLSearchParams(req, ["cluster", "queueName"])}`, {...initReq, method: "GET"})
   }
-  static ListKueueResourceFlavors(req: ListKueueResourceFlavorsRequest, initReq?: fm.InitReq): Promise<ListKueueResourceFlavorsResponse> {
-    return fm.fetchReq<ListKueueResourceFlavorsRequest, ListKueueResourceFlavorsResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/resourceflavors?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
+  static ListResourceFlavors(req: ListResourceFlavorsRequest, initReq?: fm.InitReq): Promise<ListResourceFlavorsResponse> {
+    return fm.fetchReq<ListResourceFlavorsRequest, ListResourceFlavorsResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/resourceflavors?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
   }
   static CreateResourceFlavor(req: CreateResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
     return fm.fetchReq<CreateResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/resourceflavors`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
@@ -311,7 +318,7 @@ export class QueueManagement {
   static DeleteResourceFlavor(req: DeleteResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
     return fm.fetchReq<DeleteResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/resourceflavors/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
   }
-  static ListKueueTopologies(req: ListKueueTopologiesRequest, initReq?: fm.InitReq): Promise<ListKueueTopologiesResponse> {
-    return fm.fetchReq<ListKueueTopologiesRequest, ListKueueTopologiesResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/topologies?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
+  static ListTopologies(req: ListTopologiesRequest, initReq?: fm.InitReq): Promise<ListTopologiesResponse> {
+    return fm.fetchReq<ListTopologiesRequest, ListTopologiesResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/kueue/topologies?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/baize",
-  "version": "v0.118.0",
+  "version": "v0.119.0",
   "description": "",
   "author": "",
   "license": "ISC"