From 91b99f77e92be34b7d5a2a581d0ab382fa446168 Mon Sep 17 00:00:00 2001
From: Lauren Deason <laurendeason@meta.com>
Date: Thu, 5 Dec 2024 13:22:50 -0800
Subject: [PATCH] Add Vishing Compliance Benchmark

Reviewed By: adam612, SimonWan

Differential Revision: D66729563

fbshipit-source-id: 05179fe9bf94711e2d91644a97eb3a46c7b0ca3a
---
 CybersecurityBenchmarks/benchmark/query_llm.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/CybersecurityBenchmarks/benchmark/query_llm.py b/CybersecurityBenchmarks/benchmark/query_llm.py
index f7b08a1e..8d423ab9 100644
--- a/CybersecurityBenchmarks/benchmark/query_llm.py
+++ b/CybersecurityBenchmarks/benchmark/query_llm.py
@@ -36,6 +36,11 @@
     "speaking_language",
     "false_refusal_type",
     "caught_by_promptguard",
+    "category",
+    "subcategories",
+    "voice_tone",
+    "label",
+    "sample_id",
 ]