@daocloud-proto/hydra 0.11.0 → 0.15.0

package/admin-api/storage/v1alpha1/filesstorage.pb.ts

@@ -0,0 +1,261 @@
+/* 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 EventV1alpha1Event from "../../../management-api/event/v1alpha1/event.pb"
+import * as HydraManagement_apiStorageV1alpha1Filesstorage from "../../../management-api/storage/v1alpha1/filesstorage.pb"
+import * as HydraStorageserverV1alpha1Storage_server from "../../../storageserver/v1alpha1/storage_server.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 PrePullActionRequestAction {
+  ACTION_UNSPECIFIED = "ACTION_UNSPECIFIED",
+  RERUN = "RERUN",
+}
+
+export type FilesStorageStatus = {
+  phase?: HydraManagement_apiStorageV1alpha1Filesstorage.PvcPhase
+  reason?: string
+  pvcName?: string
+  pvcNamespace?: string
+  storageServer?: StorageServerStatus
+}
+
+export type FilesStorage = {
+  name?: string
+  cluster?: string
+  pvcNamespace?: string
+  storageClassName?: string
+  capacity?: string
+  status?: FilesStorageStatus
+  createTime?: GoogleProtobufTimestamp.Timestamp
+  description?: string
+}
+
+export type StorageServerStatus = {
+  phase?: string
+  name?: string
+  namespace?: string
+}
+
+export type ListFilesStoragesRequest = {
+  page?: HydraCommonCommon.Pagination
+  cluster?: string
+}
+
+export type ListFilesStoragesResponse = {
+  items?: FilesStorage[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type GetFilesStorageRequest = {
+  name?: string
+}
+
+export type CreateFilesStorageRequest = {
+  cluster?: string
+  description?: string
+}
+
+export type UpdateFilesStorageRequest = {
+  name?: string
+  description?: string
+}
+
+export type DeleteFilesStorageRequest = {
+  name?: string
+}
+
+export type FileInfo = {
+  name?: string
+  type?: HydraStorageserverV1alpha1Storage_server.FileType
+  size?: string
+  modificationTimestamp?: string
+}
+
+export type ListFilesRequest = {
+  name?: string
+  path?: string
+  type?: HydraStorageserverV1alpha1Storage_server.FileType
+  searchKeyword?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type DeleteFileRequest = {
+  name?: string
+  fileName?: string
+  path?: string
+}
+
+export type ListFilesResponse = {
+  files?: FileInfo[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type CreateFolderRequest = {
+  name?: string
+  folderName?: string
+  path?: string
+}
+
+export type ListPrePullsRequest = {
+  storageName?: string
+  cluster?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ListPrePullsResponse = {
+  items?: HydraManagement_apiStorageV1alpha1Filesstorage.PrePull[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type CreatePrePullRequest = {
+  prepullName?: string
+  cluster?: string
+  path?: string
+  storageName?: string
+  description?: string
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+  source?: HydraManagement_apiStorageV1alpha1Filesstorage.DataSource
+  secretOptions?: HydraManagement_apiStorageV1alpha1Filesstorage.SecretOptions
+}
+
+
+type BaseUpdatePrePullRequest = {
+  cluster?: string
+  storageName?: string
+  prepullName?: string
+  description?: string
+  labels?: {[key: string]: string}
+  annotations?: {[key: string]: string}
+  secretOptions?: HydraManagement_apiStorageV1alpha1Filesstorage.SecretOptions
+  path?: string
+}
+
+export type UpdatePrePullRequest = BaseUpdatePrePullRequest
+  & OneOf<{ git: HydraManagement_apiStorageV1alpha1Filesstorage.DataSourceOptionsGit; s3: HydraManagement_apiStorageV1alpha1Filesstorage.DataSourceOptionsS3; http: HydraManagement_apiStorageV1alpha1Filesstorage.DataSourceOptionsHttp; huggingFace: HydraManagement_apiStorageV1alpha1Filesstorage.DataSourceOptionsHuggingFace; modelScope: HydraManagement_apiStorageV1alpha1Filesstorage.DataSourceOptionsModelScope }>
+
+export type DeletePrePullRequest = {
+  storageName?: string
+  prepullName?: string
+}
+
+export type PrePullActionRequest = {
+  cluster?: string
+  storageName?: string
+  prepullName?: string
+  action?: PrePullActionRequestAction
+}
+
+export type RestartPrePullRequest = {
+  storageName?: string
+  prepullName?: string
+}
+
+export type ValidateDataSourceSecretsRequest = {
+  source?: HydraManagement_apiStorageV1alpha1Filesstorage.DataSource
+  secretOptions?: HydraManagement_apiStorageV1alpha1Filesstorage.SecretOptions
+}
+
+export type ValidateDataSourceSecretsResponse = {
+  valid?: boolean
+  message?: string
+}
+
+export type ListDatasetEventsRequest = {
+  cluster?: string
+  storageName?: string
+  prepullName?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type SinglePrePullRequest = {
+  cluster?: string
+  storageName?: string
+  prepullName?: string
+}
+
+export type GetPrePullRequest = {
+  storageName?: string
+  prepullName?: string
+}
+
+export class AdminFilesStorageService {
+  static ListFilesStorages(req: ListFilesStoragesRequest, initReq?: fm.InitReq): Promise<ListFilesStoragesResponse> {
+    return fm.fetchReq<ListFilesStoragesRequest, ListFilesStoragesResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static GetFilesStorage(req: GetFilesStorageRequest, initReq?: fm.InitReq): Promise<FilesStorage> {
+    return fm.fetchReq<GetFilesStorageRequest, FilesStorage>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}?${fm.renderURLSearchParams(req, ["name"])}`, {...initReq, method: "GET"})
+  }
+  static CreateFilesStorage(req: CreateFilesStorageRequest, initReq?: fm.InitReq): Promise<FilesStorage> {
+    return fm.fetchReq<CreateFilesStorageRequest, FilesStorage>(`/apis/admin.hydra.io/v1alpha1/filesstorages`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateFilesStorage(req: UpdateFilesStorageRequest, initReq?: fm.InitReq): Promise<FilesStorage> {
+    return fm.fetchReq<UpdateFilesStorageRequest, FilesStorage>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DeleteFilesStorage(req: DeleteFilesStorageRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<DeleteFilesStorageRequest, GoogleProtobufEmpty.Empty>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}`, {...initReq, method: "DELETE"})
+  }
+  static ListFiles(req: ListFilesRequest, initReq?: fm.InitReq): Promise<ListFilesResponse> {
+    return fm.fetchReq<ListFilesRequest, ListFilesResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}/files?${fm.renderURLSearchParams(req, ["name"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteFile(req: DeleteFileRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<DeleteFileRequest, GoogleProtobufEmpty.Empty>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}/files/${req["fileName"]}:delete`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static CreateFolder(req: CreateFolderRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<CreateFolderRequest, GoogleProtobufEmpty.Empty>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}/folders/${req["folderName"]}`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListReferencedBy(req: HydraManagement_apiStorageV1alpha1Filesstorage.ListReferencedByRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.ListReferencedByResponse> {
+    return fm.fetchReq<HydraManagement_apiStorageV1alpha1Filesstorage.ListReferencedByRequest, HydraManagement_apiStorageV1alpha1Filesstorage.ListReferencedByResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["name"]}/referencedby/${req["itemType"]}?${fm.renderURLSearchParams(req, ["name", "itemType"])}`, {...initReq, method: "GET"})
+  }
+  static ListPrePulls(req: ListPrePullsRequest, initReq?: fm.InitReq): Promise<ListPrePullsResponse> {
+    return fm.fetchReq<ListPrePullsRequest, ListPrePullsResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls?${fm.renderURLSearchParams(req, ["storageName"])}`, {...initReq, method: "GET"})
+  }
+  static CreatePrePull(req: CreatePrePullRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PrePull> {
+    return fm.fetchReq<CreatePrePullRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PrePull>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdatePrePull(req: UpdatePrePullRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PrePull> {
+    return fm.fetchReq<UpdatePrePullRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PrePull>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static DeletePrePull(req: DeletePrePullRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<DeletePrePullRequest, GoogleProtobufEmpty.Empty>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}`, {...initReq, method: "DELETE"})
+  }
+  static DatasetDoAction(req: PrePullActionRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PrePull> {
+    return fm.fetchReq<PrePullActionRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PrePull>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}/actions`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListPrePullEvents(req: ListDatasetEventsRequest, initReq?: fm.InitReq): Promise<EventV1alpha1Event.ListEventsResponse> {
+    return fm.fetchReq<ListDatasetEventsRequest, EventV1alpha1Event.ListEventsResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}/events?${fm.renderURLSearchParams(req, ["storageName", "prepullName"])}`, {...initReq, method: "GET"})
+  }
+  static GetPrePullSyncProcess(req: SinglePrePullRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PrePullSyncProcess> {
+    return fm.fetchReq<SinglePrePullRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PrePullSyncProcess>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}/sync-process?${fm.renderURLSearchParams(req, ["storageName", "prepullName"])}`, {...initReq, method: "GET"})
+  }
+  static ValidateDataSourceSecrets(req: ValidateDataSourceSecretsRequest, initReq?: fm.InitReq): Promise<ValidateDataSourceSecretsResponse> {
+    return fm.fetchReq<ValidateDataSourceSecretsRequest, ValidateDataSourceSecretsResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/validate-secrets`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetPodInstanceList(req: HydraManagement_apiStorageV1alpha1Filesstorage.PodRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PodInstanceListResponse> {
+    return fm.fetchReq<HydraManagement_apiStorageV1alpha1Filesstorage.PodRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PodInstanceListResponse>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}/pods?${fm.renderURLSearchParams(req, ["storageName", "prepullName"])}`, {...initReq, method: "GET"})
+  }
+  static GetPrePull(req: GetPrePullRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.PrePull> {
+    return fm.fetchReq<GetPrePullRequest, HydraManagement_apiStorageV1alpha1Filesstorage.PrePull>(`/apis/admin.hydra.io/v1alpha1/filesstorages/${req["storageName"]}/prepulls/${req["prepullName"]}?${fm.renderURLSearchParams(req, ["storageName", "prepullName"])}`, {...initReq, method: "GET"})
+  }
+  static CreateSFTPAccess(req: HydraManagement_apiStorageV1alpha1Filesstorage.CreateSFTPAccessRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.SFTPAccessResponse> {
+    return fm.fetchReq<HydraManagement_apiStorageV1alpha1Filesstorage.CreateSFTPAccessRequest, HydraManagement_apiStorageV1alpha1Filesstorage.SFTPAccessResponse>(`/apis/hydra.io/v1alpha1/filesstorages/${req["storageName"]}/sftp-access`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetSFTPAccess(req: HydraManagement_apiStorageV1alpha1Filesstorage.GetSFTPAccessRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiStorageV1alpha1Filesstorage.SFTPAccessResponse> {
+    return fm.fetchReq<HydraManagement_apiStorageV1alpha1Filesstorage.GetSFTPAccessRequest, HydraManagement_apiStorageV1alpha1Filesstorage.SFTPAccessResponse>(`/apis/hydra.io/v1alpha1/filesstorages/${req["storageName"]}/sftp-access?${fm.renderURLSearchParams(req, ["storageName"])}`, {...initReq, method: "GET"})
+  }
+}

package/admin-api/workspace-quota/v1alpha1/workspace_quota.pb.ts

@@ -0,0 +1,61 @@
+/* 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 GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum RefreshCycle {
+  REFRESH_CYCLE_UNSPECIFIED = "REFRESH_CYCLE_UNSPECIFIED",
+  REFRESH_CYCLE_DAY = "REFRESH_CYCLE_DAY",
+  REFRESH_CYCLE_WEEK = "REFRESH_CYCLE_WEEK",
+  REFRESH_CYCLE_MONTH = "REFRESH_CYCLE_MONTH",
+}
+
+export type WorkspaceQuota = {
+  workspace?: number
+  quota?: string
+  nextRefreshTime?: GoogleProtobufTimestamp.Timestamp
+  refreshCycle?: RefreshCycle
+  usedQuota?: string
+  allocatedQuota?: string
+  updateTime?: GoogleProtobufTimestamp.Timestamp
+  unlimitedQuota?: boolean
+  workspaceName?: string
+}
+
+export type ListWorkspaceQuotasRequest = {
+  page?: HydraCommonCommon.Pagination
+  workspace?: number
+}
+
+export type ListWorkspaceQuotasResponse = {
+  items?: WorkspaceQuota[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type UpdateWorkspaceQuotaRequest = {
+  workspace?: number
+  quota?: string
+  unlimitedQuota?: boolean
+}
+
+export type UpdateWorkspaceQuotaRefreshCycleRequest = {
+  workspace?: number
+  refreshCycle?: RefreshCycle
+}
+
+export class WorkspaceQuotaService {
+  static ListWorkspaceQuotas(req: ListWorkspaceQuotasRequest, initReq?: fm.InitReq): Promise<ListWorkspaceQuotasResponse> {
+    return fm.fetchReq<ListWorkspaceQuotasRequest, ListWorkspaceQuotasResponse>(`/apis/admin.hydra.io/v1alpha1/workspace-quotas?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static UpdateWorkspaceQuota(req: UpdateWorkspaceQuotaRequest, initReq?: fm.InitReq): Promise<WorkspaceQuota> {
+    return fm.fetchReq<UpdateWorkspaceQuotaRequest, WorkspaceQuota>(`/apis/admin.hydra.io/v1alpha1/workspace-quotas/${req["workspace"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateWorkspaceQuotaRefreshCycle(req: UpdateWorkspaceQuotaRefreshCycleRequest, initReq?: fm.InitReq): Promise<WorkspaceQuota> {
+    return fm.fetchReq<UpdateWorkspaceQuotaRefreshCycleRequest, WorkspaceQuota>(`/apis/admin.hydra.io/v1alpha1/workspace-quotas/${req["workspace"]}/refresh-cycle`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+}

package/common/common.pb.ts

@@ -18,6 +18,13 @@ export enum Mode {
   Any = "Any",
 }
 
+export enum NodePhase {
+  NODE_PHASE_UNSPECIFIED = "NODE_PHASE_UNSPECIFIED",
+  Ready = "Ready",
+  Not_Ready = "Not_Ready",
+  Unknown = "Unknown",
+}
+
 export enum SearchIndexFieldBuilderType {
   FIELD_TYPE_UNSPECIFIED = "FIELD_TYPE_UNSPECIFIED",
   FILED = "FILED",
@@ -101,4 +108,52 @@ export type AuditObject = {
   kind?: string
   group?: string
   version?: string
+}
+
+export type Node = {
+  name?: string
+  arch?: string
+  labels?: {[key: string]: string}
+  gpuModels?: string[]
+  status?: NodeStatus
+}
+
+export type NodeStatus = {
+  phase?: NodePhase
+  addresses?: NodeAddress[]
+  cpuCapacity?: string
+  cpuAllocated?: number
+  cpuUsage?: number
+  memoryCapacity?: string
+  memoryAllocated?: number
+  memoryUsage?: number
+  systemInfo?: NodeSystemInfo
+  allowedPodNumber?: number
+  podAllocated?: number
+  readyPodNumber?: number
+  storageCapacity?: string
+  storageAllocated?: string
+  storageUsage?: number
+  storageDriver?: string
+  gpuTotal?: string
+  gpuAllocated?: string
+  gpuMemoryTotal?: string
+  gpuMemoryAllocated?: string
+  gpuCoreUsage?: number
+  gpuMemoryUsage?: number
+}
+
+export type NodeAddress = {
+  type?: string
+  address?: string
+}
+
+export type NodeSystemInfo = {
+  kernelVersion?: string
+  osImage?: string
+  containerRuntimeVersion?: string
+  kubeletVersion?: string
+  kubeProxyVersion?: string
+  operatingSystem?: string
+  architecture?: string
 }

package/common/errorcode.ts

@@ -39,8 +39,8 @@ export enum ErrorCode {
   // EN: out of page: %v
   // ZH: 分页错误：%v
   USER_OUT_OF_PAGE = "USER-OUT_OF_PAGE",
-  // EN: permission denied
-  // ZH: 拒绝访问
+  // EN: permission denied: %v
+  // ZH: 拒绝访问：%v
   USER_PERMISSION_DENIED = "USER-PERMISSION_DENIED",
   // EN: unauthorized request
   // ZH: 未授权的请求

package/management-api/agent/v1alpha1/event.pb.ts

@@ -8,6 +8,16 @@ 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_apiApikeyV1alpha1Apikey from "../../apikey/v1alpha1/apikey.pb"
+import * as HydraManagement_apiModel_typeV1alpha1Model_type from "../../model_type/v1alpha1/model_type.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 APIKeyEventType {
   UNKNOWN = "UNKNOWN",
@@ -15,6 +25,7 @@ export enum APIKeyEventType {
   OUT_OF_QUOTA = "OUT_OF_QUOTA",
   RECOVERY = "RECOVERY",
   UPDATE = "UPDATE",
+  CREATE = "CREATE",
 }
 
 export enum UsageReportRequestMode {
@@ -46,12 +57,18 @@ export type UsageReportRequestUsage = {
   inputImages?: UsageReportRequestUsageImage
   outputImages?: UsageReportRequestUsageImage
   requests?: number
+  cachedTokens?: string
+  inputCharacters?: string
+  outputAudioDuration?: string
+  sampleRate?: string
+  inputAudioDuration?: string
 }
 
 export type UsageReportRequest = {
   apiKeyId?: string
   requestModelName?: string
   upstreamModelName?: string
+  requestModelType?: HydraManagement_apiModel_typeV1alpha1Model_type.RequestModelType
   usage?: UsageReportRequestUsage
   firstRequestTime?: GoogleProtobufTimestamp.Timestamp
   lastRequestTime?: GoogleProtobufTimestamp.Timestamp
@@ -62,6 +79,42 @@ export type UsageReportResponse = {
   accepted?: boolean
 }
 
+
+/* hydra modified */ export type BaseMetricPoint = {
+  currentSteps?: number
+  totalSteps?: number
+}
+
+export type MetricPoint = BaseMetricPoint
+  & OneOf<{ loss: number }>
+  & OneOf<{ evalLoss: number }>
+  & OneOf<{ predictLoss: number }>
+  & OneOf<{ reward: number }>
+  & OneOf<{ accuracy: number }>
+  & OneOf<{ lr: number }>
+  & OneOf<{ epoch: number }>
+  & OneOf<{ percentage: number }>
+  & OneOf<{ elapsedTime: string }>
+  & OneOf<{ remainingTime: string }>
+
+export type ReportFineTuneMetricsRequest = {
+  jobId?: string
+  metric?: MetricPoint
+}
+
+export type ReportFineTuneMetricsResponse = {
+  accepted?: boolean
+}
+
+export type ReportEvalMetricsRequest = {
+  jobId?: string
+  metric?: {[key: string]: number}
+}
+
+export type ReportEvalMetricsResponse = {
+  accepted?: boolean
+}
+
 export class AgentCommunicationService {
   static FetchAPIKeyByKey(req: FetchAPIKeyRequest, initReq?: fm.InitReq): Promise<HydraManagement_apiApikeyV1alpha1Apikey.APIKey> {
     return fm.fetchReq<FetchAPIKeyRequest, HydraManagement_apiApikeyV1alpha1Apikey.APIKey>(`/hydra.management_api.agent.v1alpha1.AgentCommunicationService/FetchAPIKeyByKey`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
@@ -72,4 +125,10 @@ export class AgentCommunicationService {
   static UsageReport(req: UsageReportRequest, initReq?: fm.InitReq): Promise<UsageReportResponse> {
     return fm.fetchReq<UsageReportRequest, UsageReportResponse>(`/hydra.management_api.agent.v1alpha1.AgentCommunicationService/UsageReport`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
   }
+  static ReportFineTuneMetrics(req: ReportFineTuneMetricsRequest, initReq?: fm.InitReq): Promise<ReportFineTuneMetricsResponse> {
+    return fm.fetchReq<ReportFineTuneMetricsRequest, ReportFineTuneMetricsResponse>(`/apis/hydra.io/v1alpha1/agent/finetune/finetunejob-metrics:report`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ReportEvalMetrics(req: ReportEvalMetricsRequest, initReq?: fm.InitReq): Promise<ReportEvalMetricsResponse> {
+    return fm.fetchReq<ReportEvalMetricsRequest, ReportEvalMetricsResponse>(`/apis/hydra.io/v1alpha1/agent/finetune/evaljob-metrics:report`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
 }

package/management-api/agent/v1alpha1/server_registration.pb.ts

@@ -0,0 +1,33 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufEmpty from "../../../google/protobuf/empty.pb"
+export type RegisterServerRequest = {
+  clusterName?: string
+  serverName?: string
+  host?: string
+  grpcConfig?: GRPCConfig
+}
+
+export type GRPCConfig = {
+  plaintext?: boolean
+  insecure?: boolean
+}
+
+export type UnregisterServerRequest = {
+  clusterName?: string
+  serverName?: string
+}
+
+export class ServerRegistrationService {
+  static RegisterServer(req: RegisterServerRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<RegisterServerRequest, GoogleProtobufEmpty.Empty>(`/hydra.management_api.agent.v1alpha1.ServerRegistrationService/RegisterServer`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UnregisterServer(req: UnregisterServerRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<UnregisterServerRequest, GoogleProtobufEmpty.Empty>(`/hydra.management_api.agent.v1alpha1.ServerRegistrationService/UnregisterServer`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+}

package/management-api/agent/v1alpha1/sftp_user.pb.ts

@@ -0,0 +1,39 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as fm from "../../../fetch.pb"
+export type ListClusterUsersRequest = {
+  clusterName?: string
+}
+
+export type ListClusterUsersResponse = {
+  users?: SFTPUserInfo[]
+}
+
+export type SFTPUserInfo = {
+  namespace?: string
+  pvc?: string
+  username?: string
+  uid?: number
+  gid?: number
+}
+
+export type ListUserPublicKeysRequest = {
+  username?: string
+}
+
+export type ListUserPublicKeysResponse = {
+  publicKeys?: string[]
+}
+
+export class SFTPUserService {
+  static ListClusterUsers(req: ListClusterUsersRequest, initReq?: fm.InitReq): Promise<ListClusterUsersResponse> {
+    return fm.fetchReq<ListClusterUsersRequest, ListClusterUsersResponse>(`/hydra.management_api.agent.v1alpha1.SFTPUserService/ListClusterUsers`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static ListUserPublicKeys(req: ListUserPublicKeysRequest, initReq?: fm.InitReq): Promise<ListUserPublicKeysResponse> {
+    return fm.fetchReq<ListUserPublicKeysRequest, ListUserPublicKeysResponse>(`/hydra.management_api.agent.v1alpha1.SFTPUserService/ListUserPublicKeys`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+}