@daocloud-proto/baize 0.109.0 → 0.110.1

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

@@ -7,6 +7,7 @@
 import * as BaizeCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+import * as BaizeManagement_apiWorkspaceV1alpha1Workspace from "../../workspace/v1alpha1/workspace.pb"
 
 type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
 type OneOf<T> =
@@ -24,6 +25,8 @@ export enum DataSourceType {
   HTTP = "HTTP",
   PVC = "PVC",
   NFS = "NFS",
+  REFERENCE = "REFERENCE",
+  HUGGING_FACE = "HUGGING_FACE",
   CONDA = "CONDA",
 }
 
@@ -41,6 +44,18 @@ export enum DataSourceOptionsCondaPackageManager {
   CONDA = "CONDA",
 }
 
+export enum DataSourceOptionsHuggingFaceRepoType {
+  REPO_TYPE_UNSPECIFIED = "REPO_TYPE_UNSPECIFIED",
+  MODEL = "MODEL",
+  DATASET = "DATASET",
+}
+
+export enum DataSourceOptionsS3Provider {
+  PROVIDER_UNSPECIFIED = "PROVIDER_UNSPECIFIED",
+  AWS = "AWS",
+  MINIO = "MINIO",
+}
+
 export enum SecretOptionsAuthType {
   AUTH_TYPE_UNSPECIFIED = "AUTH_TYPE_UNSPECIFIED",
   BASIC = "BASIC",
@@ -83,6 +98,8 @@ export type Dataset = {
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  share?: boolean
+  sharedStoWorkspaces?: BaizeManagement_apiWorkspaceV1alpha1Workspace.Workspace[]
 }
 
 export type DataSourceOptionsConda = {
@@ -95,6 +112,13 @@ export type DataSourceOptionsConda = {
   gpuType?: string
 }
 
+export type DataSourceOptionsHuggingFace = {
+  repoType?: string
+  endpoint?: string
+  include?: string
+  exclude?: string
+}
+
 export type DataSourceOptionsGit = {
   branch?: string
   commit?: string
@@ -105,6 +129,7 @@ export type DataSourceOptionsGit = {
 export type DataSourceOptionsS3 = {
   region?: string
   endpoint?: string
+  provider?: DataSourceOptionsS3Provider
 }
 
 export type DataSourceOptionsHttp = {
@@ -116,7 +141,7 @@ export type DataSourceOptionsHttp = {
 }
 
 export type DataSourceOptions = BaseDataSourceOptions
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace }>
 
 export type SecretOptionsBasicAuth = {
   username?: string
@@ -192,6 +217,8 @@ export type SingleDatasetRequest = {
   namespace?: string
   name?: string
   secretOptions?: SecretOptions
+  share?: boolean
+  sharedStoWorkspaces?: number[]
 }
 
 export type UpdateDatasetRequest = BaseUpdateDatasetRequest

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

@@ -0,0 +1,85 @@
+/* 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 BaizeCommonK8s from "../../../common/k8s.pb"
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum LabelingInstanceType {
+  LABELING_INSTANCE_TYPE_UNSPECIFIED = "LABELING_INSTANCE_TYPE_UNSPECIFIED",
+  LABEL_STUDIO = "LABEL_STUDIO",
+}
+
+export enum Phase {
+  PHASE_UNSPECIFIED = "PHASE_UNSPECIFIED",
+  PENDING = "PENDING",
+  READY = "READY",
+}
+
+export type AccessURLs = {
+  home?: string
+  exportData?: string
+}
+
+export type LabelingInstance = {
+  cluster?: string
+  namespace?: string
+  name?: string
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  podConfig?: BaizeCommonK8s.PodConfig
+  type?: LabelingInstanceType
+  description?: string
+  labelingConfig?: string
+  phase?: Phase
+  accessUrls?: AccessURLs
+}
+
+export type CreateLabelingInstanceRequest = {
+  cluster?: string
+  namespace?: string
+  name?: string
+  workspace?: number
+  podConfig?: BaizeCommonK8s.PodConfig
+  type?: LabelingInstanceType
+  description?: string
+  uiBaseUrl?: string
+  labelingConfig?: string
+}
+
+export type ListLabelingInstanceRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListLabelingInstanceResponse = {
+  items?: LabelingInstance[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type SingleLabelingInstanceRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export class LabelingManagement {
+  static CreateLabelingInstance(req: CreateLabelingInstanceRequest, initReq?: fm.InitReq): Promise<LabelingInstance> {
+    return fm.fetchReq<CreateLabelingInstanceRequest, LabelingInstance>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/labeling-instances`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListLabelingInstance(req: ListLabelingInstanceRequest, initReq?: fm.InitReq): Promise<ListLabelingInstanceResponse> {
+    return fm.fetchReq<ListLabelingInstanceRequest, ListLabelingInstanceResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/labeling-instances?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static GetLabelingInstance(req: SingleLabelingInstanceRequest, initReq?: fm.InitReq): Promise<LabelingInstance> {
+    return fm.fetchReq<SingleLabelingInstanceRequest, LabelingInstance>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/labeling-instances/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteLabelingInstance(req: SingleLabelingInstanceRequest, initReq?: fm.InitReq): Promise<LabelingInstance> {
+    return fm.fetchReq<SingleLabelingInstanceRequest, LabelingInstance>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/labeling-instances/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
+}

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

@@ -213,6 +213,19 @@ export type NotebookImageListResponse = {
   items?: NotebookImage[]
 }
 
+export type NotebookConnectionInfoSSH = {
+  host?: string
+  port?: number
+  username?: string
+}
+
+export type NotebookConnectionInfo = {
+  cluster?: string
+  namespace?: string
+  name?: string
+  ssh?: NotebookConnectionInfoSSH
+}
+
 export class NotebookService {
   static ListNotebooks(req: ListNotebooksRequest, initReq?: fm.InitReq): Promise<ListNotebooksResponse> {
     return fm.fetchReq<ListNotebooksRequest, ListNotebooksResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/notebooks?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
@@ -220,6 +233,9 @@ export class NotebookService {
   static GetNotebook(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"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
+  static GetNotebookConnectionInfo(req: SingleNotebookRequest, initReq?: fm.InitReq): Promise<NotebookConnectionInfo> {
+    return fm.fetchReq<SingleNotebookRequest, NotebookConnectionInfo>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks/${req["name"]}/connections?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
   static CreateNotebook(req: CreateNotebookRequest, initReq?: fm.InitReq): Promise<Notebook> {
     return fm.fetchReq<CreateNotebookRequest, Notebook>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/notebooks`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }

package/package.json

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