@daocloud-proto/baize 0.106.1 → 0.107.1

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

@@ -212,6 +212,30 @@ export type GetDatasetCondaOptionsResponse = {
   pythonVersions?: string[]
 }
 
+export type RelatedInstancesRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type RelatedInstancesResponseInstance = {
+  name?: string
+  namespace?: string
+  cluster?: string
+}
+
+export type RelatedInstancesResponse = {
+  items?: RelatedInstancesResponseInstance[]
+  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"})
@@ -219,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)})
   }
@@ -234,4 +261,7 @@ export class DatasetManagement {
   static GetDatasetCondaOptions(req: GetDatasetCondaOptionsRequest, initReq?: fm.InitReq): Promise<GetDatasetCondaOptionsResponse> {
     return fm.fetchReq<GetDatasetCondaOptionsRequest, GetDatasetCondaOptionsResponse>(`/apis/baize.io/v1alpha1/datasets/conda/options?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
+  static GetRelatedInstances(req: RelatedInstancesRequest, initReq?: fm.InitReq): Promise<RelatedInstancesResponse> {
+    return fm.fetchReq<RelatedInstancesRequest, RelatedInstancesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}/related-instances?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -37,6 +37,7 @@ export enum RestartPolicy {
 export enum JobActionRequestAction {
   JOB_ACTION_UNSPECIFIED = "JOB_ACTION_UNSPECIFIED",
   RESTART = "RESTART",
+  CHANGE_PRIORITY = "CHANGE_PRIORITY",
 }
 
 export enum JobSchedulersResponseSchedulerFeature {
@@ -112,6 +113,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 +127,7 @@ export type JobActionRequest = {
   namespace?: string
   name?: string
   action?: JobActionRequestAction
+  params?: JobActionRequestParams
 }
 
 export type JobSchedulersResponseScheduler = {

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.1",
+  "version": "v0.107.1",
   "description": "",
   "author": "",
   "license": "ISC"