@daocloud-proto/baize 0.107.4 → 0.108.1

package/common/k8s.pb.ts

@@ -11,6 +11,12 @@ export enum KubeVolumeVolumeType {
   DATASET_RUNTIME_ENV = "DATASET_RUNTIME_ENV",
 }
 
+export enum KubeVolumeDataDirection {
+  DATA_DIRECTION_UNSPECIFIED = "DATA_DIRECTION_UNSPECIFIED",
+  INPUT = "INPUT",
+  OUTPUT = "OUTPUT",
+}
+
 export type Resources = {
   requests?: {[key: string]: string}
   limits?: {[key: string]: string}
@@ -26,6 +32,7 @@ export type KubeVolume = {
   name?: string
   mountPath?: string
   readOnly?: boolean
+  dataDirection?: KubeVolumeDataDirection
 }
 
 export type PodConfig = {

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

@@ -0,0 +1,158 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+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",
+  REGISTRY_TYPE_DOCKER = "REGISTRY_TYPE_DOCKER",
+  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?: number
+  global?: boolean
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListRegistriesResponse = {
+  items?: Registry[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type Registry = {
+  host?: string
+  name?: string
+  type?: RegistryType
+  scope?: Scope
+}
+
+export type ListProjectsRequest = {
+  workspace?: number
+  registry?: string
+  public?: boolean
+  page?: BaizeCommonCommon.Pagination
+  onlyBind?: boolean
+  all?: boolean
+}
+
+export type ListProjectsResponse = {
+  items?: Project[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type Project = {
+  name?: string
+  visible?: Visibility
+}
+
+export type ListRepositoriesRequest = {
+  workspace?: number
+  registry?: string
+  project?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListRepositoriesResponse = {
+  items?: Repository[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type Repository = {
+  name?: string
+  updateTimestamp?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type ListArtifactsRequest = {
+  workspace?: number
+  registry?: string
+  project?: string
+  repository?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListArtifactsResponse = {
+  items?: Artifact[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type Artifact = {
+  digest?: string
+  tags?: Tag[]
+  imageSize?: string
+  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 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
+  meta?: KangarooImageMetadata
+  registryName?: string
+}
+
+export class ImageService {
+  static ListRegistries(req: ListRegistriesRequest, initReq?: fm.InitReq): Promise<ListRegistriesResponse> {
+    return fm.fetchReq<ListRegistriesRequest, ListRegistriesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/registries?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static ListProjects(req: ListProjectsRequest, initReq?: fm.InitReq): Promise<ListProjectsResponse> {
+    return fm.fetchReq<ListProjectsRequest, ListProjectsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/registries/${req["registry"]}/projects?${fm.renderURLSearchParams(req, ["workspace", "registry"])}`, {...initReq, method: "GET"})
+  }
+  static ListProjectRepositories(req: ListRepositoriesRequest, initReq?: fm.InitReq): Promise<ListRepositoriesResponse> {
+    return fm.fetchReq<ListRepositoriesRequest, ListRepositoriesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/registries/${req["registry"]}/projects/${req["project"]}/repositories?${fm.renderURLSearchParams(req, ["workspace", "registry", "project"])}`, {...initReq, method: "GET"})
+  }
+  static ListArtifacts(req: ListArtifactsRequest, initReq?: fm.InitReq): Promise<ListArtifactsResponse> {
+    return fm.fetchReq<ListArtifactsRequest, ListArtifactsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/registries/${req["registry"]}/projects/${req["project"]}/repositories/${req["repository"]}/artifacts?${fm.renderURLSearchParams(req, ["workspace", "registry", "project", "repository"])}`, {...initReq, method: "GET"})
+  }
+}

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

@@ -5,6 +5,7 @@
 */
 
 import * as BaizeCommonK8s from "../../../common/k8s.pb"
+import * as BaizeManagement_apiImageV1alpha1Image from "../../image/v1alpha1/image.pb"
 
 export enum QueueType {
   QUEUE_TYPE_UNSPECIFIED = "QUEUE_TYPE_UNSPECIFIED",
@@ -60,6 +61,7 @@ export type JobCreationBaseConfig = {
   workingDir?: string
   shmSize?: number
   noOverrideEnvPath?: boolean
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type JobRoleDifferenceConfig = {

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

@@ -9,6 +9,7 @@ import * as BaizeCommonK8s from "../../../common/k8s.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 import * as BaizeManagement_apiAnalysisV1alpha1Analysis from "../../analysis/v1alpha1/analysis.pb"
+import * as BaizeManagement_apiImageV1alpha1Image from "../../image/v1alpha1/image.pb"
 
 type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
 type OneOf<T> =
@@ -70,6 +71,7 @@ export type NotebookConfig = {
   enableSSH?: boolean
   affinity?: BaizeCommonK8s.Affinity
   tolerationSeconds?: string
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type NotebookStatus = {
@@ -91,6 +93,7 @@ export type Notebook = {
   priorityClass?: string
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   description?: string
+  snapshotConfig?: SnapshotConfig
 }
 
 export type ListNotebooksResponse = {
@@ -120,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 = {
@@ -132,6 +141,7 @@ export type UpdateNotebookRequest = {
   priorityClass?: string
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
   description?: string
+  snapshotConfig?: SnapshotConfig
 }
 
 export type NotebookActionRequest = {
@@ -142,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
@@ -171,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/pod/v1alpha1/pod.pb.ts

@@ -14,10 +14,11 @@ export enum ResourceType {
   PYTORCH = "PYTORCH",
   TENSORFLOW = "TENSORFLOW",
   PADDLE = "PADDLE",
-  NOTEBOOK = "NOTEBOOK",
-  INFERENCE = "INFERENCE",
   MXNET = "MXNET",
   MPI = "MPI",
+  NOTEBOOK = "NOTEBOOK",
+  INFERENCE = "INFERENCE",
+  DATASET = "DATASET",
 }
 
 export type PodRequest = {

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