INNER CODE UNIT · Python

main

prompt-security/ps-fuzz · ps_fuzz/cli.py:28

def main():
    # Print the logo
    print_logo()

    # Parse command line arguments
    args = parse_cmdline_args()

    # Execute immediate commands
    if args.list_providers:
        print("Available providers:")
        for provider_name, provider_info in get_langchain_chat_models_info().items():
            print(f"  {BRIGHT}{provider_name}{RESET}: {provider_info.short_doc}")
        sys.exit(0)

    if args.list_attacks:
        client_config = ClientConfig(FakeChatClient(), [])
        attack_config = AttackConfig(FakeChatClient(), 1, embedding_config=None)
        tests = instantiate_tests(client_config, attack_config,[],True)

View source record →