@daocloud-proto/baize 0.126.0 → 0.127.1

package/common/common.pb.ts

@@ -24,6 +24,12 @@ export enum TolerationOperator {
   Equal = "Equal",
 }
 
+export enum APIVersion {
+  API_VERSION_UNSPECIFIED = "API_VERSION_UNSPECIFIED",
+  API_V1_ALPHA1 = "API_V1_ALPHA1",
+  API_V1_ALPHA2 = "API_V1_ALPHA2",
+}
+
 export type NoParamsQuery = {
 }
 

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

@@ -32,6 +32,7 @@ export enum DataSourceType {
   MODEL_SCOPE = "MODEL_SCOPE",
   CONDA = "CONDA",
   DATABASE = "DATABASE",
+  HADOOP = "HADOOP",
 }
 
 export enum DatasetStatusPhase {
@@ -195,12 +196,19 @@ export type DataSourceOptionsDatabase = {
   extraParams?: {[key: string]: string}
 }
 
+export type DataSourceOptionsHadoop = {
+  coreSiteXml?: string
+  hdfsSiteXml?: string
+  sourcePath?: string
+  username?: string
+}
+
 
 /* baize modified */ export type BaseDataSourceOptions = {
 }
 
 export type DataSourceOptions = BaseDataSourceOptions
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference; database: DataSourceOptionsDatabase }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference; database: DataSourceOptionsDatabase; hadoop: DataSourceOptionsHadoop }>
 
 export type SecretOptionsBasicAuth = {
   username?: string
@@ -288,7 +296,7 @@ export type SingleDatasetRequest = {
 }
 
 export type UpdateDatasetRequest = BaseUpdateDatasetRequest
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference; database: DataSourceOptionsDatabase }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda; huggingFace: DataSourceOptionsHuggingFace; modelScope: DataSourceOptionsModelScope; reference: DataSourceOptionsReference; database: DataSourceOptionsDatabase; hadoop: DataSourceOptionsHadoop }>
 
 export type DatasetActionRequest = {
   workspace?: number

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

@@ -37,6 +37,11 @@ export enum RestartPolicy {
   RESTART_POLICY_ON_FAILURE = "RESTART_POLICY_ON_FAILURE",
 }
 
+export enum SuspendReason {
+  SUSPEND_REASON_UNSPECIFIED = "SUSPEND_REASON_UNSPECIFIED",
+  SUSPEND_REASON_USER_ACTION = "SUSPEND_REASON_USER_ACTION",
+}
+
 export enum ImagePullPolicy {
   IMAGE_PULL_POLICY_UNSPECIFIED = "IMAGE_PULL_POLICY_UNSPECIFIED",
   IMAGE_PULL_POLICY_ALWAYS = "IMAGE_PULL_POLICY_ALWAYS",
@@ -49,6 +54,7 @@ export enum JobActionRequestAction {
   RESTART = "RESTART",
   CHANGE_PRIORITY = "CHANGE_PRIORITY",
   SUSPEND = "SUSPEND",
+  RESUME = "RESUME",
 }
 
 export enum JobSchedulersResponseSchedulerFeature {
@@ -77,6 +83,8 @@ export type Job = {
   recoveryConfig?: BaizeManagement_apiJobV1alpha1Common.RecoveryConfig
   cleanCheckpointConfig?: CleanCheckpointConfig
   imagePullPolicy?: ImagePullPolicy
+  suspendReason?: SuspendReason
+  uid?: string
 }
 
 export type ListJobsRequest = {

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

@@ -6,14 +6,93 @@
 
 import * as BaizeCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
-export type ListJobsRequest = {
+import * as BaizeManagement_apiEventV1alpha1Event from "../../event/v1alpha1/event.pb"
+import * as BaizeManagement_apiJobV1alpha1Common from "../v1alpha1/common.pb"
+import * as BaizeManagement_apiJobV1alpha1Job from "../v1alpha1/job.pb"
+export type Trainer = {
+  command?: string[]
+  args?: string[]
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+  queue?: string
+  trainingMode?: BaizeManagement_apiJobV1alpha1Common.TrainingMode
+  imagePullPolicy?: BaizeManagement_apiJobV1alpha1Job.ImagePullPolicy
+  checkpointConfig?: BaizeManagement_apiJobV1alpha1Job.CleanCheckpointConfig
+}
+
+export type RuntimeRef = {
+  name?: string
+  apiGroup?: string
+  kind?: string
+}
+
+export type DatasetInitializer = {
+}
+
+export type ModelInitializer = {
+}
+
+export type Initializer = {
+  model?: ModelInitializer
+  dataset?: DatasetInitializer
+}
+
+export type CreateJobRequest = {
   workspace?: number
   cluster?: string
   namespace?: string
-  page?: BaizeCommonCommon.Pagination
+  name?: string
+  runtimeRef?: RuntimeRef
+  initializer?: Initializer
+  trainer?: Trainer
+  suspend?: boolean
+  description?: string
 }
 
 export type Job = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+  runtimeRef?: RuntimeRef
+  initializer?: Initializer
+  trainer?: Trainer
+  suspend?: boolean
+  description?: string
+}
+
+export type GetJobRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type DeleteJobRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type ListRuntimesRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListClusterRuntimesRequest = {
+  workspace?: number
+  cluster?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListJobsRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: BaizeCommonCommon.Pagination
 }
 
 export type ListJobsResponse = {
@@ -21,8 +100,106 @@ export type ListJobsResponse = {
   page?: BaizeCommonCommon.Pagination
 }
 
+export type CreateRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type Runtime = {
+  name?: string
+}
+
+export type ListRuntimesResponse = {
+  items?: Runtime[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type ListClusterRuntimesResponse = {
+  items?: Runtime[]
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type GetRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type GetClusterRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+}
+
+export type DeleteRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type DeleteClusterRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+}
+
+export type CreateClusterRuntimeRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+}
+
 export class JobService {
-  static ListJobs(req: ListJobsRequest, initReq?: fm.InitReq): Promise<ListJobsResponse> {
-    return fm.fetchReq<ListJobsRequest, ListJobsResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/jobs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  static CreateJob(req: BaizeManagement_apiJobV1alpha1Job.CreateJobRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.Job> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.CreateJobRequest, BaizeManagement_apiJobV1alpha1Job.Job>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetJob(req: BaizeManagement_apiJobV1alpha1Job.SingleJobRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.Job> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.SingleJobRequest, BaizeManagement_apiJobV1alpha1Job.Job>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteJob(req: BaizeManagement_apiJobV1alpha1Job.SingleJobRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.Job> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.SingleJobRequest, BaizeManagement_apiJobV1alpha1Job.Job>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListJobs(req: BaizeManagement_apiJobV1alpha1Job.ListJobsRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.ListJobsResponse> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.ListJobsRequest, BaizeManagement_apiJobV1alpha1Job.ListJobsResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/jobs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static CreateRuntime(req: CreateRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<CreateRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/runtimes`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListRuntimes(req: ListRuntimesRequest, initReq?: fm.InitReq): Promise<ListRuntimesResponse> {
+    return fm.fetchReq<ListRuntimesRequest, ListRuntimesResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/runtimes?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
+  }
+  static GetRuntime(req: GetRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<GetRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/runtimes/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteRuntime(req: DeleteRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<DeleteRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/runtimes/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
+  static CreateClusterRuntime(req: CreateClusterRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<CreateClusterRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/clusterruntimes`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListClusterRuntimes(req: ListClusterRuntimesRequest, initReq?: fm.InitReq): Promise<ListClusterRuntimesResponse> {
+    return fm.fetchReq<ListClusterRuntimesRequest, ListClusterRuntimesResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/clusterruntimes?${fm.renderURLSearchParams(req, ["workspace", "cluster"])}`, {...initReq, method: "GET"})
+  }
+  static GetClusterRuntime(req: GetClusterRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<GetClusterRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/clusterruntimes/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteClusterRuntime(req: DeleteClusterRuntimeRequest, initReq?: fm.InitReq): Promise<Runtime> {
+    return fm.fetchReq<DeleteClusterRuntimeRequest, Runtime>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/clusterruntimes/${req["name"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DoJobAction(req: BaizeManagement_apiJobV1alpha1Job.JobActionRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.Job> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.JobActionRequest, BaizeManagement_apiJobV1alpha1Job.Job>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/actions`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListSchedulers(req: BaizeManagement_apiJobV1alpha1Job.CreateJobRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.JobSchedulersResponse> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.CreateJobRequest, BaizeManagement_apiJobV1alpha1Job.JobSchedulersResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/schedulers`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListRdmas(req: BaizeManagement_apiJobV1alpha1Job.ListRdmasRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiJobV1alpha1Job.ListRdmasResponse> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.ListRdmasRequest, BaizeManagement_apiJobV1alpha1Job.ListRdmasResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/rdmas?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
+  }
+  static ListJobEvents(req: BaizeManagement_apiJobV1alpha1Job.ListJobEventsRequest, initReq?: fm.InitReq): Promise<BaizeManagement_apiEventV1alpha1Event.ListEventsResponse> {
+    return fm.fetchReq<BaizeManagement_apiJobV1alpha1Job.ListJobEventsRequest, BaizeManagement_apiEventV1alpha1Event.ListEventsResponse>(`/apis/baize.io/v1alpha2/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/events?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
   }
 }

package/package.json

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