@daocloud-proto/baize 0.115.0 → 0.116.1

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

@@ -158,12 +158,16 @@ export type DataSourceOptionsHttp = {
   headers?: {[key: string]: string}
 }
 
+export type DataSourceOptionsReference = {
+  sourceWorkspace?: number
+}
+
 
 /* baize modified */ export type BaseDataSourceOptions = {
 }
 
 export type DataSourceOptions = BaseDataSourceOptions
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference }>
 
 export type SecretOptionsBasicAuth = {
   username?: string
@@ -218,6 +222,7 @@ export type ListDatasetRequest = {
   workspace?: number
   cluster?: string
   namespace?: string
+  sharedTo?: number
   page?: BaizeCommonCommon.Pagination
 }
 
@@ -250,7 +255,7 @@ export type SingleDatasetRequest = {
 }
 
 export type UpdateDatasetRequest = BaseUpdateDatasetRequest
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference }>
 
 export type DatasetActionRequest = {
   workspace?: number

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/package.json

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