@daocloud-proto/hydra 0.0.0-dev-593036d8 → 0.0.0-dev-87f727ca

package/admin-api/core/v1alpha1/core.pb.ts

@@ -7,6 +7,7 @@
 import * as HydraCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufEmpty from "../../../google/protobuf/empty.pb"
+import * as HydraManagement_apiCoreV1alpha1Permissions from "../../../management-api/core/v1alpha1/permissions.pb"
 import * as HydraManagement_apiWorkspaceV1alpha1Workspace from "../../../management-api/workspace/v1alpha1/workspace.pb"
 export type ListClusterRequest = {
   page?: HydraCommonCommon.Pagination
@@ -22,6 +23,10 @@ export type ConstantsResponse = {
   defaultMountPath?: string
 }
 
+export type ListPlatformRolePermissionsForCurrentUserResponse = {
+  permissions?: HydraManagement_apiCoreV1alpha1Permissions.Permission[]
+}
+
 export class CoreService {
   static AdminListCluster(req: ListClusterRequest, initReq?: fm.InitReq): Promise<ListClusterResponse> {
     return fm.fetchReq<ListClusterRequest, ListClusterResponse>(`/apis/admin.hydra.io/v1alpha1/clusters?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
@@ -29,4 +34,7 @@ export class CoreService {
   static AdminListConstants(req: GoogleProtobufEmpty.Empty, initReq?: fm.InitReq): Promise<ConstantsResponse> {
     return fm.fetchReq<GoogleProtobufEmpty.Empty, ConstantsResponse>(`/apis/admin.hydra.io/v1alpha1/constants?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
+  static ListPlatformRolePermissionsForCurrentUser(req: GoogleProtobufEmpty.Empty, initReq?: fm.InitReq): Promise<ListPlatformRolePermissionsForCurrentUserResponse> {
+    return fm.fetchReq<GoogleProtobufEmpty.Empty, ListPlatformRolePermissionsForCurrentUserResponse>(`/apis/admin.hydra.io/v1alpha1/permissions?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
 }

package/management-api/core/v1alpha1/permissions.pb.ts

@@ -5,26 +5,44 @@
 */
 
 import * as fm from "../../../fetch.pb"
-export type ListRolePermissionsByWorkspaceForCurrentUserRequest = {
-  workspace?: number
-}
 
-export type ListRolePermissionsByWorkspaceForCurrentUserResponse = {
-  permissions?: string[]
+export enum Permission {
+  MODEL_GET = "MODEL_GET",
+  MODEL_CREATE = "MODEL_CREATE",
+  MODEL_UPDATE = "MODEL_UPDATE",
+  MODEL_DELETE = "MODEL_DELETE",
+  MAAS_MODEL_GET = "MAAS_MODEL_GET",
+  MAAS_MODEL_CREATE = "MAAS_MODEL_CREATE",
+  MAAS_MODEL_UPDATE = "MAAS_MODEL_UPDATE",
+  MODEL_SERVING_GET = "MODEL_SERVING_GET",
+  MODEL_SERVING_CREATE = "MODEL_SERVING_CREATE",
+  MODEL_SERVING_UPDATE = "MODEL_SERVING_UPDATE",
+  MODEL_SERVING_DELETE = "MODEL_SERVING_DELETE",
+  APIKEY_GET = "APIKEY_GET",
+  APIKEY_CREATE = "APIKEY_CREATE",
+  APIKEY_DELETE = "APIKEY_DELETE",
+  CLUSTER_GET = "CLUSTER_GET",
+  FILES_STORAGE_GET = "FILES_STORAGE_GET",
+  FILES_STORAGE_CREATE = "FILES_STORAGE_CREATE",
+  FILES_STORAGE_UPDATE = "FILES_STORAGE_UPDATE",
+  FILES_STORAGE_DELETE = "FILES_STORAGE_DELETE",
+  QUEUE_GET = "QUEUE_GET",
+  QUEUE_CREATE = "QUEUE_CREATE",
+  QUEUE_UPDATE = "QUEUE_UPDATE",
+  QUEUE_DELETE = "QUEUE_DELETE",
+  ALL = "ALL",
 }
 
-export type ListPlatformRolePermissionsForCurrentUserRequest = {
+export type ListRolePermissionsByWorkspaceForCurrentUserRequest = {
+  workspace?: number
 }
 
-export type ListPlatformRolePermissionsForCurrentUserResponse = {
-  permissions?: string[]
+export type ListRolePermissionsByWorkspaceForCurrentUserResponse = {
+  permissions?: Permission[]
 }
 
 export class Core {
   static ListRolePermissionsByWorkspaceForCurrentUser(req: ListRolePermissionsByWorkspaceForCurrentUserRequest, initReq?: fm.InitReq): Promise<ListRolePermissionsByWorkspaceForCurrentUserResponse> {
     return fm.fetchReq<ListRolePermissionsByWorkspaceForCurrentUserRequest, ListRolePermissionsByWorkspaceForCurrentUserResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/permissions?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }
-  static ListPlatformRolePermissionsForCurrentUser(req: ListPlatformRolePermissionsForCurrentUserRequest, initReq?: fm.InitReq): Promise<ListPlatformRolePermissionsForCurrentUserResponse> {
-    return fm.fetchReq<ListPlatformRolePermissionsForCurrentUserRequest, ListPlatformRolePermissionsForCurrentUserResponse>(`/apis/hydra.io/v1alpha1/permissions?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
-  }
 }

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

@@ -0,0 +1,182 @@
+/* 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 TrainingStage {
+  TRAINING_STAGE_UNSPECIFIED = "TRAINING_STAGE_UNSPECIFIED",
+  SFT = "SFT",
+  DPO = "DPO",
+}
+
+export enum FineTuningType {
+  FINE_TUNING_TYPE_UNSPECIFIED = "FINE_TUNING_TYPE_UNSPECIFIED",
+  LORA = "LORA",
+  FULL = "FULL",
+  FREEZE = "FREEZE",
+}
+
+export type ModelConfig = {
+  modelId?: string
+  modelVolumeId?: string
+  template?: string
+  trustRemoteCode?: boolean
+}
+
+export type LoraConfig = {
+  loraRank?: number
+  loraAlpha?: number
+  loraDropout?: number
+  loraTarget?: string
+  additionalTarget?: string
+  loraPlusLrRatio?: number
+  loraPlusLrEmbedding?: number
+  useRslora?: boolean
+  useDora?: boolean
+  pissaInit?: boolean
+  pissaIter?: number
+  pissaConvert?: boolean
+  createNewAdapter?: boolean
+  quantizationBit?: number
+  quantizationMethod?: string
+  ropeScaling?: string
+  booster?: string
+}
+
+export type DatasetConfig = {
+  datasetId?: string
+  datasetVolumeId?: string
+  cutoffLen?: number
+  maxSamples?: number
+  valSize?: string
+}
+
+export type TrainingConfig = {
+  numTrainEpochs?: string
+  perDeviceTrainBatchSize?: number
+  gradientAccumulationSteps?: number
+  learningRate?: string
+  lrSchedulerType?: string
+  warmupRatio?: string
+  warmupSteps?: number
+  bf16?: boolean
+  optimizer?: string
+  seed?: number
+  ddpTimeout?: number
+  maxGradNorm?: string
+  neftAlpha?: number
+  deepspeedStage?: string
+  deepspeedOffload?: boolean
+  gradientCheckpointing?: boolean
+  loggingSteps?: number
+  saveSteps?: number
+  outputSubPath?: string
+}
+
+export type DpoConfig = {
+  prefBeta?: string
+  prefLoss?: string
+  prefFtx?: string
+  dpoLabelSmoothing?: string
+  refModel?: string
+}
+
+export type ResourceQuota = {
+  request?: string
+  limit?: string
+}
+
+export type ResourceConfig = {
+  replicas?: number
+  gpuType?: string
+  gpuCount?: number
+  cpu?: ResourceQuota
+  memory?: ResourceQuota
+}
+
+export type FineTuneConfig = {
+  name?: string
+  cluster?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+  createdAt?: GoogleProtobufTimestamp.Timestamp
+  updatedAt?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type CreateWSFineTuneConfigRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+}
+
+export type UpdateWSFineTuneConfigRequest = {
+  workspace?: number
+  cluster?: string
+  name?: string
+  stage?: TrainingStage
+  finetuningType?: FineTuningType
+  model?: ModelConfig
+  lora?: LoraConfig
+  dataset?: DatasetConfig
+  training?: TrainingConfig
+  dpo?: DpoConfig
+  resources?: ResourceConfig
+}
+
+export type GetWSFineTuneConfigRequest = {
+  workspace?: number
+  name?: string
+}
+
+export type ListWSFineTuneConfigsRequest = {
+  workspace?: number
+  cluster?: string
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ListWSFineTuneConfigsResponse = {
+  items?: FineTuneConfig[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type DeleteWSFineTuneConfigRequest = {
+  workspace?: number
+  name?: string
+}
+
+export class WSFineTuneConfigManagement {
+  static CreateWSFineTuneConfig(req: CreateWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<CreateWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+  static UpdateWSFineTuneConfig(req: UpdateWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<UpdateWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static GetWSFineTuneConfig(req: GetWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<GetWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}?${fm.renderURLSearchParams(req, ["workspace", "name"])}`, {...initReq, method: "GET"})
+  }
+  static ListWSFineTuneConfigs(req: ListWSFineTuneConfigsRequest, initReq?: fm.InitReq): Promise<ListWSFineTuneConfigsResponse> {
+    return fm.fetchReq<ListWSFineTuneConfigsRequest, ListWSFineTuneConfigsResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
+  static DeleteWSFineTuneConfig(req: DeleteWSFineTuneConfigRequest, initReq?: fm.InitReq): Promise<FineTuneConfig> {
+    return fm.fetchReq<DeleteWSFineTuneConfigRequest, FineTuneConfig>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/train-configs/${req["name"]}`, {...initReq, method: "DELETE"})
+  }
+}

package/package.json

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