portkey-ai--gateway/plugins/index.ts

import { handler as defaultregexMatch } from './default/regexMatch';
import { handler as defaultsentenceCount } from './default/sentenceCount';
import { handler as defaultwordCount } from './default/wordCount';
import { handler as defaultcharacterCount } from './default/characterCount';
import { handler as defaultjsonSchema } from './default/jsonSchema';
import { handler as defaultjsonKeys } from './default/jsonKeys';
import { handler as defaultcontains } from './default/contains';
import { handler as defaultvalidUrls } from './default/validUrls';
import { handler as defaultwebhook } from './default/webhook';
import { handler as defaultlog } from './default/log';
import { handler as defaultcontainsCode } from './default/containsCode';
import { handler as defaultalluppercase } from './default/alluppercase';
import { handler as defaultalllowercase } from './default/alllowercase';
import { handler as defaultendsWith } from './default/endsWith';
import { handler as defaultmodelWhitelist } from './default/modelWhitelist';
import { handler as defaultnotNull } from './default/notNull';
import { handler as qualifireContentModeration } from './qualifire/contentModeration';
import { handler as qualifireGrounding } from './qualifire/grounding';
import { handler as qualifirePolicy } from './qualifire/policy';
import { handler as qualifireToolUseQuality } from './qualifire/toolUseQuality';
import { handler as qualifireHallucinations } from './qualifire/hallucinations';
import { handler as qualifirePii } from './qualifire/pii';
import { handler as qualifirePromptInjections } from './qualifire/promptInjections';
import { handler as defaultaddPrefix } from './default/addPrefix';
import { handler as defaultmodelRules } from './default/modelRules';
import { handler as portkeymoderateContent } from './portkey/moderateContent';
import { handler as portkeylanguage } from './portkey/language';
import { handler as portkeypii } from './portkey/pii';
import { handler as portkeygibberish } from './portkey/gibberish';
import { handler as aporiavalidateProject } from './aporia/validateProject';
import { handler as sydelabssydeguard } from './sydelabs/sydeguard';
import { handler as pillarscanPrompt } from './pillar/scanPrompt';
import { handler as pillarscanResponse } from './pillar/scanResponse';
import { handler as patronusphi } from './patronus/phi';
import { handler as patronuspii } from './patronus/pii';
import { handler as patronusisConcise } from './patronus/isConcise';
import { handler as patronusisHelpful } from './patronus/isHelpful';
import { handler as patronusisPolite } from './patronus/isPolite';
import { handler as patronusnoApologies } from './patronus/noApologies';
import { handler as patronusnoGenderBias } from './patronus/noGenderBias';
import { handler as patronusnoRacialBias } from './patronus/noRacialBias';
import { handler as patronusretrievalAnswerRelevance } from './patronus/retrievalAnswerRelevance';
import { handler as patronustoxicity } from './patronus/toxicity';
import { handler as patronuscustom } from './patronus/custom';
import { mistralGuardrailHandler } from './mistral';
import { handler as pangeatextGuard } from './pangea/textGuard';
import { handler as promptfooPii } from './promptfoo/pii';
import { handler as promptfooHarm } from './promptfoo/harm';
import { handler as promptfooGuard } from './promptfoo/guard';
import { handler as pangeapii } from './pangea/pii';
import { pluginHandler as bedrockHandler } from './bedrock/index';
import { handler as acuvityScan } from './acuvity/scan';
import { handler as lassoclassify } from './lasso/classify';
import { handler as exaonline } from './exa/online';
import { handler as azurePii } from './azure/pii';
import { handler as azureContentSafety } from './azure/contentSafety';
import { handler as promptSecurityProtectPrompt } from './promptsecurity/protectPrompt';
import { handler as promptSecurityProtectResponse } from './promptsecurity/protectResponse';
import { handler as panwPrismaAirsintercept } from './panw-prisma-airs/intercept';
import { handler as defaultjwt } from './default/jwt';
import { handler as defaultrequiredMetadataKeys } from './default/requiredMetadataKeys';
import { handler as walledaiguardrails } from './walledai/walledprotect';
import { handler as defaultregexReplace } from './default/regexReplace';
import { handler as defaultallowedRequestTypes } from './default/allowedRequestTypes';
import { handler as javelinguardrails } from './javelin/guardrails';
import { handler as f5GuardrailsScan } from './f5-guardrails/scan';
import { handler as azureShieldPrompt } from './azure/shieldPrompt';
import { handler as azureProtectedMaterial } from './azure/protectedMaterial';
import { handler as crowdstrikeAidrGuardChatCompletions } from './crowdstrike-aidr/guardChatCompletion';

