@daocloud-proto/baize 0.108.3 → 0.110.0

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

@@ -41,6 +41,12 @@ export enum DataSourceOptionsCondaPackageManager {
   CONDA = "CONDA",
 }
 
+export enum DataSourceOptionsS3Provider {
+  PROVIDER_UNSPECIFIED = "PROVIDER_UNSPECIFIED",
+  AWS = "AWS",
+  MINIO = "MINIO",
+}
+
 export enum SecretOptionsAuthType {
   AUTH_TYPE_UNSPECIFIED = "AUTH_TYPE_UNSPECIFIED",
   BASIC = "BASIC",
@@ -105,6 +111,7 @@ export type DataSourceOptionsGit = {
 export type DataSourceOptionsS3 = {
   region?: string
   endpoint?: string
+  provider?: DataSourceOptionsS3Provider
 }
 
 export type DataSourceOptionsHttp = {

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

@@ -0,0 +1,153 @@
+/* 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"
+
+type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
+type OneOf<T> =
+  | { [k in keyof T]?: undefined }
+  | (
+    keyof T extends infer K ?
+      (K extends string & keyof T ? { [k in K]: T[K] } & Absent<T, K>
+        : never)
+    : never);
+
+export enum Format {
+  FORMAT_UNSPECIFIED = "FORMAT_UNSPECIFIED",
+  PYTORCH = "PYTORCH",
+  TENSORFLOW = "TENSORFLOW",
+  ONNX = "ONNX",
+}
+
+export enum Architecture {
+  ARCHITECTURE_UNSPECIFIED = "ARCHITECTURE_UNSPECIFIED",
+  CNN = "CNN",
+  RNN = "RNN",
+  LSTM = "LSTM",
+  TRANSFORMER = "TRANSFORMER",
+  MLP = "MLP",
+  BERT = "BERT",
+  GAN = "GAN",
+  RESNET = "RESNET",
+  OTHER = "OTHER",
+}
+
+export enum ModelVersionSourceType {
+  VERSION_SOURCE_TYPE_UNSPECIFIED = "VERSION_SOURCE_TYPE_UNSPECIFIED",
+  DATASET = "DATASET",
+}
+
+export enum ModelStatusPhase {
+  PHASE_UNSPECIFIED = "PHASE_UNSPECIFIED",
+  PENDING = "PENDING",
+  CREATING = "CREATING",
+  READY = "READY",
+  ERROR = "ERROR",
+}
+
+export type Workspace = {
+  workspaceId?: number
+  alias?: string
+}
+
+export type DatasetRef = {
+  namespace?: string
+  name?: string
+  path?: string
+}
+
+
+/* baize modified */ export type BaseModelVersionSource = {
+  type?: ModelVersionSourceType
+}
+
+export type ModelVersionSource = BaseModelVersionSource
+  & OneOf<{ datasetRef: DatasetRef }>
+
+export type Version = {
+  version?: string
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  source?: ModelVersionSource
+}
+
+export type ModelStatus = {
+  phase?: ModelStatusPhase
+}
+
+export type Model = {
+  name?: string
+  cluster?: string
+  namespace?: string
+  share?: boolean
+  sharedToWorkspaces?: Workspace[]
+  format?: Format
+  architecture?: Architecture
+  versions?: Version[]
+  license?: string
+  isLLM?: boolean
+  latestVersion?: string
+  iconUrl?: string
+  tags?: string[]
+  readme?: string
+  status?: ModelStatus
+}
+
+export type ListModelsRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListModelsResponse = {
+  items?: Model[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type SingleModelRequest = {
+  name?: string
+  cluster?: string
+  namespace?: string
+  workspace?: number
+}
+
+export type CreateModelRequest = {
+  workspace?: number
+  name?: string
+  cluster?: string
+  namespace?: string
+  format?: Format
+  architecture?: Architecture
+  versions?: Version[]
+  license?: string
+  isLLM?: boolean
+  latestVersion?: string
+  iconUrl?: string
+  tags?: string[]
+  readme?: string
+  share?: boolean
+  sharedToWorkspaces?: Workspace[]
+}
+
+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"})
+  }
+  static ListSharedModels(req: ListModelsRequest, initReq?: fm.InitReq): Promise<ListModelsResponse> {
+    return fm.fetchReq<ListModelsRequest, ListModelsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/shared-models?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static GetModel(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"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  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 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)})
+  }
+}

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

@@ -46,6 +46,13 @@ export enum NotebookActionRequestAction {
   STOP = "STOP",
 }
 
+export enum SnapshotTaskStatus {
+  Status_UNSPECIFIED = "Status_UNSPECIFIED",
+  CREATED = "CREATED",
+  FAILED = "FAILED",
+  COMPLETED = "COMPLETED",
+}
+
 export type ListNotebooksRequest = {
   workspace?: number
   cluster?: string
@@ -157,6 +164,7 @@ export type CreateNotebookSnapshotTaskRequest = {
   cluster?: string
   namespace?: string
   name?: string
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
 }
 
 export type SnapshotTask = {
@@ -165,6 +173,31 @@ export type SnapshotTask = {
   namespace?: string
   name?: string
   triggerRound?: number
+  imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  status?: SnapshotTaskStatus
+  notebookName?: string
+}
+
+export type ListSnapshotTasksRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListSnapshotTasksResponse = {
+  items?: SnapshotTask[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ResetNotebookImageRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  snapshotTaskName?: string
 }
 
 export type NotebookImage = {
@@ -205,4 +238,10 @@ export class NotebookService {
   static GetNotebookImageList(req: NotebookImageListRequest, initReq?: fm.InitReq): Promise<NotebookImageListResponse> {
     return fm.fetchReq<NotebookImageListRequest, NotebookImageListResponse>(`/apis/baize.io/v1alpha1/notebook-images?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
+  static ListNotebookSnapshotTasks(req: ListSnapshotTasksRequest, initReq?: fm.InitReq): Promise<ListSnapshotTasksResponse> {
+    return fm.fetchReq<ListSnapshotTasksRequest, ListSnapshotTasksResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}/snapshottasks?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static ResetNotebookImage(req: ResetNotebookImageRequest, initReq?: fm.InitReq): Promise<SnapshotTask> {
+    return fm.fetchReq<ResetNotebookImageRequest, SnapshotTask>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}/image`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
 }

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

@@ -134,6 +134,7 @@ export type InferenceServing = {
   status?: InferenceServingStatus
   lastUpdated?: GoogleProtobufTimestamp.Timestamp
   authType?: ServingAuthAuthType
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
 }
 
 export type InferenceServingStatusModelStatus = {

package/package.json

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