add one test case field

Reviewed By: SimonWan Differential Revision: D66690954 fbshipit-source-id: 7e6e9682810b3398de868be505d3a65868cc4260
meta-llama · Dec 3, 2024 · 76825e1 · 76825e1
1 parent 0884335
commit 76825e1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/CybersecurityBenchmarks/benchmark/query_llm.py b/CybersecurityBenchmarks/benchmark/query_llm.py
@@ -35,6 +35,7 @@
     "language",
     "speaking_language",
     "false_refusal_type",
+    "caught_by_promptguard",
 ]
 
 
@@ -144,6 +145,7 @@ def process_prompt_and_user_input(
         "risk_category",
         "speaking_language",
         "false_refusal_type",
+        "caught_by_promptguard",
     ]
     for field in test_case_fields:
         if field in test_case: