File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 1414from .interactive_mode import interactive_shell
1515from .prompt_injection_fuzzer import run_fuzzer
1616from .logo import print_logo
17+ from .util import wrap_text
1718
1819# Initialize colorama
1920colorama .init ()
@@ -43,8 +44,8 @@ def main():
4344 attack_config = AttackConfig (FakeChatClient (), 1 )
4445 tests = instantiate_tests (client_config , attack_config )
4546 print ("Available attacks:" )
46- for test_name , test_description in [(cls .test_name , cls .test_description ) for cls in tests ]:
47- print (f" { BRIGHT } { test_name } { RESET } : { test_description } " )
47+ for test_name , test_description in sorted ( [(cls .test_name , cls .test_description ) for cls in tests ]) :
48+ print (f"{ BRIGHT } { test_name } { RESET } :\n { wrap_text ( test_description ) } " )
4849 sys .exit (0 )
4950
5051 # Setup debug level (default 1)
You can’t perform that action at this time.
0 commit comments