@daocloud-proto/hydra 0.0.0-dev-5297a7b6 → 0.0.0-dev-96adbe0d

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

@@ -0,0 +1,179 @@
+/* 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 GoogleApiHttpbody from "../../../google/api/httpbody.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum BlockStrategy {
+  BLOCK_STRATEGY_UNSPECIFIED = "BLOCK_STRATEGY_UNSPECIFIED",
+  BLOCK_STRATEGY_ANY = "BLOCK_STRATEGY_ANY",
+  BLOCK_STRATEGY_ALL = "BLOCK_STRATEGY_ALL",
+}
+
+export enum MatchStrategy {
+  MATCH_STRATEGY_UNSPECIFIED = "MATCH_STRATEGY_UNSPECIFIED",
+  MATCH_STRATEGY_ANY = "MATCH_STRATEGY_ANY",
+  MATCH_STRATEGY_ALL = "MATCH_STRATEGY_ALL",
+}
+
+export enum PluginAction {
+  PLUGIN_ACTION_UNSPECIFIED = "PLUGIN_ACTION_UNSPECIFIED",
+  PLUGIN_ACTION_BLOCK = "PLUGIN_ACTION_BLOCK",
+  PLUGIN_ACTION_LOG = "PLUGIN_ACTION_LOG",
+  PLUGIN_ACTION_REPLACE_RANDOM = "PLUGIN_ACTION_REPLACE_RANDOM",
+  PLUGIN_ACTION_REPLACE_MASK = "PLUGIN_ACTION_REPLACE_MASK",
+}
+
+export enum GuardrailsPlugin {
+  GUARDRAILS_PLUGIN_UNSPECIFIED = "GUARDRAILS_PLUGIN_UNSPECIFIED",
+  GUARDRAILS_PLUGIN_CONTAINS = "GUARDRAILS_PLUGIN_CONTAINS",
+  GUARDRAILS_PLUGIN_REGEX = "GUARDRAILS_PLUGIN_REGEX",
+  GUARDRAILS_PLUGIN_JSON_SCHEMA = "GUARDRAILS_PLUGIN_JSON_SCHEMA",
+  GUARDRAILS_PLUGIN_JWT = "GUARDRAILS_PLUGIN_JWT",
+  GUARDRAILS_PLUGIN_MODEL_WHITELIST = "GUARDRAILS_PLUGIN_MODEL_WHITELIST",
+}
+
+export enum GuardrailsLevel {
+  GUARDRAILS_LEVEL_UNSPECIFIED = "GUARDRAILS_LEVEL_UNSPECIFIED",
+  GUARDRAILS_LEVEL_HIGH = "GUARDRAILS_LEVEL_HIGH",
+  GUARDRAILS_LEVEL_MEDIUM = "GUARDRAILS_LEVEL_MEDIUM",
+  GUARDRAILS_LEVEL_LOW = "GUARDRAILS_LEVEL_LOW",
+}
+
+export enum ExportType {
+  EXPORT_TYPE_UNSPECIFIED = "EXPORT_TYPE_UNSPECIFIED",
+  EXPORT_TYPE_CSV = "EXPORT_TYPE_CSV",
+  EXPORT_TYPE_EXCEL = "EXPORT_TYPE_EXCEL",
+}
+
+export type AIGuardrailsConfig = {
+  blockStrategy?: BlockStrategy
+  denyMessage?: string
+  regexMatch?: RegexMatchPlugin
+  contains?: ContainsPlugin
+  jsonSchema?: JsonSchemaPlugin
+  jwt?: JWTPlugin
+  modelWhitelist?: ModelWhitelistPlugin
+  cluster?: string
+  creationTime?: GoogleProtobufTimestamp.Timestamp
+  updateTime?: GoogleProtobufTimestamp.Timestamp
+}
+
+export type RegexMatchPlugin = {
+  enabled?: boolean
+  patterns?: string[]
+  matchStrategy?: MatchStrategy
+  action?: PluginAction
+  responses?: string[]
+}
+
+export type ContainsPlugin = {
+  enabled?: boolean
+  words?: string[]
+  matchStrategy?: MatchStrategy
+  action?: PluginAction
+  responses?: string[]
+}
+
+export type JsonSchemaPlugin = {
+  enabled?: boolean
+  action?: PluginAction
+}
+
+export type JWTPlugin = {
+  enabled?: boolean
+  action?: PluginAction
+  headerKey?: string
+}
+
+export type ModelWhitelistPlugin = {
+  enabled?: boolean
+  models?: string[]
+  isBlocklist?: boolean
+  action?: PluginAction
+}
+
+export type GetAIGuardrailsConfigRequest = {
+  cluster?: string
+}
+
+export type UpdateAIGuardrailsConfigRequest = {
+  cluster?: string
+  config?: AIGuardrailsConfig
+}
+
+export type ListAIGuardrailsConfigRequest = {
+  page?: HydraCommonCommon.Pagination
+  cluster?: string
+}
+
+export type ListAIGuardrailsConfigResponse = {
+  items?: AIGuardrailsConfig[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type QueryAIAuditLogRequest = {
+  page?: HydraCommonCommon.Pagination
+  startTime?: GoogleProtobufTimestamp.Timestamp
+  endTime?: GoogleProtobufTimestamp.Timestamp
+  clusterName?: string
+  operation?: PluginAction
+  pluginId?: GuardrailsPlugin
+  level?: GuardrailsLevel
+  routeName?: string
+}
+
+export type AIAuditLog = {
+  timestamp?: GoogleProtobufTimestamp.Timestamp
+  duration?: string
+  clusterName?: string
+  routeName?: string
+  operation?: PluginAction
+  pluginId?: GuardrailsPlugin
+  level?: GuardrailsLevel
+  reason?: string
+  denyMessage?: string
+  matchSource?: string
+  functionName?: string
+  functionArgs?: string
+  matchRule?: string
+}
+
+export type QueryAIAuditLogResponse = {
+  items?: AIAuditLog[]
+  page?: HydraCommonCommon.Pagination
+}
+
+export type ExportAIAuditLogRequest = {
+  startTime?: GoogleProtobufTimestamp.Timestamp
+  endTime?: GoogleProtobufTimestamp.Timestamp
+  clusterName?: string
+  operation?: PluginAction
+  pluginId?: GuardrailsPlugin
+  level?: GuardrailsLevel
+  exportType?: ExportType
+  routeName?: string
+}
+
+export class AIGuardrailsService {
+  static GetAIGuardrailsConfig(req: GetAIGuardrailsConfigRequest, initReq?: fm.InitReq): Promise<AIGuardrailsConfig> {
+    return fm.fetchReq<GetAIGuardrailsConfigRequest, AIGuardrailsConfig>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails/${req["cluster"]}?${fm.renderURLSearchParams(req, ["cluster"])}`, {...initReq, method: "GET"})
+  }
+  static ListAIGuardrailsConfig(req: ListAIGuardrailsConfigRequest, initReq?: fm.InitReq): Promise<ListAIGuardrailsConfigResponse> {
+    return fm.fetchReq<ListAIGuardrailsConfigRequest, ListAIGuardrailsConfigResponse>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static UpdateAIGuardrailsConfig(req: UpdateAIGuardrailsConfigRequest, initReq?: fm.InitReq): Promise<AIGuardrailsConfig> {
+    return fm.fetchReq<UpdateAIGuardrailsConfigRequest, AIGuardrailsConfig>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+  static QueryAIAuditLog(req: QueryAIAuditLogRequest, initReq?: fm.InitReq): Promise<QueryAIAuditLogResponse> {
+    return fm.fetchReq<QueryAIAuditLogRequest, QueryAIAuditLogResponse>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails-logs?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static ExportAIAuditLog(req: ExportAIAuditLogRequest, initReq?: fm.InitReq): Promise<GoogleApiHttpbody.HttpBody> {
+    return fm.fetchReq<ExportAIAuditLogRequest, GoogleApiHttpbody.HttpBody>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails-logs/export`, {...initReq, method: "POST", body: JSON.stringify(req, fm.replacer)})
+  }
+}

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

@@ -102,25 +102,26 @@ export type DatasetConfig = {
   datasetVolumeId?: string
   cutoffLen?: number
   maxSamples?: number
-  valSize?: string
+  valSize?: number
   datasetName?: string
+  datasetTag?: string
 }
 
 export type TrainingConfig = {
-  numTrainEpochs?: string
+  numTrainEpochs?: number
   perDeviceTrainBatchSize?: number
   gradientAccumulationSteps?: number
-  learningRate?: string
+  learningRate?: number
   lrSchedulerType?: LrSchedulerType
-  warmupRatio?: string
+  warmupRatio?: number
   warmupSteps?: number
   computeType?: ComputeType
   optimizer?: OptimizerType
   seed?: number
   ddpTimeout?: number
-  maxGradNorm?: string
+  maxGradNorm?: number
   neftAlpha?: number
-  deepspeedStage?: string
+  deepspeedStage?: number
   deepspeedOffload?: boolean
   gradientCheckpointing?: boolean
   loggingSteps?: number
@@ -129,10 +130,10 @@ export type TrainingConfig = {
 }
 
 export type DpoConfig = {
-  prefBeta?: string
+  prefBeta?: number
   prefLoss?: string
-  prefFtx?: string
-  dpoLabelSmoothing?: string
+  prefFtx?: number
+  dpoLabelSmoothing?: number
   refModel?: string
 }
 
@@ -163,6 +164,7 @@ export type FineTuneConfig = {
   createdAt?: GoogleProtobufTimestamp.Timestamp
   updatedAt?: GoogleProtobufTimestamp.Timestamp
   modelSource?: boolean
+  workspace?: number
 }
 
 export type CreateWSFineTuneConfigRequest = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.0.0-dev-5297a7b6",
+  "version": "v0.0.0-dev-96adbe0d",
   "description": "",
   "author": "",
   "license": "ISC"