Merge pull request #1263 from indranil786/main

Fix : WalledAI guardrail parameters

Author: VisargD

plugins/walledai/guardrails.ts

@@ -8,6 +8,18 @@ import { post, getText, getCurrentContentPart } from '../utils';
 
 const API_URL = 'https://services.walled.ai/v1/guardrail/moderate';
 
+const DEFAULT_PII_LIST = [
+  "Person's Name",
+  'Address',
+  'Email Id',
+  'Contact No',
+  'Date Of Birth',
+  'Unique Id',
+  'Financial Data',
+];
+
+const DEFAULT_GREETINGS_LIST = ['Casual & Friendly', 'Professional & Polite'];
+
 export const handler: PluginHandler = async (
   context: PluginContext,
   parameters: PluginParameters,
@@ -40,13 +52,15 @@ export const handler: PluginHandler = async (
     text: text,
     text_type: parameters.text_type || 'prompt',
     generic_safety_check: parameters.generic_safety_check ?? true,
-    greetings_list: parameters.greetings_list || ['generalgreetings'],
+    greetings_list: parameters.greetings_list || DEFAULT_GREETINGS_LIST,
+    pii_list: parameters.pii_list || DEFAULT_PII_LIST,
+    compliance_list: parameters.compliance_list || [],
   };
   // Prepare headers
   const requestOptions = {
     headers: {
       'Content-Type': 'application/json',
-      Authorization: `Bearer ${parameters.credentials.apiKey}`, // Uncomment if API key is required
+      Authorization: `Bearer ${parameters.credentials.apiKey}`,
     },
   };
 

plugins/walledai/manifest.json

@@ -60,9 +60,9 @@
               }
             ],
             "items": {
-              "type": "string",
-              "default": ["generalgreetings"]
-            }
+              "type": "string"
+            },
+            "default": ["Casual & Friendly", "Professional & Polite"]
           },
           "pii_list": {
             "type": "array",
@@ -84,7 +84,16 @@
                 "Unique Id",
                 "Financial Data"
               ]
-            }
+            },
+            "default": [
+              "Person's Name",
+              "Address",
+              "Email Id",
+              "Contact No",
+              "Date Of Birth",
+              "Unique Id",
+              "Financial Data"
+            ]
           },
           "compliance_list": {
             "type": "array",
@@ -97,7 +106,8 @@
             ],
             "items": {
               "type": "string"
-            }
+            },
+            "default": []
           }
         }
       }

plugins/walledai/walledai.test.ts

@@ -11,9 +11,9 @@ describe('WalledAI Guardrail Plugin Handler (integration)', () => {
     credentials: testCreds,
     text_type: 'prompt',
     generic_safety_check: true,
-    greetings_list: ['generalgreetings'],
+    greetings_list: ['Casual & Friendly', 'Professional & Polite'],
     pii_list: ["Person's Name", 'Address'],
-    compliance_list: [],
+    compliance_list: ['questions on medicine'],
   };
 
   const makeContext = (text: string): PluginContext => ({
@@ -80,4 +80,23 @@ describe('WalledAI Guardrail Plugin Handler (integration)', () => {
     expect(result.verdict).toBe(true);
     expect(result.error).toBeNull();
   });
+
+  it('handles compliance_list parameter', async () => {
+    const context = makeContext('This is a test for compliance.');
+
+    const paramsWithCompliance: PluginParameters = {
+      ...baseParams,
+      compliance_list: ['GDPR', 'PCI-DSS'],
+    };
+
+    const result = await handler(
+      context,
+      paramsWithCompliance,
+      'beforeRequestHook'
+    );
+
+    expect(result.error).toBeNull();
+    expect(result.data).toBeDefined();
+    // Optionally, check if compliance_list was respected in the response if API supports it
+  });
 });