@daocloud-proto/hydra 0.0.0-dev-d87e43f9 → 0.0.0-dev-997245b5

package/management-api/finetune/v1alpha1/export_job.pb.ts

@@ -0,0 +1,84 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as HydraCommonCommon from "../../../common/common.pb"
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufEmpty from "../../../google/protobuf/empty.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+import * as HydraManagement_apiFinetuneV1alpha1Common from "./common.pb"
+
+export enum ExportDevice {
+  cpu = "cpu",
+  auto = "auto",
+}
+
+export type ExportJob = {
+  workspace?: number
+  name?: string
+  cluster?: string
+  namespace?: string
+  finetuneConfigName?: string
+  finetuneJobName?: string
+  filesStorageName?: string
+  exportSubdir?: string
+  exportSize?: number
+  exportDevice?: ExportDevice
+  exportLegacyFormat?: boolean
+  status?: HydraManagement_apiFinetuneV1alpha1Common.JobStatus
+  reason?: string
+  creationTime?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type SaveExportJobRequest = {
+  workspace?: number
+  name?: string
+  cluster?: string
+  namespace?: string
+  finetuneConfigName?: string
+  finetuneJobName?: string
+  filesStorageName?: string
+  exportSubdir?: string
+  exportSize?: number
+  exportDevice?: ExportDevice
+  exportLegacyFormat?: boolean
+}
+
+export type SingleExportJobRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  name?: string
+}
+
+export type ListExportJobRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ListExportJobResponse = {
+  exportJobs?: ExportJob[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export class WSExportJobService {
+  static CreatExportJob(req: SaveExportJobRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<SaveExportJobRequest, GoogleProtobufEmpty.Empty>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/finetune-exportjobs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DeleteExportJob(req: SingleExportJobRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<SingleExportJobRequest, GoogleProtobufEmpty.Empty>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/finetune-exportjobs/${req["name"]}`, {...initReq, method: "DELETE"})
+  }
+  static DefaultExportJob(req: GoogleProtobufEmpty.Empty, initReq?: fm.InitReq): Promise<ExportJob> {
+    return fm.fetchReq<GoogleProtobufEmpty.Empty, ExportJob>(`/apis/hydra.io/v1alpha1/finetune-exportjobs:default?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static GetExportJob(req: SingleExportJobRequest, initReq?: fm.InitReq): Promise<ExportJob> {
+    return fm.fetchReq<SingleExportJobRequest, ExportJob>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/finetune-exportjobs/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static ListExportJob(req: ListExportJobRequest, initReq?: fm.InitReq): Promise<ListExportJobResponse> {
+    return fm.fetchReq<ListExportJobRequest, ListExportJobResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/finetune-exportjobs?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
+  }
+}

package/management-api/finetune/v1alpha1/finetuneconfig.pb.ts

@@ -8,6 +8,15 @@ import * as HydraCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.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 TrainingStage {
   TRAINING_STAGE_UNSPECIFIED = "TRAINING_STAGE_UNSPECIFIED",
   SFT = "SFT",
@@ -86,19 +95,18 @@ export type DpoConfig = {
   refModel?: string
 }
 
-export type ResourceQuota = {
-  request?: string
-  limit?: string
-}
 
-export type ResourceConfig = {
-  replicas?: number
+/* hydra modified */ export type BaseResourceConfig = {
   gpuType?: string
   gpuCount?: number
-  cpu?: ResourceQuota
-  memory?: ResourceQuota
+  cpu?: number
+  memory?: number
+  perGpuMemory?: number
 }
 
+export type ResourceConfig = BaseResourceConfig
+  & OneOf<{ gpuCores: number }>
+
 export type FineTuneConfig = {
   name?: string
   cluster?: string

package/management-api/finetune/v1alpha1/finetunejob.pb.ts

@@ -51,6 +51,7 @@ export type FineTuneJob = {
   creationTimestamp?: GoogleProtobufTimestamp.Timestamp
   completionTimestamp?: GoogleProtobufTimestamp.Timestamp
   metrics?: MetricPoint[]
+  trainingParams?: GoogleProtobufStruct.Struct
 }
 
 export type CreateWSFineTuneJobRequest = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.0.0-dev-d87e43f9",
+  "version": "v0.0.0-dev-997245b5",
   "description": "",
   "author": "",
   "license": "ISC"