export const plugins = {
  default: {
    regexMatch: defaultregexMatch,
    sentenceCount: defaultsentenceCount,
    wordCount: defaultwordCount,
    characterCount: defaultcharacterCount,
    jsonSchema: defaultjsonSchema,
    jsonKeys: defaultjsonKeys,
    contains: defaultcontains,
    validUrls: defaultvalidUrls,
    webhook: defaultwebhook,
    log: defaultlog,
    containsCode: defaultcontainsCode,
    alluppercase: defaultalluppercase,
    alllowercase: defaultalllowercase,
    endsWith: defaultendsWith,
    modelWhitelist: defaultmodelWhitelist,
    modelRules: defaultmodelRules,
    jwt: defaultjwt,
    requiredMetadataKeys: defaultrequiredMetadataKeys,
    addPrefix: defaultaddPrefix,
    regexReplace: defaultregexReplace,
    allowedRequestTypes: defaultallowedRequestTypes,
    notNull: defaultnotNull,
  },
  qualifire: {
    contentModeration: qualifireContentModeration,
    grounding: qualifireGrounding,
    policy: qualifirePolicy,
    toolUseQuality: qualifireToolUseQuality,
    hallucinations: qualifireHallucinations,
    pii: qualifirePii,
    promptInjections: qualifirePromptInjections,
  },
  portkey: {
    moderateContent: portkeymoderateContent,
    language: portkeylanguage,
    pii: portkeypii,
    gibberish: portkeygibberish,
  },
  aporia: {
    validateProject: aporiavalidateProject,
  },
  sydelabs: {
    sydeguard: sydelabssydeguard,
  },
  pillar: {
    scanPrompt: pillarscanPrompt,
    scanResponse: pillarscanResponse,
  },
  patronus: {
    phi: patronusphi,
    pii: patronuspii,
    isConcise: patronusisConcise,
    isHelpful: patronusisHelpful,
    isPolite: patronusisPolite,
    noApologies: patronusnoApologies,
    noGenderBias: patronusnoGenderBias,
    noRacialBias: patronusnoRacialBias,
    retrievalAnswerRelevance: patronusretrievalAnswerRelevance,
    toxicity: patronustoxicity,
    custom: patronuscustom,
  },
  mistral: {
    moderateContent: mistralGuardrailHandler,
  },
  pangea: {
    textGuard: pangeatextGuard,
    pii: pangeapii,
  },
  promptfoo: {
    pii: promptfooPii,
    harm: promptfooHarm,
    guard: promptfooGuard,
  },
  bedrock: {
    guard: bedrockHandler,
  },
  acuvity: {
    scan: acuvityScan,
  },
  lasso: {
    classify: lassoclassify,
  },
  exa: {
    online: exaonline,
  },
  azure: {
    pii: azurePii,
    contentSafety: azureContentSafety,
    shieldPrompt: azureShieldPrompt,
    protectedMaterial: azureProtectedMaterial,
  },
  promptsecurity: {
    protectPrompt: promptSecurityProtectPrompt,
    protectResponse: promptSecurityProtectResponse,
  },
  'panw-prisma-airs': {
    intercept: panwPrismaAirsintercept,
  },
  walledai: {
    walledprotect: walledaiguardrails,
  },
  javelin: {
    guardrails: javelinguardrails,
  },
  'f5-guardrails': {
    scan: f5GuardrailsScan,
  },
  'crowdstrike-aidr': {
    guardChatCompletions: crowdstrikeAidrGuardChatCompletions,
  },
};