@daocloud-proto/baize 0.105.0 → 0.105.1

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

@@ -24,6 +24,7 @@ export enum DataSourceType {
   HTTP = "HTTP",
   PVC = "PVC",
   NFS = "NFS",
+  CONDA = "CONDA",
 }
 
 export enum DatasetStatusPhase {
@@ -34,6 +35,12 @@ export enum DatasetStatusPhase {
   FAILED = "FAILED",
 }
 
+export enum DataSourceOptionsCondaPackageManager {
+  PACKAGE_MANAGER_UNSPECIFIED = "PACKAGE_MANAGER_UNSPECIFIED",
+  PIP = "PIP",
+  CONDA = "CONDA",
+}
+
 export enum SecretOptionsAuthType {
   AUTH_TYPE_UNSPECIFIED = "AUTH_TYPE_UNSPECIFIED",
   BASIC = "BASIC",
@@ -94,12 +101,21 @@ export type DataSourceOptionsHttp = {
   headers?: {[key: string]: string}
 }
 
+export type DataSourceOptionsConda = {
+  pythonVersion?: string
+  packageManager?: DataSourceOptionsCondaPackageManager
+  condaEnvironmentYml?: string
+  pipRequirementsTxt?: string
+  pipIndexUrl?: string
+  gpuType?: string
+}
+
 
 /* baize modified */ export type BaseDataSourceOptions = {
 }
 
 export type DataSourceOptions = BaseDataSourceOptions
-  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp }>
+  & OneOf<{ git: DataSourceOptionsGit; s3: DataSourceOptionsS3; http: DataSourceOptionsHttp; conda: DataSourceOptionsConda }>
 
 export type SecretOptionsBasicAuth = {
   username?: string

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

@@ -52,6 +52,8 @@ export type JobCreationBaseConfig = {
   podConfig?: BaizeCommonK8s.PodConfig
   labels?: {[key: string]: string}
   annotations?: {[key: string]: string}
+  workingDir?: string
+  shmSize?: number
 }
 
 export type JobRoleDifferenceConfig = {

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

@@ -12,6 +12,15 @@ import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 import * as BaizeManagement_apiAnalysisV1alpha1Analysis from "../../analysis/v1alpha1/analysis.pb"
 import * as BaizeManagement_apiJobV1alpha1Common from "./common.pb"
 
+type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
+type OneOf<T> =
+  | { [k in keyof T]?: undefined }
+  | (
+    keyof T extends infer K ?
+      (K extends string & keyof T ? { [k in K]: T[K] } & Absent<T, K>
+        : never)
+    : never);
+
 export enum JobType {
   JOB_TYPE_UNSPECIFIED = "JOB_TYPE_UNSPECIFIED",
   PYTORCH = "PYTORCH",
@@ -19,6 +28,12 @@ export enum JobType {
   PADDLE = "PADDLE",
 }
 
+export enum RestartPolicy {
+  RESTART_POLICY_UNSPECIFIED = "RESTART_POLICY_UNSPECIFIED",
+  RESTART_POLICY_NEVER = "RESTART_POLICY_NEVER",
+  RESTART_POLICY_ON_FAILURE = "RESTART_POLICY_ON_FAILURE",
+}
+
 export enum JobActionRequestAction {
   JOB_ACTION_UNSPECIFIED = "JOB_ACTION_UNSPECIFIED",
   STOP = "STOP",
@@ -46,6 +61,7 @@ export type Job = {
   runningDuration?: number
   totalResources?: BaizeCommonK8s.Resources
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
+  trainingConfig?: TrainingConfig
 }
 
 export type ListJobsRequest = {
@@ -80,8 +96,18 @@ export type CreateJobRequest = {
   trainingMode?: BaizeManagement_apiJobV1alpha1Common.TrainingMode
   roleConfig?: {[key: string]: BaizeManagement_apiJobV1alpha1Common.JobRoleDifferenceConfig}
   analysis?: BaizeManagement_apiAnalysisV1alpha1Analysis.AnalysisConfig
+  trainingConfig?: TrainingConfig
 }
 
+
+/* baize modified */ export type BaseTrainingConfig = {
+}
+
+export type TrainingConfig = BaseTrainingConfig
+  & OneOf<{ restartPolicy: RestartPolicy }>
+  & OneOf<{ maxRetries: number }>
+  & OneOf<{ maxTrainingDuration: string }>
+
 export type JobActionRequest = {
   workspace?: number
   type?: JobType
@@ -119,6 +145,6 @@ export class JobsManagement {
     return fm.fetchReq<JobActionRequest, Job>(`/apis/baize.io/v1alpha1/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: CreateJobRequest, initReq?: fm.InitReq): Promise<JobSchedulersResponse> {
-    return fm.fetchReq<CreateJobRequest, JobSchedulersResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/schedulers?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+    return fm.fetchReq<CreateJobRequest, JobSchedulersResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/schedulers`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
   }
 }

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

@@ -58,6 +58,8 @@ export type NotebookConfig = {
   runAsRoot?: boolean
   serviceType?: NotebookConfigServiceType
   enableSSH?: boolean
+  affinity?: BaizeCommonK8s.Affinity
+  tolerationSeconds?: string
 }
 
 export type NotebookStatus = {

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

@@ -15,6 +15,7 @@ export enum ResourceType {
   TENSORFLOW = "TENSORFLOW",
   PADDLE = "PADDLE",
   NOTEBOOK = "NOTEBOOK",
+  INFERENCE = "INFERENCE",
 }
 
 export type PodRequest = {

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

@@ -119,6 +119,7 @@ export type InferenceServing = {
   podConfig?: BaizeCommonK8s.PodConfig
   status?: InferenceServingStatus
   lastUpdated?: GoogleProtobufTimestamp.Timestamp
+  authType?: ServingAuthAuthType
 }
 
 export type InferenceServingStatusModelStatus = {

package/package.json

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