@daocloud-proto/hydra 0.0.0-dev-5c60a30e → 0.0.0-dev-c571dd90

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

@@ -0,0 +1,60 @@
+/* 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
+}
+
+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/management-api/apikey/v1alpha1/apikey.pb.ts

@@ -17,13 +17,6 @@ type OneOf<T> =
         : 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",
@@ -65,9 +58,11 @@ export type APIKey = {
   expireTime?: GoogleProtobufTimestamp.Timestamp
   remainingQuota?: string
   remainingQuotaSyncedAt?: GoogleProtobufTimestamp.Timestamp
-  status?: APIKeyStatus
   unlimitedQuota?: boolean
   usedQuota?: string
+  lastUsedTime?: GoogleProtobufTimestamp.Timestamp
+  expired?: boolean
+  disabled?: boolean
 }
 
 export type ListWSAPIKeyRequest = {

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

@@ -30,6 +30,44 @@ export enum FineTuningType {
   FREEZE = "FREEZE",
 }
 
+export enum ComputeType {
+  COMPUTE_TYPE_UNSPECIFIED = "COMPUTE_TYPE_UNSPECIFIED",
+  BF16 = "BF16",
+  FP16 = "FP16",
+  FP32 = "FP32",
+  PURE_BF16 = "PURE_BF16",
+}
+
+export enum LrSchedulerType {
+  LR_SCHEDULER_TYPE_UNSPECIFIED = "LR_SCHEDULER_TYPE_UNSPECIFIED",
+  COSINE = "COSINE",
+  LINEAR = "LINEAR",
+  COSINE_WITH_RESTARTS = "COSINE_WITH_RESTARTS",
+  POLYNOMIAL = "POLYNOMIAL",
+  CONSTANT = "CONSTANT",
+  CONSTANT_WITH_WARMUP = "CONSTANT_WITH_WARMUP",
+  INVERSE_SQRT = "INVERSE_SQRT",
+  REDUCE_LR_ON_PLATEAU = "REDUCE_LR_ON_PLATEAU",
+  COSINE_WITH_MIN_LR = "COSINE_WITH_MIN_LR",
+  WARMUP_STABLE_DECAY = "WARMUP_STABLE_DECAY",
+}
+
+export enum OptimizerType {
+  OPTIMIZER_TYPE_UNSPECIFIED = "OPTIMIZER_TYPE_UNSPECIFIED",
+  ADAMW_TORCH = "ADAMW_TORCH",
+  ADAMW_HF = "ADAMW_HF",
+  SGD = "SGD",
+  ADAFACTOR = "ADAFACTOR",
+  ADAGRAD = "ADAGRAD",
+}
+
+export enum QuantizationMethodType {
+  QUANTIZATION_METHOD_TYPE_UNSPECIFIED = "QUANTIZATION_METHOD_TYPE_UNSPECIFIED",
+  BITSANDBYTES = "BITSANDBYTES",
+  HQQ = "HQQ",
+  EETQ = "EETQ",
+}
+
 export type ModelConfig = {
   modelId?: string
   modelVolumeId?: string
@@ -52,7 +90,7 @@ export type LoraConfig = {
   pissaConvert?: boolean
   createNewAdapter?: boolean
   quantizationBit?: number
-  quantizationMethod?: string
+  quantizationMethod?: QuantizationMethodType
   ropeScaling?: string
   booster?: string
 }
@@ -70,11 +108,11 @@ export type TrainingConfig = {
   perDeviceTrainBatchSize?: number
   gradientAccumulationSteps?: number
   learningRate?: string
-  lrSchedulerType?: string
+  lrSchedulerType?: LrSchedulerType
   warmupRatio?: string
   warmupSteps?: number
-  bf16?: boolean
-  optimizer?: string
+  computeType?: ComputeType
+  optimizer?: OptimizerType
   seed?: number
   ddpTimeout?: number
   maxGradNorm?: string

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.0.0-dev-5c60a30e",
+  "version": "v0.0.0-dev-c571dd90",
   "description": "",
   "author": "",
   "license": "ISC"