@daocloud-proto/baize 0.102.4 → 0.103.0

package/management-api/job/v1alpha1/job.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 BaizeManagement_apiJobV1alpha1Paddle from "./paddle.pb"
 import * as BaizeManagement_apiJobV1alpha1Pytorch from "./pytorch.pb"
 import * as BaizeManagement_apiJobV1alpha1Tfjob from "./tfjob.pb"
 
@@ -22,6 +24,7 @@ export enum JobType {
   JOB_TYPE_UNSPECIFIED = "JOB_TYPE_UNSPECIFIED",
   PYTORCH = "PYTORCH",
   TENSORFLOW = "TENSORFLOW",
+  PADDLE = "PADDLE",
 }
 
 export enum JobActionRequestAction {
@@ -35,10 +38,11 @@ export enum JobActionRequestAction {
   queueName?: string
   priorityClass?: string
   runningDuration?: number
+  totalResources?: BaizeCommonK8s.Resources
 }
 
 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
@@ -74,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

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

@@ -101,6 +101,10 @@ export type DevDashboardVectorQueryResponse = {
   metric?: DevDashboardQueryMetric
 }
 
+export type MultipleDevDashboardVectorQueryResponse = {
+  items?: DevDashboardVectorQueryResponse[]
+}
+
 export type TimeSeriesVectorQueryResponse = {
   matrix?: Matrix
   error?: string
@@ -118,16 +122,13 @@ export type QueryDevDashboardMultipleVectorRequest = {
   queryMetrics?: DevDashboardQueryMetric[]
   time?: string
   workspace?: number
+  cluster?: string
 }
 
 export type MultipleVectorQueryResponse = {
   items?: VectorQueryResponse[]
 }
 
-export type MultipleDevDashboardVectorQueryResponse = {
-  items?: VectorQueryResponse[]
-}
-
 export type QueryTimeSeriesVectorsRequest = {
   queryMetrics?: QueryTimeSeriesMetric[]
   queryMetricRange?: QueryTimeSeriesMetricRange
@@ -138,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"})
@@ -145,7 +164,13 @@ export class MetricsService {
   static QueryTimeSeriesVectors(req: QueryTimeSeriesVectorsRequest, initReq?: fm.InitReq): Promise<QueryTimeSeriesVectorsResponse> {
     return fm.fetchReq<QueryTimeSeriesVectorsRequest, QueryTimeSeriesVectorsResponse>(`/apis/baize.io/v1alpha1/metrics/time-series-queries?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
-  static QueryDevDashboardMultipleVectors(req: QueryDevDashboardMultipleVectorRequest, initReq?: fm.InitReq): Promise<DevDashboardVectorQueryResponse> {
-    return fm.fetchReq<QueryDevDashboardMultipleVectorRequest, DevDashboardVectorQueryResponse>(`/apis/baize.io/v1alpha1/metrics/dev-dashboard-multiple-queries?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  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/queue/v1alpha1/queue.pb.ts

@@ -68,6 +68,22 @@ export type CreateQueueRequest = {
   strategy?: QueueStrategy
 }
 
+export type UpdateQueueRequest = {
+  type?: QueueType
+  cluster?: string
+  name?: string
+  workspace?: number
+  description?: string
+  resources?: QueueResource[]
+  strategy?: QueueStrategy
+}
+
+export type GetQueueRequest = {
+  type?: QueueType
+  cluster?: string
+  name?: string
+}
+
 export type DeleteQueueRequest = {
   type?: QueueType
   cluster?: string
@@ -106,11 +122,17 @@ export type CreateLocalQueueResponse = {
 
 export class QueueManagement {
   static ListQueues(req: ListQueueRequest, initReq?: fm.InitReq): Promise<ListQueueResponse> {
-    return fm.fetchReq<ListQueueRequest, ListQueueResponse>(`/apis/baize.io/v1alpha1/clusters/${req["cluster"]}/queues?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
+    return fm.fetchReq<ListQueueRequest, ListQueueResponse>(`/apis/baize.io/v1alpha1/queues?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
   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.4",
+  "version": "v0.103.0",
   "description": "",
   "author": "",
   "license": "ISC"