pangea-node-sdk: let action be a string

pangeacyber · Feb 11, 2025 · 805948b · 805948b
1 parent 85d292b
commit 805948b
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/packages/pangea-node-sdk/src/types.ts b/packages/pangea-node-sdk/src/types.ts
@@ -294,23 +294,16 @@ export namespace Audit {
 }
 
 export namespace AIGuard {
-  type DetectorAction =
-    | "detected"
-    | "redacted"
-    | "defanged"
-    | "reported"
-    | "blocked";
-
   export interface MaliciousEntity {
     raw?: Record<string, unknown>;
-    action: DetectorAction;
+    action: string;
     start_pos?: number;
     type: string;
     value: string;
   }
 
   export interface PIIEntity {
-    action: DetectorAction;
+    action: string;
     start_pos?: number;
     type: string;
     value: string;
@@ -340,7 +333,7 @@ export namespace AIGuard {
     /** Result of the recipe analyzing and input prompt. */
     detectors: {
       prompt_injection: Detector<{
-        action: DetectorAction;
+        action: string;
         analyzer_responses: { analyzer: string; confidence: number }[];
       }>;
       pii_entity?: Detector<{ entities: PIIEntity[] }>;