@daocloud-proto/baize 0.118.0 → 0.120.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/cluster/v1alpha1/cluster.pb.ts

@@ -15,6 +15,11 @@ export enum ClusterStatus {
   CLUSTER_STATUS_UNKNOWN = "CLUSTER_STATUS_UNKNOWN",
 }
 
+export enum Prerequisite {
+  PREREQUISITE_UNSPECIFIED = "PREREQUISITE_UNSPECIFIED",
+  Dind = "Dind",
+}
+
 export type AICluster = {
   name?: string
   withAiSuite?: boolean
@@ -147,6 +152,15 @@ export type ListResourceAffinitiesResponse = {
   page?: BaizeCommonCommon.Pagination
 }
 
+export type GetPrerequisiteRequest = {
+  cluster?: string
+  prerequisite?: Prerequisite
+}
+
+export type GetPrerequisiteResponse = {
+  ok?: boolean
+}
+
 export class ClusterService {
   static ListClusters(req: ListAIClustersRequest, initReq?: fm.InitReq): Promise<ListAIClustersResponse> {
     return fm.fetchReq<ListAIClustersRequest, ListAIClustersResponse>(`/apis/baize.io/v1alpha1/clusters?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -169,4 +183,7 @@ export class ClusterService {
   static ListClusterNodes(req: ListClusterNodesRequest, initReq?: fm.InitReq): Promise<ListClusterNodesResponse> {
     return fm.fetchReq<ListClusterNodesRequest, ListClusterNodesResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/nodes?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
   }
+  static GetPrerequisite(req: GetPrerequisiteRequest, initReq?: fm.InitReq): Promise<GetPrerequisiteResponse> {
+    return fm.fetchReq<GetPrerequisiteRequest, GetPrerequisiteResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/prerequisites/${req["prerequisite"]}?${fm.renderURLSearchParams(req, ["cluster", "prerequisite"])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -81,6 +81,7 @@ export type NotebookConfig = {
   tolerationSeconds?: string
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
   shmSize?: number
+  enableDind?: boolean
 }
 
 export type NotebookStatus = {

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,16 @@ export type ResourceFlavor = {
   nodeTaints?: BaizeCommonCommon.Taint[]
   tolerations?: BaizeCommonCommon.Toleration[]
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  nodeDetails?: BaizeCommonCommon.Node[]
+  default?: boolean
 }
 
-export type ListKueueResourceFlavorsRequest = {
+export type ListResourceFlavorsRequest = {
   cluster?: string
   page?: BaizeCommonCommon.Pagination
 }
 
-export type ListKueueResourceFlavorsResponse = {
+export type ListResourceFlavorsResponse = {
   items?: ResourceFlavor[]
   page?: BaizeCommonCommon.Pagination
 }
@@ -230,12 +238,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 +304,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 +319,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/management-api/serving/v1alpha1/serving.pb.ts

@@ -153,6 +153,7 @@ export type InferenceServing = {
   auth?: ServingAuth
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   hpaConfig?: HPAConfig
+  image?: string
 }
 
 export type InferenceServingStatusModelStatus = {
@@ -197,7 +198,6 @@ export type Metrics = {
 }
 
 export type CreateInferenceServingRequest = {
-  workspace?: number
   name?: string
   cluster?: string
   namespace?: string
@@ -210,10 +210,11 @@ export type CreateInferenceServingRequest = {
   framework?: Framework
   auth?: ServingAuth
   hpaConfig?: HPAConfig
+  workspace?: number
+  image?: string
 }
 
 export type UpdateInferenceServingRequest = {
-  workspace?: number
   name?: string
   cluster?: string
   namespace?: string
@@ -225,6 +226,8 @@ export type UpdateInferenceServingRequest = {
   framework?: Framework
   auth?: ServingAuth
   hpaConfig?: HPAConfig
+  workspace?: number
+  image?: string
 }
 
 export type SingleInferenceServingRequest = {

package/package.json

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