Merge pull request #1223 from Portkey-AI/chore/handle-content-array-in-pk-plugins

chore: support array content values in portkey plugin checks

Author: VisargD

plugins/portkey/gibberish.ts

@@ -4,7 +4,7 @@ import {
   PluginHandler,
   PluginParameters,
 } from '../types';
-import { getText } from '../utils';
+import { getCurrentContentPart } from '../utils';
 import { PORTKEY_ENDPOINTS, fetchPortkey } from './globals';
 
 export const handler: PluginHandler = async (
@@ -16,9 +16,17 @@ export const handler: PluginHandler = async (
   let error = null;
   let verdict = false;
   let data: any = null;
-
+  let text = '';
   try {
-    const text = getText(context, eventType);
+    const { content, textArray } = getCurrentContentPart(context, eventType);
+    if (!content) {
+      return {
+        error: { message: 'request or response json is empty' },
+        verdict: true,
+        data: null,
+      };
+    }
+    text = textArray.filter((text) => text).join('\n');
     const not = parameters.not || false;
 
     const response: any = await fetchPortkey(
@@ -47,7 +55,6 @@ export const handler: PluginHandler = async (
     };
   } catch (e) {
     error = e as Error;
-    const text = getText(context, eventType);
     data = {
       explanation: `An error occurred while checking for gibberish: ${error.message}`,
       not: parameters.not || false,

plugins/portkey/language.ts

@@ -4,7 +4,7 @@ import {
   PluginHandler,
   PluginParameters,
 } from '../types';
-import { getText } from '../utils';
+import { getCurrentContentPart } from '../utils';
 import { PORTKEY_ENDPOINTS, fetchPortkey } from './globals';
 
 export const handler: PluginHandler = async (
@@ -16,9 +16,17 @@ export const handler: PluginHandler = async (
   let error = null;
   let verdict = false;
   let data: any = null;
-
+  let text = '';
   try {
-    const text = getText(context, eventType);
+    const { content, textArray } = getCurrentContentPart(context, eventType);
+    if (!content) {
+      return {
+        error: { message: 'request or response json is empty' },
+        verdict: true,
+        data: null,
+      };
+    }
+    text = textArray.filter((text) => text).join('\n');
     const languages = parameters.language;
     const not = parameters.not || false;
 
@@ -51,7 +59,6 @@ export const handler: PluginHandler = async (
     };
   } catch (e) {
     error = e as Error;
-    const text = getText(context, eventType);
     data = {
       explanation: `An error occurred while checking language: ${error.message}`,
       not: parameters.not || false,

plugins/portkey/moderateContent.ts

@@ -4,7 +4,7 @@ import {
   PluginHandler,
   PluginParameters,
 } from '../types';
-import { getText } from '../utils';
+import { getCurrentContentPart } from '../utils';
 import { PORTKEY_ENDPOINTS, fetchPortkey } from './globals';
 
 export const handler: PluginHandler = async (
@@ -16,9 +16,17 @@ export const handler: PluginHandler = async (
   let error = null;
   let verdict = false;
   let data: any = null;
-
+  let text = '';
   try {
-    const text = getText(context, eventType);
+    const { content, textArray } = getCurrentContentPart(context, eventType);
+    if (!content) {
+      return {
+        error: { message: 'request or response json is empty' },
+        verdict: true,
+        data: null,
+      };
+    }
+    text = textArray.filter((text) => text).join('\n');
     const categories = parameters.categories;
     const not = parameters.not || false;
 
@@ -59,7 +67,6 @@ export const handler: PluginHandler = async (
     };
   } catch (e) {
     error = e as Error;
-    const text = getText(context, eventType);
     data = {
       explanation: `An error occurred during content moderation: ${error.message}`,
       not: parameters.not || false,