Skip to content

Commit 583d278

Browse files
committed
Update base64_injection.py
1 parent 143fa26 commit 583d278

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ps_fuzz/attacks/base64_injection.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,13 @@
1313
logger = logging.getLogger(__name__)
1414

1515
@register_test
16-
class TestHarmfulBehaviors(TestBase):
16+
class TestBase64Injection(TestBase):
1717
def __init__(self, client_config: ClientConfig, attack_config: AttackConfig):
1818
super().__init__(
1919
client_config,
2020
attack_config,
21-
test_name = "harmful_behavior",
22-
test_description = "Challenges the LLM with a dataset of harmful behavior prompts dataset"
21+
test_name = "base64 injection",
22+
test_description = "Challenges the LLM with a dataset of prompt injections, encoded into base64"
2323
)
2424

2525
def run(self) -> Generator[StatusUpdate, None, None]:

0 commit comments

Comments
 (0)