@daocloud-proto/baize 0.110.4 → 0.111.0

package/common/common.pb.ts

@@ -31,13 +31,22 @@ export type Auth = {
   operator?: AuthResourceAction
 }
 
+export type AuditObject = {
+  nameKey?: string
+  namespaceKey?: string
+  clusterKey?: string
+}
+
+export type AuditObjectKind = {
+  kindName?: string
+  kindGroup?: string
+  kindVersion?: string
+  kindKey?: string
+}
+
 export type Audit = {
-  objectNsKey?: string
-  objectNameKey?: string
-  objectKindName?: string
-  objectKindGroup?: string
-  objectKindVersion?: string
-  objectKindKey?: string
+  object?: AuditObject
+  objectKind?: AuditObjectKind
   action?: string
   actionKey?: string
   customObject?: boolean

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

@@ -27,6 +27,7 @@ export enum DataSourceType {
   NFS = "NFS",
   REFERENCE = "REFERENCE",
   HUGGING_FACE = "HUGGING_FACE",
+  MODEL_SCOPE = "MODEL_SCOPE",
   CONDA = "CONDA",
 }
 
@@ -50,6 +51,12 @@ export enum DataSourceOptionsHuggingFaceRepoType {
   DATASET = "DATASET",
 }
 
+export enum DataSourceOptionsModelScopeRepoType {
+  REPO_TYPE_UNSPECIFIED = "REPO_TYPE_UNSPECIFIED",
+  MODEL = "MODEL",
+  DATASET = "DATASET",
+}
+
 export enum DataSourceOptionsS3Provider {
   PROVIDER_UNSPECIFIED = "PROVIDER_UNSPECIFIED",
   AWS = "AWS",
@@ -115,10 +122,18 @@ export type DataSourceOptionsConda = {
 }
 
 export type DataSourceOptionsHuggingFace = {
-  repoType?: string
+  repoType?: DataSourceOptionsHuggingFaceRepoType
   endpoint?: string
   include?: string
   exclude?: string
+  revision?: string
+}
+
+export type DataSourceOptionsModelScope = {
+  repoType?: DataSourceOptionsModelScopeRepoType
+  include?: string
+  exclude?: string
+  revision?: string
 }
 
 export type DataSourceOptionsGit = {
@@ -143,7 +158,7 @@ export type DataSourceOptionsHttp = {
 }
 
 export type DataSourceOptions = BaseDataSourceOptions
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope }>
 
 export type SecretOptionsBasicAuth = {
   username?: string
@@ -265,6 +280,16 @@ export type DatasetSyncProcess = {
   syncProcess?: number
 }
 
+export type ValidateDataSourceSecretsRequest = {
+  source?: DataSource
+  secretOptions?: SecretOptions
+}
+
+export type ValidateDataSourceSecretsResponse = {
+  valid?: boolean
+  message?: string
+}
+
 export class DatasetManagement {
   static ListDatasets(req: ListDatasetRequest, initReq?: fm.InitReq): Promise<ListDatasetResponse> {
     return fm.fetchReq<ListDatasetRequest, ListDatasetResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/datasets?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
@@ -293,4 +318,7 @@ export class DatasetManagement {
   static GetRelatedInstances(req: RelatedInstancesRequest, initReq?: fm.InitReq): Promise<RelatedInstancesResponse> {
     return fm.fetchReq<RelatedInstancesRequest, RelatedInstancesResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}/related-instances?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
+  static ValidateDataSourceSecrets(req: ValidateDataSourceSecretsRequest, initReq?: fm.InitReq): Promise<ValidateDataSourceSecretsResponse> {
+    return fm.fetchReq<ValidateDataSourceSecretsRequest, ValidateDataSourceSecretsResponse>(`/apis/baize.io/v1alpha1/datasets/validate-secrets`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
 }

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

@@ -176,6 +176,14 @@ export type AdminOverviewDashboardRequest = {
   to?: string
 }
 
+export type NvidiaGPUGrafanaDashboardRequest = {
+  cluster?: string
+  node?: string
+  gpuIndex?: string
+  from?: string
+  to?: string
+}
+
 export class MetricsService {
   static QueryMultipleVectors(req: QueryMultipleVectorRequest, initReq?: fm.InitReq): Promise<MultipleVectorQueryResponse> {
     return fm.fetchReq<QueryMultipleVectorRequest, MultipleVectorQueryResponse>(`/apis/baize.io/v1alpha1/metrics/multiple-queries?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -198,4 +206,7 @@ export class MetricsService {
   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"})
   }
+  static GetNvidiaGPUGrafanaDashboard(req: NvidiaGPUGrafanaDashboardRequest, initReq?: fm.InitReq): Promise<GrafanaDashboard> {
+    return fm.fetchReq<NvidiaGPUGrafanaDashboardRequest, GrafanaDashboard>(`/apis/baize.io/v1alpha1/grafana-dashboards/nvidia-gpu-overview?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -108,6 +108,7 @@ export type Model = {
   tags?: string[]
   readme?: string
   status?: ModelStatus
+  displayName?: string
 }
 
 export type ListModelsRequest = {
@@ -145,6 +146,7 @@ export type CreateModelRequest = {
   readme?: string
   share?: boolean
   sharedToWorkspaces?: Workspace[]
+  displayName?: string
 }
 
 export type UpdateModelRequest = {
@@ -162,6 +164,7 @@ export type UpdateModelRequest = {
   readme?: string
   share?: boolean
   sharedToWorkspaces?: Workspace[]
+  displayName?: string
 }
 
 export type CreateModelVersionRequest = {

package/package.json

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