@daocloud-proto/baize 0.108.0 → 0.108.2

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

@@ -238,7 +238,7 @@ export type DatasetSyncProcess = {
 
 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"})
+    return fm.fetchReq<ListDatasetRequest, ListDatasetResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/datasets?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }
   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"})

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

@@ -8,6 +8,12 @@ import * as BaizeCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 
+export enum Scope {
+  SCOPE_UNKNOWN = "SCOPE_UNKNOWN",
+  SCOPE_GLOBAL = "SCOPE_GLOBAL",
+  SCOPE_WORKSPACE = "SCOPE_WORKSPACE",
+}
+
 export enum RegistryType {
   REGISTRY_TYPE_UNSPECIFIED = "REGISTRY_TYPE_UNSPECIFIED",
   REGISTRY_TYPE_HARBOR = "REGISTRY_TYPE_HARBOR",
@@ -15,8 +21,14 @@ export enum RegistryType {
   REGISTRY_TYPE_JFROG = "REGISTRY_TYPE_JFROG",
 }
 
+export enum Visibility {
+  VISIBLE_UNKNOWN = "VISIBLE_UNKNOWN",
+  VISIBLE_PUBLIC = "VISIBLE_PUBLIC",
+  VISIBLE_PRIVATE = "VISIBLE_PRIVATE",
+}
+
 export type ListRegistriesRequest = {
-  workspace?: string
+  workspace?: number
   global?: boolean
   page?: BaizeCommonCommon.Pagination
 }
@@ -30,10 +42,11 @@ export type Registry = {
   host?: string
   name?: string
   type?: RegistryType
+  scope?: Scope
 }
 
 export type ListProjectsRequest = {
-  workspace?: string
+  workspace?: number
   registry?: string
   public?: boolean
   page?: BaizeCommonCommon.Pagination
@@ -48,13 +61,13 @@ export type ListProjectsResponse = {
 
 export type Project = {
   name?: string
+  visible?: Visibility
 }
 
 export type ListRepositoriesRequest = {
-  workspace?: string
+  workspace?: number
   registry?: string
   project?: string
-  showArtifacts?: boolean
   page?: BaizeCommonCommon.Pagination
 }
 
@@ -65,13 +78,11 @@ export type ListRepositoriesResponse = {
 
 export type Repository = {
   name?: string
-  public?: boolean
   updateTimestamp?: GoogleProtobufTimestamp.Timestamp
-  artifacts?: Artifact[]
 }
 
 export type ListArtifactsRequest = {
-  workspace?: string
+  workspace?: number
   registry?: string
   project?: string
   repository?: string
@@ -90,19 +101,45 @@ export type Artifact = {
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
 }
 
+export type ProjectInfo = {
+  registry?: Registry
+  project?: Project
+}
+
+export type ArtifactInfo = {
+  registry?: Registry
+  project?: Project
+  repository?: Repository
+  artifact?: Artifact
+}
+
 export type Tag = {
   name?: string
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
 }
 
-export type ImageInfo = {
+export type ImageConfig = {
+  input?: InputImage
+  kangaroo?: KangarooImage
+}
+
+export type InputImage = {
+  name?: string
+  pullSecret?: string
+}
+
+export type KangarooImageMetadata = {
+  visible?: Visibility
+  scope?: Scope
+}
+
+export type KangarooImage = {
   registry?: string
   project?: string
   image?: string
   tag?: string
-  public?: boolean
-  global?: boolean
-  origin?: string
+  meta?: KangarooImageMetadata
+  registryName?: string
 }
 
 export class ImageService {

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

@@ -61,10 +61,14 @@ export type JobCreationBaseConfig = {
   workingDir?: string
   shmSize?: number
   noOverrideEnvPath?: boolean
-  imageInfo?: BaizeManagement_apiImageV1alpha1Image.ImageInfo
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type JobRoleDifferenceConfig = {
   replicas?: number
   resources?: BaizeCommonK8s.Resources
+}
+
+export type RecoveryConfig = {
+  enable?: boolean
 }

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

@@ -65,6 +65,7 @@ export type Job = {
   totalResources?: BaizeCommonK8s.Resources
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   trainingConfig?: TrainingConfig
+  recoveryConfig?: BaizeManagement_apiJobV1alpha1Common.RecoveryConfig
 }
 
 export type ListJobsRequest = {
@@ -104,6 +105,7 @@ export type CreateJobRequest = {
   roleConfig?: {[key: string]: BaizeManagement_apiJobV1alpha1Common.JobRoleDifferenceConfig}
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   trainingConfig?: TrainingConfig
+  recoveryConfig?: BaizeManagement_apiJobV1alpha1Common.RecoveryConfig
 }
 
 

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

@@ -149,6 +149,15 @@ export type WorkloadGrafanaDashboardRequest = {
   to?: string
 }
 
+export type InferenceServiceGrafanaDashboardRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  inference?: string
+  from?: string
+  to?: string
+}
+
 export type GrafanaDashboard = {
   urlEN?: string
   urlZH?: string
@@ -186,4 +195,7 @@ export class MetricsService {
   static GetWorkloadGrafanaDashboard(req: WorkloadGrafanaDashboardRequest, initReq?: fm.InitReq): Promise<GrafanaDashboard> {
     return fm.fetchReq<WorkloadGrafanaDashboardRequest, GrafanaDashboard>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/grafana-dashboards?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
   }
+  static GetInferenceServiceGrafanaDashboard(req: InferenceServiceGrafanaDashboardRequest, initReq?: fm.InitReq): Promise<GrafanaDashboard> {
+    return fm.fetchReq<InferenceServiceGrafanaDashboardRequest, GrafanaDashboard>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/inference-serving/${req["inference"]}/service-grafana-dashboards?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "inference"])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -71,7 +71,7 @@ export type NotebookConfig = {
   enableSSH?: boolean
   affinity?: BaizeCommonK8s.Affinity
   tolerationSeconds?: string
-  imageInfo?: BaizeManagement_apiImageV1alpha1Image.ImageInfo
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type NotebookStatus = {
@@ -93,6 +93,7 @@ export type Notebook = {
   priorityClass?: string
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   description?: string
+  snapshotConfig?: SnapshotConfig
 }
 
 export type ListNotebooksResponse = {
@@ -122,6 +123,12 @@ export type CreateNotebookRequest = {
   priorityClass?: string
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   description?: string
+  snapshotConfig?: SnapshotConfig
+}
+
+export type SnapshotConfig = {
+  autoSaveOnTermination?: boolean
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type UpdateNotebookRequest = {
@@ -134,6 +141,7 @@ export type UpdateNotebookRequest = {
   priorityClass?: string
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   description?: string
+  snapshotConfig?: SnapshotConfig
 }
 
 export type NotebookActionRequest = {
@@ -144,6 +152,21 @@ export type NotebookActionRequest = {
   action?: NotebookActionRequestAction
 }
 
+export type CreateNotebookSnapshotTaskRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type SnapshotTask = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  triggerRound?: number
+}
+
 export type NotebookImage = {
   name?: string
   type?: NotebookType
@@ -173,6 +196,9 @@ export class NotebookService {
   static NotebookDoAction(req: NotebookActionRequest, initReq?: fm.InitReq): Promise<Notebook> {
     return fm.fetchReq<NotebookActionRequest, Notebook>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}/actions`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }
+  static CreateNotebookSnapshotTask(req: CreateNotebookSnapshotTaskRequest, initReq?: fm.InitReq): Promise<SnapshotTask> {
+    return fm.fetchReq<CreateNotebookSnapshotTaskRequest, SnapshotTask>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}/snapshot`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
   static DeleteNotebook(req: SingleNotebookRequest, initReq?: fm.InitReq): Promise<Notebook> {
     return fm.fetchReq<SingleNotebookRequest, Notebook>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
   }

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

@@ -23,6 +23,27 @@ export enum QueueStrategy {
   BEST_EFFORT_FIFO = "BEST_EFFORT_FIFO",
 }
 
+export enum QueuePreemptionWithinClusterQueuePolicy {
+  QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_UNSPECIFIED = "QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_UNSPECIFIED",
+  QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_NEVER = "QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_NEVER",
+  QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_LOWER_PRIORITY = "QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_LOWER_PRIORITY",
+  QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_LOWER_OR_NEWER_EQUALPRIORITY = "QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_LOWER_OR_NEWER_EQUALPRIORITY",
+  QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_ANY = "QUEUE_PREEMPTION_WITHIN_CLUSTER_QUEUE_POLICY_ANY",
+}
+
+export enum QueuePreemptionReclaimWithinCohortPolicy {
+  QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_UNSPECIFIED = "QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_UNSPECIFIED",
+  QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_NEVER = "QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_NEVER",
+  QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_ANY = "QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_ANY",
+  QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_LOWER_PRIORITY = "QUEUE_PREEMPTION_RECLAIM_WITHIN_COHORT_POLICY_LOWER_PRIORITY",
+}
+
+export enum QueuePreemptionBorrowWithinCohortPolicy {
+  QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_UNSPECIFIED = "QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_UNSPECIFIED",
+  QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_NEVER = "QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_NEVER",
+  QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_LOWER_PRIORITY = "QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_LOWER_PRIORITY",
+}
+
 export type QueueResource = {
   name?: string
   value?: string
@@ -74,6 +95,7 @@ export type CreateQueueRequest = {
   strategy?: QueueStrategy
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  preemptionWithinClusterQueue?: QueuePreemptionWithinClusterQueuePolicy
 }
 
 export type UpdateQueueRequest = {
@@ -86,6 +108,7 @@ export type UpdateQueueRequest = {
   strategy?: QueueStrategy
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  preemptionWithinClusterQueue?: QueuePreemptionWithinClusterQueuePolicy
 }
 
 export type GetQueueRequest = {

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

@@ -216,7 +216,7 @@ export class InferenceServingManagement {
     return fm.fetchReq<SingleInferenceServingRequest, InferenceServing>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/inference-serving/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
   static ListInferenceServings(req: ListInferenceServingsRequest, initReq?: fm.InitReq): Promise<ListInferenceServingsResponse> {
-    return fm.fetchReq<ListInferenceServingsRequest, ListInferenceServingsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/inference-serving?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
+    return fm.fetchReq<ListInferenceServingsRequest, ListInferenceServingsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/inference-serving?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }
   static UpdateInferenceServing(req: UpdateInferenceServingRequest, initReq?: fm.InitReq): Promise<InferenceServing> {
     return fm.fetchReq<UpdateInferenceServingRequest, InferenceServing>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/inference-serving/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})

package/package.json

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