minor changes

kritinv · Oct 7, 2024 · ae4da95 · ae4da95
1 parent cf934e1
commit ae4da95
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/deepeval/red_teaming/types.py b/deepeval/red_teaming/types.py
@@ -5,7 +5,6 @@ class AttackEnhancement(Enum):
     GRAY_BOX_ATTACK = "Gray Box Attack"
     PROMPT_INJECTION = "Prompt Injection"
     PROMPT_PROBING = "Prompt Probing"
-    JAILBREAKING = "Jailbreaking"
     JAILBREAK_LINEAR = "Linear Jailbreak"
     JAILBREAK_TREE = "Tree Jailbreak"
     ROT13 = "ROT13 Encoding"

diff --git a/docs/docs/evaluation-red-teaming.mdx b/docs/docs/evaluation-red-teaming.mdx
@@ -104,7 +104,7 @@ print("Vulnerability Scores: ", results)
 There are 2 required parameters and 2 optional parameters when calling the scan method inside `RedTeamer`:
 
 - `target_model`: a [custom LLM model](guides-using-custom-llms) of type `DeepEvalBaseLLM` representing the model you wish to scan.
-- `attacks_per_vulnerability`: An integer specifying the number of adversarial attacks to be generated per vulnerability.
+- `attacks_per_vulnerability`: An integer specifying the maximum number of adversarial attacks to be generated per vulnerability.
 - [Optional] `vulnerabilities`: A list of `Vulnerability` enums specifying the vulnerabilities to be tested. Defaulted to all available `Vulnerability`.
 - [Optional] `attack_enhancements`: A dict of `AttackEnhancement` enum keys specifying the distribution of AttackEnhancements to be used. Defaulted to uniform distribution of all available `AttackEnhancements`.