@daocloud-proto/baize 0.110.5 → 0.111.1

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",
@@ -119,6 +126,14 @@ export type DataSourceOptionsHuggingFace = {
   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
@@ -221,10 +236,11 @@ export type SingleDatasetRequest = {
   secretOptions?: SecretOptions
   share?: boolean
   sharedStoWorkspaces?: number[]
+  uri?: string
 }
 
 export type UpdateDatasetRequest = BaseUpdateDatasetRequest
-  & OneOf<{ conda: DataSourceOptionsConda }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope }>
 
 export type DatasetActionRequest = {
   workspace?: number
@@ -265,6 +281,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 +319,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,8 @@ export type Model = {
   tags?: string[]
   readme?: string
   status?: ModelStatus
+  displayName?: string
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
 }
 
 export type ListModelsRequest = {
@@ -145,6 +147,7 @@ export type CreateModelRequest = {
   readme?: string
   share?: boolean
   sharedToWorkspaces?: Workspace[]
+  displayName?: string
 }
 
 export type UpdateModelRequest = {
@@ -162,6 +165,7 @@ export type UpdateModelRequest = {
   readme?: string
   share?: boolean
   sharedToWorkspaces?: Workspace[]
+  displayName?: string
 }
 
 export type CreateModelVersionRequest = {

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

@@ -5,7 +5,9 @@
 */
 
 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 QueueType {
   QUEUE_TYPE_UNSPECIFIED = "QUEUE_TYPE_UNSPECIFIED",
@@ -44,6 +46,12 @@ export enum QueuePreemptionBorrowWithinCohortPolicy {
   QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_LOWER_PRIORITY = "QUEUE_PREEMPTION_BORROW_WITHIN_COHORT_POLICY_LOWER_PRIORITY",
 }
 
+export enum QueueWorkloadPhase {
+  PHASE_UNSPECIFIED = "PHASE_UNSPECIFIED",
+  PENDING = "PENDING",
+  ADMITTED = "ADMITTED",
+}
+
 export type QueueResource = {
   name?: string
   value?: string
@@ -153,6 +161,34 @@ export type CreateLocalQueueResponse = {
   status?: QueueStatus
 }
 
+export type QueueWorkload = {
+  cluster?: string
+  namespace?: string
+  queueName?: string
+  name?: string
+  totalPods?: number
+  totalResources?: BaizeCommonK8s.Resources
+  priorityClassName?: string
+  priority?: number
+  schedulerName?: string
+  preemptionPolicy?: string
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  phase?: QueueWorkloadPhase
+}
+
+export type ListQueueWorkloadRequest = {
+  cluster?: string
+  namespace?: string
+  queueName?: string
+  type?: QueueType
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListQueueWorkloadResponse = {
+  items?: QueueWorkload[]
+  page?: BaizeCommonCommon.Pagination
+}
+
 export class QueueManagement {
   static ListQueues(req: ListQueueRequest, initReq?: fm.InitReq): Promise<ListQueueResponse> {
     return fm.fetchReq<ListQueueRequest, ListQueueResponse>(`/apis/baize.io/v1alpha1/queues?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -178,4 +214,7 @@ export class QueueManagement {
   static CreateLocalQueue(req: CreateLocalQueueRequest, initReq?: fm.InitReq): Promise<CreateLocalQueueResponse> {
     return fm.fetchReq<CreateLocalQueueRequest, CreateLocalQueueResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/localqueues`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }
+  static ListQueueWorkloads(req: ListQueueWorkloadRequest, initReq?: fm.InitReq): Promise<ListQueueWorkloadResponse> {
+    return fm.fetchReq<ListQueueWorkloadRequest, ListQueueWorkloadResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues/${req["queueName"]}/workloads?${fm.renderURLSearchParams(req, ["cluster", "queueName"])}`, {...initReq, method: "GET"})
+  }
 }

package/package.json

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