Review fixes

Author: roh26it

plugins/azure/azure.test.ts

@@ -54,7 +54,6 @@ describe('Azure Plugins', () => {
           messages: [{ role: 'user', content: "hello, I'm a harmless string" }],
         };
         const result = await piiHandler(context, params, 'beforeRequestHook');
-        console.log('result', result);
         expect(result.error).toBeNull();
         expect(result.verdict).toBe(true);
         expect(result.transformed).toBe(false);
@@ -66,7 +65,6 @@ describe('Azure Plugins', () => {
           { ...params, redact: false },
           'beforeRequestHook'
         );
-        console.log('result', result);
         expect(result.error).toBeNull();
         expect(result.verdict).toBe(false);
         expect(result.transformed).toBe(false);

plugins/azure/pii.ts

@@ -134,19 +134,22 @@ export const handler: PluginHandler<{ pii: AzureCredentials }> = async (
 
   try {
     const response = await redact(documents, parameters, pluginOptions);
+    if (!response?.results?.documents) {
+      throw new Error('Invalid response from Azure PII API');
+    }
     data = response.results.documents;
     const containsPII =
       data.length > 0 && data.some((doc: any) => doc.entities.length > 0);
     if (containsPII) {
       verdict = false;
     }
     if (parameters.redact && containsPII) {
+      verdict = true;
       const redactedData = (response.results.documents ?? []).map(
         (doc: any) => doc.redactedText
       );
       setCurrentContentPart(context, eventType, transformedData, redactedData);
       transformed = true;
-      verdict = true;
     }
   } catch (e) {
     error = e;