@daocloud-proto/baize 0.110.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,12 @@ 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",
@@ -89,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 = {
@@ -101,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
@@ -123,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
@@ -199,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/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.110.0",
+  "version": "v0.110.1",
   "description": "",
   "author": "",
   "license": "ISC"