@daocloud-proto/baize 0.110.2 → 0.110.4

package/common/k8s.pb.ts

@@ -9,6 +9,7 @@ export enum KubeVolumeVolumeType {
   PERSISTENT_VOLUME_CLAIM = "PERSISTENT_VOLUME_CLAIM",
   DATASET = "DATASET",
   DATASET_RUNTIME_ENV = "DATASET_RUNTIME_ENV",
+  MODEL = "MODEL",
 }
 
 export enum KubeVolumeDataDirection {
@@ -17,6 +18,11 @@ export enum KubeVolumeDataDirection {
   OUTPUT = "OUTPUT",
 }
 
+export enum ResourceAffinityType {
+  RESOURCE_AFFINITY_TYPE_UNSPECIFIED = "RESOURCE_AFFINITY_TYPE_UNSPECIFIED",
+  HAMI_VGPU = "HAMI_VGPU",
+}
+
 export type Resources = {
   requests?: {[key: string]: string}
   limits?: {[key: string]: string}
@@ -35,6 +41,11 @@ export type KubeVolume = {
   dataDirection?: KubeVolumeDataDirection
 }
 
+export type ResourceAffinity = {
+  type?: ResourceAffinityType
+  value?: string
+}
+
 export type PodConfig = {
   kubeEnvs?: KubeEnv[]
   kubeVolumes?: KubeVolume[]
@@ -44,6 +55,7 @@ export type PodConfig = {
   priorityClass?: string
   queue?: string
   tolerationSeconds?: string
+  resourceAffinities?: ResourceAffinity[]
 }
 
 export type Affinity = {

package/management-api/cluster/v1alpha1/cluster.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"
 
 export enum ClusterStatus {
@@ -117,6 +118,22 @@ export type ClusterSettingsRequest = {
   workspace?: number
 }
 
+export type ClusterResourceType = {
+  type?: BaizeCommonK8s.ResourceAffinityType
+  values?: string[]
+}
+
+export type ListResourceAffinitiesRequest = {
+  workspace?: number
+  cluster?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListResourceAffinitiesResponse = {
+  items?: ClusterResourceType[]
+  page?: BaizeCommonCommon.Pagination
+}
+
 export class ClusterService {
   static ListClusters(req: ListAIClustersRequest, initReq?: fm.InitReq): Promise<ListAIClustersResponse> {
     return fm.fetchReq<ListAIClustersRequest, ListAIClustersResponse>(`/apis/baize.io/v1alpha1/clusters?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -133,4 +150,7 @@ export class ClusterService {
   static ListStorageClasses(req: ListStorageClassesRequest, initReq?: fm.InitReq): Promise<ListStorageClassesResponse> {
     return fm.fetchReq<ListStorageClassesRequest, ListStorageClassesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/storage-classes?${fm.renderURLSearchParams(req, ["workspace", "cluster"])}`, {...initReq, method: "GET"})
   }
+  static ListResourceAffinities(req: ListResourceAffinitiesRequest, initReq?: fm.InitReq): Promise<ListResourceAffinitiesResponse> {
+    return fm.fetchReq<ListResourceAffinitiesRequest, ListResourceAffinitiesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/resource-affinities?${fm.renderURLSearchParams(req, ["workspace", "cluster"])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -101,6 +101,7 @@ export type Dataset = {
   share?: boolean
   sharedStoWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
   boundPVC?: DatasetBoundPVC
+  lastSync?: GoogleProtobufTimestamp.Timestamp
 }
 
 export type DataSourceOptionsConda = {

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

@@ -42,6 +42,14 @@ export enum ModelVersionSourceType {
   DATASET = "DATASET",
 }
 
+export enum ModelVersionStatusPhase {
+  PHASE_UNSPECIFIED = "PHASE_UNSPECIFIED",
+  PENDING = "PENDING",
+  CREATING = "CREATING",
+  READY = "READY",
+  ERROR = "ERROR",
+}
+
 export enum ModelStatusPhase {
   PHASE_UNSPECIFIED = "PHASE_UNSPECIFIED",
   PENDING = "PENDING",
@@ -69,10 +77,15 @@ export type DatasetRef = {
 export type ModelVersionSource = BaseModelVersionSource
   & OneOf<{ datasetRef: DatasetRef }>
 
+export type ModelVersionStatus = {
+  phase?: ModelVersionStatusPhase
+}
+
 export type Version = {
   version?: string
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   source?: ModelVersionSource
+  status?: ModelVersionStatus
 }
 
 export type ModelStatus = {
@@ -134,6 +147,49 @@ export type CreateModelRequest = {
   sharedToWorkspaces?: Workspace[]
 }
 
+export type UpdateModelRequest = {
+  workspace?: number
+  name?: string
+  cluster?: string
+  namespace?: string
+  format?: Format
+  architecture?: Architecture
+  license?: string
+  isLLM?: boolean
+  latestVersion?: string
+  iconUrl?: string
+  tags?: string[]
+  readme?: string
+  share?: boolean
+  sharedToWorkspaces?: Workspace[]
+}
+
+export type CreateModelVersionRequest = {
+  name?: string
+  cluster?: string
+  namespace?: string
+  workspace?: number
+  version?: string
+  source?: ModelVersionSource
+}
+
+export type UpdateModelVersionRequest = {
+  name?: string
+  cluster?: string
+  namespace?: string
+  workspace?: number
+  version?: string
+  source?: ModelVersionSource
+}
+
+export type DeleteModelVersionRequest = {
+  name?: string
+  cluster?: string
+  namespace?: string
+  workspace?: number
+  version?: string
+}
+
 export class ModelManagement {
   static ListModels(req: ListModelsRequest, initReq?: fm.InitReq): Promise<ListModelsResponse> {
     return fm.fetchReq<ListModelsRequest, ListModelsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/models?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
@@ -147,7 +203,19 @@ export class ModelManagement {
   static CreateModel(req: CreateModelRequest, initReq?: fm.InitReq): Promise<Model> {
     return fm.fetchReq<CreateModelRequest, Model>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }
+  static UpdateModel(req: UpdateModelRequest, initReq?: fm.InitReq): Promise<Model> {
+    return fm.fetchReq<UpdateModelRequest, Model>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
   static DeleteModel(req: SingleModelRequest, initReq?: fm.InitReq): Promise<Model> {
     return fm.fetchReq<SingleModelRequest, Model>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
   }
+  static CreateModelVersion(req: CreateModelVersionRequest, initReq?: fm.InitReq): Promise<Version> {
+    return fm.fetchReq<CreateModelVersionRequest, Version>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models/${req["name"]}/versions`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateModelVersion(req: UpdateModelVersionRequest, initReq?: fm.InitReq): Promise<Version> {
+    return fm.fetchReq<UpdateModelVersionRequest, Version>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models/${req["name"]}/versions/${req["version"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DeleteModelVersion(req: DeleteModelVersionRequest, initReq?: fm.InitReq): Promise<Version> {
+    return fm.fetchReq<DeleteModelVersionRequest, Version>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/models/${req["name"]}/versions/${req["version"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
 }

package/package.json

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