@daocloud-proto/baize 0.110.2 → 0.110.3

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 = {

package/package.json

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