@daocloud-proto/baize 0.102.5 → 0.103.0

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

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/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
@@ -111,6 +127,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.0",
   "description": "",
   "author": "",
   "license": "ISC"