@daocloud-proto/baize 0.106.2 → 0.107.2

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

@@ -6,9 +6,18 @@
 
 import * as BaizeCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
+
+export enum ClusterStatus {
+  CLUSTER_STATUS_UNSPECIFIED = "CLUSTER_STATUS_UNSPECIFIED",
+  CLUSTER_STATUS_RUNNING = "CLUSTER_STATUS_RUNNING",
+  CLUSTER_STATUS_NOT_RUNNING = "CLUSTER_STATUS_NOT_RUNNING",
+  CLUSTER_STATUS_UNKNOWN = "CLUSTER_STATUS_UNKNOWN",
+}
+
 export type AICluster = {
   name?: string
   withAiSuite?: boolean
+  clusterStatus?: ClusterStatus
 }
 
 export type ListAIClustersRequest = {
@@ -30,6 +39,7 @@ export type ListClusterNamespaceRequest = {
 export type ClusterNamespace = {
   cluster?: string
   name?: string
+  clusterStatus?: ClusterStatus
 }
 
 export type ListClusterNamespaceResponse = {

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

@@ -231,6 +231,11 @@ export type RelatedInstancesResponse = {
   page?: BaizeCommonCommon.Pagination
 }
 
+export type DatasetSyncProcess = {
+  status?: DatasetStatus
+  syncProcess?: number
+}
+
 export class DatasetManagement {
   static ListDatasets(req: ListDatasetRequest, initReq?: fm.InitReq): Promise<ListDatasetResponse> {
     return fm.fetchReq<ListDatasetRequest, ListDatasetResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
@@ -238,6 +243,9 @@ export class DatasetManagement {
   static GetDataset(req: SingleDatasetRequest, initReq?: fm.InitReq): Promise<Dataset> {
     return fm.fetchReq<SingleDatasetRequest, Dataset>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
+  static GetDatasetSyncProcess(req: SingleDatasetRequest, initReq?: fm.InitReq): Promise<DatasetSyncProcess> {
+    return fm.fetchReq<SingleDatasetRequest, DatasetSyncProcess>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}/sync-process?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
   static CreateDataset(req: CreateDatasetRequest, initReq?: fm.InitReq): Promise<Dataset> {
     return fm.fetchReq<CreateDatasetRequest, Dataset>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }

package/management-api/job/v1alpha1/common.pb.ts

@@ -36,6 +36,11 @@ export enum JobRole {
   TF_EVALUATOR = "TF_EVALUATOR",
   PD_MASTER = "PD_MASTER",
   PD_WORKER = "PD_WORKER",
+  MX_SCHEDULER = "MX_SCHEDULER",
+  MX_SERVER = "MX_SERVER",
+  MX_WORKER = "MX_WORKER",
+  MPI_LAUNCHER = "MPI_LAUNCHER",
+  MPI_WORKER = "MPI_WORKER",
 }
 
 export type CommonConfig = {

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

@@ -26,6 +26,8 @@ export enum JobType {
   PYTORCH = "PYTORCH",
   TENSORFLOW = "TENSORFLOW",
   PADDLE = "PADDLE",
+  MPI = "MPI",
+  MXNET = "MXNET",
 }
 
 export enum RestartPolicy {
@@ -37,6 +39,7 @@ export enum RestartPolicy {
 export enum JobActionRequestAction {
   JOB_ACTION_UNSPECIFIED = "JOB_ACTION_UNSPECIFIED",
   RESTART = "RESTART",
+  CHANGE_PRIORITY = "CHANGE_PRIORITY",
 }
 
 export enum JobSchedulersResponseSchedulerFeature {
@@ -112,6 +115,13 @@ export type TrainingConfig = BaseTrainingConfig
   & OneOf<{ maxRetries: number }>
   & OneOf<{ maxTrainingDuration: string }>
 
+
+/* baize modified */ export type BaseJobActionRequestParams = {
+}
+
+export type JobActionRequestParams = BaseJobActionRequestParams
+  & OneOf<{ priorityClass: string }>
+
 export type JobActionRequest = {
   workspace?: number
   type?: JobType
@@ -119,6 +129,7 @@ export type JobActionRequest = {
   namespace?: string
   name?: string
   action?: JobActionRequestAction
+  params?: JobActionRequestParams
 }
 
 export type JobSchedulersResponseScheduler = {

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

@@ -16,6 +16,8 @@ export enum ResourceType {
   PADDLE = "PADDLE",
   NOTEBOOK = "NOTEBOOK",
   INFERENCE = "INFERENCE",
+  MXNET = "MXNET",
+  MPI = "MPI",
 }
 
 export type PodRequest = {

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

@@ -29,6 +29,7 @@ export enum ServiceType {
 export enum FrameworkType {
   FRAMEWORK_TYPE_UNSPECIFIED = "FRAMEWORK_TYPE_UNSPECIFIED",
   FRAMEWORK_TYPE_TRITON = "FRAMEWORK_TYPE_TRITON",
+  FRAMEWORK_TYPE_VLLM = "FRAMEWORK_TYPE_VLLM",
 }
 
 export enum FrameworkTritonBackend {
@@ -58,13 +59,16 @@ export type FrameworkTriton = {
   backend?: FrameworkTritonBackend
 }
 
+export type FrameworkVLLM = {
+}
+
 
 /* baize modified */ export type BaseFramework = {
   type?: FrameworkType
 }
 
 export type Framework = BaseFramework
-  & OneOf<{ triton: FrameworkTriton }>
+  & OneOf<{ triton: FrameworkTriton; vllm: FrameworkVLLM }>
 
 export type Model = {
   name?: string
@@ -72,9 +76,19 @@ export type Model = {
   modelPath?: string
 }
 
+export type ServingConfigVLLMLoraModel = {
+  volume?: BaizeCommonK8s.KubeVolume
+  relativePath?: string
+}
+
 export type ServingConfigVLLM = {
   trustRemoteCode?: boolean
   tensorParallelSize?: number
+  enableLora?: boolean
+  loraModels?: ServingConfigVLLMLoraModel[]
+  maxLorasRank?: number
+  maxLoras?: number
+  maxCpuLoras?: number
 }
 
 export type ServingConfigTritonModelConfig = {
@@ -100,7 +114,7 @@ export type ServingConfigTritonServingConfig = BaseServingConfigTritonServingCon
 }
 
 export type ServingConfig = BaseServingConfig
-  & OneOf<{ triton: ServingConfigTritonServingConfig }>
+  & OneOf<{ triton: ServingConfigTritonServingConfig; vllm: ServingConfigVLLM }>
 
 export type ServiceConfig = {
   serviceType?: ServiceType

package/package.json

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