fix: update types, guardrail functions

plugins/index.ts

@@ -33,19 +33,6 @@ import { handler as patronusretrievalAnswerRelevance } from './patronus/retrieva
 import { handler as patronustoxicity } from './patronus/toxicity';
 import { handler as patronuscustom } from './patronus/custom';
 import { mistralGuardrailHandler } from './mistral';
-import { PluginHandler } from './types';
-
-const mistralGuardCategories = [
-  'sexual',
-  'hate_and_discrimination',
-  'violence_and_threats',
-  'dangerous_and_criminal_content',
-  'selfharm',
-  'health',
-  'financial',
-  'law',
-  'pii',
-];
 import { handler as pangeatextGuard } from './pangea/textGuard';
 
 export const plugins = {
@@ -95,13 +82,9 @@ export const plugins = {
     toxicity: patronustoxicity,
     custom: patronuscustom,
   },
-  mistral: mistralGuardCategories.reduce(
-    (config, category) => {
-      config[category] = mistralGuardrailHandler;
-      return config;
-    },
-    {} as Record<string, PluginHandler>
-  ),
+  mistral: {
+    moderateContent: mistralGuardrailHandler,
+  },
   pangea: {
     textGuard: pangeatextGuard,
   },

plugins/types.ts

@@ -24,7 +24,5 @@ export type PluginHandler = (
   eventType: HookEventType,
   options?: {
     env: Record<string, any>;
-  },
-  // Handler function, useful in cases for a provider with multiple guardrails ex: mistral
-  fn?: string
+  }
 ) => Promise<PluginHandlerResponse>;

src/middlewares/hooks/index.ts

@@ -261,8 +261,7 @@ export class HooksManager {
         context,
         check.parameters,
         eventType,
-        options,
-        fn
+        options
       );
       return {
         ...result,