@daocloud-proto/hydra 0.0.0-dev-87f727ca → 0.0.0-dev-8c334a47

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

@@ -9,12 +9,22 @@ 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"
 
+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",
   DELETE = "DELETE",
   OUT_OF_QUOTA = "OUT_OF_QUOTA",
   RECOVERY = "RECOVERY",
   UPDATE = "UPDATE",
+  CREATE = "CREATE",
 }
 
 export enum UsageReportRequestMode {
@@ -62,6 +72,33 @@ 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 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 +109,7 @@ 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/fine-tune-metrics:report`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
 }

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

@@ -8,6 +8,22 @@ 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 APIKeyStatus {
+  API_KEY_STATUS_UNSPECIFIED = "API_KEY_STATUS_UNSPECIFIED",
+  API_KEY_STATUS_ACTIVE = "API_KEY_STATUS_ACTIVE",
+  API_KEY_STATUS_DISABLED = "API_KEY_STATUS_DISABLED",
+  API_KEY_STATUS_EXPIRED = "API_KEY_STATUS_EXPIRED",
+}
+
 export enum TimePeriod {
   TIME_PERIOD_UNSPECIFIED = "TIME_PERIOD_UNSPECIFIED",
   TIME_PERIOD_HOUR = "TIME_PERIOD_HOUR",
@@ -16,15 +32,25 @@ export enum TimePeriod {
   TIME_PERIOD_MONTH = "TIME_PERIOD_MONTH",
 }
 
-export type CreateWSAPIKeyRequest = {
+
+/* hydra modified */ export type BaseCreateWSAPIKeyRequest = {
   name?: string
   workspace?: number
+  quota?: string
+  unlimitedQuota?: boolean
 }
 
-export type CreateAPIKeyRequest = {
+export type CreateWSAPIKeyRequest = BaseCreateWSAPIKeyRequest
+  & OneOf<{ expireTime: GoogleProtobufTimestamp.Timestamp }>
+
+
+/* hydra modified */ export type BaseCreateAPIKeyRequest = {
   name?: string
 }
 
+export type CreateAPIKeyRequest = BaseCreateAPIKeyRequest
+  & OneOf<{ expireTime: GoogleProtobufTimestamp.Timestamp }>
+
 export type APIKey = {
   id?: string
   key?: string
@@ -35,6 +61,13 @@ export type APIKey = {
   denyModels?: string[]
   workspace?: number
   createBy?: string
+  quota?: string
+  expireTime?: GoogleProtobufTimestamp.Timestamp
+  remainingQuota?: string
+  remainingQuotaSyncedAt?: GoogleProtobufTimestamp.Timestamp
+  status?: APIKeyStatus
+  unlimitedQuota?: boolean
+  usedQuota?: string
 }
 
 export type ListWSAPIKeyRequest = {

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

@@ -0,0 +1,14 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+export enum JobStatus {
+  UNKNOWN = "UNKNOWN",
+  PENDING = "PENDING",
+  RUNNING = "RUNNING",
+  COMPLETED = "COMPLETED",
+  FAILED = "FAILED",
+  DELETING = "DELETING",
+}

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

@@ -0,0 +1,100 @@
+/* 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 GoogleProtobufStruct from "../../../google/protobuf/struct.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+import * as HydraManagement_apiFinetuneV1alpha1Common from "./common.pb"
+import * as HydraManagement_apiFinetuneV1alpha1Finetuneconfig from "./finetuneconfig.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);
+
+/* 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 FineTuneJob = {
+  id?: string
+  name?: string
+  configName?: string
+  status?: HydraManagement_apiFinetuneV1alpha1Common.JobStatus
+  cluster?: string
+  namespace?: string
+  storageSpec?: GoogleProtobufStruct.Struct
+  runtimeType?: string
+  resourceConfig?: HydraManagement_apiFinetuneV1alpha1Finetuneconfig.ResourceConfig
+  creationTimestamp?: GoogleProtobufTimestamp.Timestamp
+  completionTimestamp?: GoogleProtobufTimestamp.Timestamp
+  metrics?: MetricPoint[]
+}
+
+export type CreateWSFineTuneJobRequest = {
+  workspace?: number
+  namespace?: string
+  name?: string
+  configName?: string
+  runtimeType?: string
+  resourceConfig?: HydraManagement_apiFinetuneV1alpha1Finetuneconfig.ResourceConfig
+}
+
+export type GetWSFineTuneJobRequest = {
+  workspace?: number
+  id?: string
+}
+
+export type ListWSFineTuneJobsRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ListWSFineTuneJobsResponse = {
+  items?: FineTuneJob[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type DeleteWSFineTuneJobRequest = {
+  workspace?: number
+  id?: string
+}
+
+export class WSFineTuneJobManagement {
+  static CreateWSFineTuneJob(req: CreateWSFineTuneJobRequest, initReq?: fm.InitReq): Promise<FineTuneJob> {
+    return fm.fetchReq<CreateWSFineTuneJobRequest, FineTuneJob>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/finetune/jobs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetWSFineTuneJob(req: GetWSFineTuneJobRequest, initReq?: fm.InitReq): Promise<FineTuneJob> {
+    return fm.fetchReq<GetWSFineTuneJobRequest, FineTuneJob>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/finetune/jobs/${req["id"]}?${fm.renderURLSearchParams(req, ["workspace", "id"])}`, {...initReq, method: "GET"})
+  }
+  static ListWSFineTuneJobs(req: ListWSFineTuneJobsRequest, initReq?: fm.InitReq): Promise<ListWSFineTuneJobsResponse> {
+    return fm.fetchReq<ListWSFineTuneJobsRequest, ListWSFineTuneJobsResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/finetune/jobs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteWSFineTuneJob(req: DeleteWSFineTuneJobRequest, initReq?: fm.InitReq): Promise<GoogleProtobufEmpty.Empty> {
+    return fm.fetchReq<DeleteWSFineTuneJobRequest, GoogleProtobufEmpty.Empty>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/finetune/jobs/${req["id"]}`, {...initReq, method: "DELETE"})
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.0.0-dev-87f727ca",
+  "version": "v0.0.0-dev-8c334a47",
   "description": "",
   "author": "",
   "license": "ISC"