@daocloud-proto/baize 0.102.5 → 0.103.1

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

@@ -49,6 +49,11 @@ export enum DatasetBoundPVCAccessMode {
   READ_WRITE_MANY = "READ_WRITE_MANY",
 }
 
+export enum DatasetActionRequestAction {
+  ACTION_UNSPECIFIED = "ACTION_UNSPECIFIED",
+  RERUN = "RERUN",
+}
+
 export type DataSource = {
   type?: DataSourceType
   uri?: string
@@ -163,6 +168,22 @@ export type SingleDatasetRequest = {
   name?: string
 }
 
+export type UpdateDatasetRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  secretOptions?: SecretOptions
+}
+
+export type DatasetActionRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  action?: DatasetActionRequestAction
+}
+
 export class DatasetManagement {
   static ListDatasets(req: ListDatasetRequest, initReq?: fm.InitReq): Promise<ListDatasetResponse> {
     return fm.fetchReq<ListDatasetRequest, ListDatasetResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
@@ -176,4 +197,10 @@ export class DatasetManagement {
   static DeleteDataset(req: SingleDatasetRequest, initReq?: fm.InitReq): Promise<Dataset> {
     return fm.fetchReq<SingleDatasetRequest, Dataset>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}`, {...initReq, method: "DELETE"})
   }
+  static UpdateDataset(req: UpdateDatasetRequest, initReq?: fm.InitReq): Promise<Dataset> {
+    return fm.fetchReq<UpdateDatasetRequest, Dataset>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DatasetDoAction(req: DatasetActionRequest, initReq?: fm.InitReq): Promise<Dataset> {
+    return fm.fetchReq<DatasetActionRequest, Dataset>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/datasets/${req["name"]}/actions`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
 }

package/management-api/job/v1alpha1/common.pb.ts

@@ -23,6 +23,7 @@ export enum JobPhase {
   RUNNING = "RUNNING",
   FAILED = "FAILED",
   SUCCEEDED = "SUCCEEDED",
+  SUSPENDED = "SUSPENDED",
 }
 
 export type CommonConfig = {

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

@@ -7,6 +7,7 @@
 import * as BaizeCommonCommon from "../../../common/common.pb"
 import * as BaizeCommonK8s from "../../../common/k8s.pb"
 import * as fm from "../../../fetch.pb"
+import * as BaizeManagement_apiJobV1alpha1Paddle from "./paddle.pb"
 import * as BaizeManagement_apiJobV1alpha1Pytorch from "./pytorch.pb"
 import * as BaizeManagement_apiJobV1alpha1Tfjob from "./tfjob.pb"
 
@@ -23,6 +24,7 @@ export enum JobType {
   JOB_TYPE_UNSPECIFIED = "JOB_TYPE_UNSPECIFIED",
   PYTORCH = "PYTORCH",
   TENSORFLOW = "TENSORFLOW",
+  PADDLE = "PADDLE",
 }
 
 export enum JobActionRequestAction {
@@ -40,7 +42,7 @@ export enum JobActionRequestAction {
 }
 
 export type Job = BaseJob
-  & OneOf<{ pytorch: BaizeManagement_apiJobV1alpha1Pytorch.PyTorchJob; tensorflow: BaizeManagement_apiJobV1alpha1Tfjob.TFJob }>
+  & OneOf<{ pytorch: BaizeManagement_apiJobV1alpha1Pytorch.PyTorchJob; tensorflow: BaizeManagement_apiJobV1alpha1Tfjob.TFJob; paddle: BaizeManagement_apiJobV1alpha1Paddle.PaddleJob }>
 
 export type ListJobsRequest = {
   workspace?: number
@@ -76,7 +78,7 @@ export type SingleJobRequest = {
 }
 
 export type CreateJobRequest = BaseCreateJobRequest
-  & OneOf<{ pytorch: BaizeManagement_apiJobV1alpha1Pytorch.PytorchJobCreationConfig; tensorflow: BaizeManagement_apiJobV1alpha1Tfjob.TFJobCreationConfig }>
+  & OneOf<{ pytorch: BaizeManagement_apiJobV1alpha1Pytorch.PytorchJobCreationConfig; tensorflow: BaizeManagement_apiJobV1alpha1Tfjob.TFJobCreationConfig; paddle: BaizeManagement_apiJobV1alpha1Paddle.PaddleJobCreationConfig }>
 
 export type JobActionRequest = {
   workspace?: number
@@ -87,23 +89,6 @@ export type JobActionRequest = {
   action?: JobActionRequestAction
 }
 
-export type JobInstanceContainer = {
-  name?: string
-}
-
-export type JobInstance = {
-  cluster?: string
-  namespace?: string
-  name?: string
-  phase?: string
-  containers?: JobInstanceContainer[]
-}
-
-export type JobInstanceListResponse = {
-  items?: JobInstance[]
-  page?: BaizeCommonCommon.Pagination
-}
-
 export class JobsManagement {
   static ListJobs(req: ListJobsRequest, initReq?: fm.InitReq): Promise<ListJobsResponse> {
     return fm.fetchReq<ListJobsRequest, ListJobsResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/jobs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
@@ -111,9 +96,6 @@ export class JobsManagement {
   static GetJob(req: SingleJobRequest, initReq?: fm.InitReq): Promise<Job> {
     return fm.fetchReq<SingleJobRequest, Job>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
-  static GetJobInstanceList(req: SingleJobRequest, initReq?: fm.InitReq): Promise<JobInstanceListResponse> {
-    return fm.fetchReq<SingleJobRequest, JobInstanceListResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/instances?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
-  }
   static CreateJob(req: CreateJobRequest, initReq?: fm.InitReq): Promise<Job> {
     return fm.fetchReq<CreateJobRequest, Job>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }

package/management-api/job/v1alpha1/paddle.pb.ts

@@ -0,0 +1,34 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as GoogleProtobufStruct from "../../../google/protobuf/struct.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+import * as BaizeManagement_apiJobV1alpha1Common from "./common.pb"
+
+export enum PaddleRole {
+  PADDLE_ROLE_UNSPECIFIED = "PADDLE_ROLE_UNSPECIFIED",
+  PD_MASTER = "PD_MASTER",
+  PD_WORKER = "PD_WORKER",
+}
+
+export type PaddleJob = {
+  cluster?: string
+  namespace?: string
+  name?: string
+  trainingMode?: BaizeManagement_apiJobV1alpha1Common.TrainingMode
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  phase?: BaizeManagement_apiJobV1alpha1Common.JobPhase
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+  jobSpec?: GoogleProtobufStruct.Struct
+  description?: string
+}
+
+export type PaddleJobCreationConfig = {
+  trainingMode?: BaizeManagement_apiJobV1alpha1Common.TrainingMode
+  baseConfig?: BaizeManagement_apiJobV1alpha1Common.JobCreationBaseConfig
+  roleConfig?: {[key: string]: BaizeManagement_apiJobV1alpha1Common.JobRoleDifferenceConfig}
+}

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

@@ -139,6 +139,24 @@ export type QueryTimeSeriesVectorsResponse = {
   items?: TimeSeriesVectorQueryResponse[]
 }
 
+export type GrafanaDashboard = {
+  urlEN?: string
+  urlZH?: string
+}
+
+export type DevOverviewDashboardRequest = {
+  workspace?: number
+  cluster?: string
+  from?: string
+  to?: string
+}
+
+export type AdminOverviewDashboardRequest = {
+  cluster?: 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"})
@@ -149,4 +167,10 @@ export class MetricsService {
   static QueryDevDashboardMultipleVectors(req: QueryDevDashboardMultipleVectorRequest, initReq?: fm.InitReq): Promise<MultipleDevDashboardVectorQueryResponse> {
     return fm.fetchReq<QueryDevDashboardMultipleVectorRequest, MultipleDevDashboardVectorQueryResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/metrics/dev-dashboard-multiple-queries?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }
+  static GetDevGrafanaDashboard(req: DevOverviewDashboardRequest, initReq?: fm.InitReq): Promise<GrafanaDashboard> {
+    return fm.fetchReq<DevOverviewDashboardRequest, GrafanaDashboard>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/grafana-dashboards/dev-overview?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static GetAdminGrafanaDashboard(req: AdminOverviewDashboardRequest, initReq?: fm.InitReq): Promise<GrafanaDashboard> {
+    return fm.fetchReq<AdminOverviewDashboardRequest, GrafanaDashboard>(`/apis/baize.io/v1alpha1/grafana-dashboards/admin-overview?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
 }

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

@@ -74,6 +74,7 @@ export type Notebook = {
   config?: NotebookConfig
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   status?: NotebookStatus
+  priorityClass?: string
 }
 
 export type ListNotebooksResponse = {
@@ -96,6 +97,7 @@ export type CreateNotebookRequest = {
   queueName?: string
   type?: NotebookType
   config?: NotebookConfig
+  priorityClass?: string
 }
 
 export type UpdateNotebookRequest = {
@@ -104,6 +106,7 @@ export type UpdateNotebookRequest = {
   namespace?: string
   name?: string
   config?: NotebookConfig
+  priorityClass?: string
 }
 
 export type NotebookActionRequest = {

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

@@ -0,0 +1,55 @@
+/* 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 ResourceType {
+  JOB_TYPE_UNSPECIFIED = "JOB_TYPE_UNSPECIFIED",
+  PYTORCH = "PYTORCH",
+  TENSORFLOW = "TENSORFLOW",
+  PADDLE = "PADDLE",
+  NOTEBOOK = "NOTEBOOK",
+}
+
+export type PodRequest = {
+  workspace?: number
+  type?: ResourceType
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type PodInstanceContainer = {
+  name?: string
+}
+
+export type PodInstance = {
+  cluster?: string
+  namespace?: string
+  name?: string
+  phase?: string
+  containerReadyCount?: number
+  containerTotalCount?: number
+  podIp?: string
+  nodeName?: string
+  resources?: BaizeCommonK8s.Resources
+  containers?: PodInstanceContainer[]
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type PodInstanceListResponse = {
+  items?: PodInstance[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export class podsManagement {
+  static GetPodInstanceList(req: PodRequest, initReq?: fm.InitReq): Promise<PodInstanceListResponse> {
+    return fm.fetchReq<PodRequest, PodInstanceListResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/resources/${req["name"]}/instances?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+}

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

@@ -41,6 +41,8 @@ export type Queue = {
   description?: string
   resources?: QueueResource[]
   resourcesAvailable?: QueueResource[]
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
   status?: QueueStatus
   strategy?: QueueStrategy
   workspaceAlias?: string
@@ -66,6 +68,26 @@ export type CreateQueueRequest = {
   description?: string
   resources?: QueueResource[]
   strategy?: QueueStrategy
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+}
+
+export type UpdateQueueRequest = {
+  type?: QueueType
+  cluster?: string
+  name?: string
+  workspace?: number
+  description?: string
+  resources?: QueueResource[]
+  strategy?: QueueStrategy
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+}
+
+export type GetQueueRequest = {
+  type?: QueueType
+  cluster?: string
+  name?: string
 }
 
 export type DeleteQueueRequest = {
@@ -111,6 +133,12 @@ export class QueueManagement {
   static CreateQueue(req: CreateQueueRequest, initReq?: fm.InitReq): Promise<Queue> {
     return fm.fetchReq<CreateQueueRequest, Queue>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }
+  static UpdateQueue(req: UpdateQueueRequest, initReq?: fm.InitReq): Promise<Queue> {
+    return fm.fetchReq<UpdateQueueRequest, Queue>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetQueue(req: GetQueueRequest, initReq?: fm.InitReq): Promise<Queue> {
+    return fm.fetchReq<GetQueueRequest, Queue>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues/${req["name"]}?${fm.renderURLSearchParams(req, ["cluster", "name"])}`, {...initReq, method: "GET"})
+  }
   static DeleteQueue(req: DeleteQueueRequest, initReq?: fm.InitReq): Promise<Queue> {
     return fm.fetchReq<DeleteQueueRequest, Queue>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
   }

package/package.json

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