@daocloud-proto/baize 0.103.0 → 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

@@ -89,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"})
@@ -113,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/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,8 @@ export type CreateQueueRequest = {
   description?: string
   resources?: QueueResource[]
   strategy?: QueueStrategy
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
 }
 
 export type UpdateQueueRequest = {
@@ -76,6 +80,8 @@ export type UpdateQueueRequest = {
   description?: string
   resources?: QueueResource[]
   strategy?: QueueStrategy
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
 }
 
 export type GetQueueRequest = {

package/package.json

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