@daocloud-proto/baize 0.114.1 → 0.116.0

package/common/k8s.pb.ts

@@ -4,6 +4,15 @@
 * This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
 */
 
+type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
+type OneOf<T> =
+  | { [k in keyof T]?: undefined }
+  | (
+    keyof T extends infer K ?
+      (K extends string & keyof T ? { [k in K]: T[K] } & Absent<T, K>
+        : never)
+    : never);
+
 export enum KubeVolumeVolumeType {
   VOLUME_TYPE_UNSPECIFIED = "VOLUME_TYPE_UNSPECIFIED",
   PERSISTENT_VOLUME_CLAIM = "PERSISTENT_VOLUME_CLAIM",
@@ -57,6 +66,51 @@ export type PodConfig = {
   tolerationSeconds?: string
   resourceAffinities?: ResourceAffinity[]
   shmSize?: number
+  livenessProbe?: Probe
+  readinessProbe?: Probe
+}
+
+
+type BaseProbe = {
+  exec?: ExecAction
+  httpGet?: HTTPGetAction
+  tcpSocket?: TCPSocketAction
+  grpc?: GRPCAction
+}
+
+export type Probe = BaseProbe
+  & OneOf<{ initialDelaySeconds: number }>
+  & OneOf<{ timeoutSeconds: number }>
+  & OneOf<{ periodSeconds: number }>
+  & OneOf<{ successThreshold: number }>
+  & OneOf<{ failureThreshold: number }>
+  & OneOf<{ terminationGracePeriodSeconds: number }>
+
+export type ExecAction = {
+  command?: string[]
+}
+
+export type HTTPGetAction = {
+  path?: string
+  port?: number
+  host?: string
+  scheme?: string
+  httpHeaders?: HTTPHeader[]
+}
+
+export type HTTPHeader = {
+  name?: string
+  value?: string
+}
+
+export type TCPSocketAction = {
+  port?: string
+  host?: string
+}
+
+export type GRPCAction = {
+  port?: number
+  service?: string
 }
 
 export type Affinity = {

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

@@ -73,6 +73,7 @@ export type GPUSetting = {
   type?: string
   alias?: string
   resource?: GPUResourceSetting[]
+  maxUnits?: number
 }
 
 export type GPUResourceSetting = {

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

@@ -5,6 +5,7 @@
 */
 
 import * as BaizeCommonCommon from "../../../common/common.pb"
+import * as BaizeCommonK8s from "../../../common/k8s.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 import * as BaizeManagement_apiEventV1alpha1Event from "../../event/v1alpha1/event.pb"
@@ -87,6 +88,8 @@ export enum DatasetActionRequestAction {
 export type DataSource = {
   type?: DataSourceType
   uri?: string
+  share?: boolean
+  sharedToWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
   options?: DataSourceOptions
 }
 
@@ -107,9 +110,10 @@ export type Dataset = {
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
   share?: boolean
-  sharedStoWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
+  sharedToWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
   boundPVC?: DatasetBoundPVC
   lastSync?: GoogleProtobufTimestamp.Timestamp
+  resources?: BaizeCommonK8s.Resources
 }
 
 export type DataSourceOptionsConda = {
@@ -207,6 +211,7 @@ export type CreateDatasetRequest = {
   description?: string
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  resources?: BaizeCommonK8s.Resources
 }
 
 export type ListDatasetRequest = {
@@ -235,12 +240,13 @@ export type SingleDatasetRequest = {
   namespace?: string
   name?: string
   secretOptions?: SecretOptions
-  share?: boolean
-  sharedStoWorkspaces?: number[]
   uri?: string
   description?: string
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  share?: boolean
+  sharedToWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
+  resources?: BaizeCommonK8s.Resources
 }
 
 export type UpdateDatasetRequest = BaseUpdateDatasetRequest

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

@@ -138,6 +138,8 @@ export type CreateNotebookRequest = {
 export type SnapshotConfig = {
   autoSaveOnTermination?: boolean
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  containerNoPause?: boolean
+  timeout?: number
 }
 
 export type UpdateNotebookRequest = {

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

@@ -203,9 +203,11 @@ export type ListQueueWorkloadResponse = {
 export type ResourceFlavor = {
   name?: string
   topologyName?: string
+  nodeLabels?: {[key: string]: string}
 }
 
 export type ListKueueResourceFlavorsRequest = {
+  workspace?: number
   cluster?: string
   page?: BaizeCommonCommon.Pagination
 }
@@ -215,6 +217,32 @@ export type ListKueueResourceFlavorsResponse = {
   page?: BaizeCommonCommon.Pagination
 }
 
+export type CreateResourceFlavorRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  flavor?: ResourceFlavor
+}
+
+export type UpdateResourceFlavorRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  flavor?: ResourceFlavor
+}
+
+export type GetResourceFlavorRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+}
+
+export type DeleteResourceFlavorRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+}
+
 export class QueueManagement {
   static ListQueues(req: ListQueueRequest, initReq?: fm.InitReq): Promise<ListQueueResponse> {
     return fm.fetchReq<ListQueueRequest, ListQueueResponse>(`/apis/baize.io/v1alpha1/queues?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -246,4 +274,16 @@ export class QueueManagement {
   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 CreateResourceFlavor(req: CreateResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
+    return fm.fetchReq<CreateResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/kueue/resourceflavors`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateResourceFlavor(req: UpdateResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
+    return fm.fetchReq<UpdateResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/kueue/resourceflavors/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetResourceFlavor(req: GetResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
+    return fm.fetchReq<GetResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/kueue/resourceflavors/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteResourceFlavor(req: DeleteResourceFlavorRequest, initReq?: fm.InitReq): Promise<ResourceFlavor> {
+    return fm.fetchReq<DeleteResourceFlavorRequest, ResourceFlavor>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/kueue/resourceflavors/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
 }

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

@@ -68,6 +68,8 @@ export type FrameworkVLLM = {
 
 /* baize modified */ export type BaseFramework = {
   type?: FrameworkType
+  preCmds?: string
+  postCmds?: string
 }
 
 export type Framework = BaseFramework
@@ -79,16 +81,21 @@ export type Model = {
   modelPath?: string
 }
 
+export type CustomArg = {
+  name?: string
+  values?: string[]
+}
+
+export type CustomCmd = {
+  command?: string
+  args?: CustomArg[]
+}
+
 export type ServingConfigVLLMLoraModel = {
   volume?: BaizeCommonK8s.KubeVolume
   relativePath?: string
 }
 
-export type ServingConfigVLLMCustomOptionalArg = {
-  argName?: string
-  argValues?: string[]
-}
-
 export type ServingConfigVLLM = {
   trustRemoteCode?: boolean
   tensorParallelSize?: number
@@ -97,7 +104,8 @@ export type ServingConfigVLLM = {
   maxLorasRank?: number
   maxLoras?: number
   maxCpuLoras?: number
-  customOptionalArgs?: ServingConfigVLLMCustomOptionalArg[]
+  customArgs?: string
+  shouldSkip?: boolean
 }
 
 export type ServingConfigTritonModelConfig = {
@@ -214,6 +222,7 @@ export type UpdateInferenceServingRequest = {
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
   podConfig?: BaizeCommonK8s.PodConfig
+  framework?: Framework
   auth?: ServingAuth
   hpaConfig?: HPAConfig
 }

package/package.json

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