@daocloud-proto/baize 0.114.0 → 0.115.0

package/common/common.pb.ts

@@ -17,7 +17,7 @@ export type Pagination = {
 export type SearchIndex = {
   searchKey?: string
   objectPath?: string
-  customPath?: string
+  customPath?: {[key: string]: string}
 }
 
 export type AuthResourceAction = {

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/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/serving/v1alpha1/serving.pb.ts

@@ -55,6 +55,7 @@ export enum InferenceServingStatusPhase {
 export enum ServingAuthAuthType {
   AUTH_TYPE_UNSPECIFIED = "AUTH_TYPE_UNSPECIFIED",
   TRITON_RESTRICTED_KEY = "TRITON_RESTRICTED_KEY",
+  VLLM_API_KEY = "VLLM_API_KEY",
 }
 
 export type FrameworkTriton = {
@@ -67,6 +68,8 @@ export type FrameworkVLLM = {
 
 /* baize modified */ export type BaseFramework = {
   type?: FrameworkType
+  preCmds?: string
+  postCmds?: string
 }
 
 export type Framework = BaseFramework
@@ -78,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
@@ -96,7 +104,8 @@ export type ServingConfigVLLM = {
   maxLorasRank?: number
   maxLoras?: number
   maxCpuLoras?: number
-  customOptionalArgs?: ServingConfigVLLMCustomOptionalArg[]
+  customArgs?: string
+  shouldSkip?: boolean
 }
 
 export type ServingConfigTritonModelConfig = {
@@ -141,7 +150,7 @@ export type InferenceServing = {
   podConfig?: BaizeCommonK8s.PodConfig
   status?: InferenceServingStatus
   lastUpdated?: GoogleProtobufTimestamp.Timestamp
-  authType?: ServingAuthAuthType
+  auth?: ServingAuth
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   hpaConfig?: HPAConfig
 }
@@ -169,7 +178,7 @@ export type ServingAuthTritonRestrictedKeyValue = {
 }
 
 export type ServingAuth = BaseServingAuth
-  & OneOf<{ tritonRestrictedKeyValue: ServingAuthTritonRestrictedKeyValue }>
+  & OneOf<{ tritonRestrictedKeyValue: ServingAuthTritonRestrictedKeyValue; vllmApiKey: string }>
 
 export type HPAConfig = {
   enabled?: boolean
@@ -213,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.0",
+  "version": "v0.115.0",
   "description": "",
   "author": "",
   "license": "ISC"