adversaria 0.1.0

Adversarial Testing Harness for Large Language Models
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

mod commands;

use anyhow::Result;
use clap::{Parser, Subcommand};
use commands::{list::ListCommand, report::ReportCommand, run::RunCommand};

#[derive(Parser)]
#[command(name = "adversaria")]
#[command(author = "Adversaria Contributors")]
#[command(version = "0.1.0")]
#[command(about = "Adversarial Testing Harness for Large Language Models", long_about = None)]
pub struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    #[command(about = "Run attack suites against a model")]
    Run(RunCommand),

    #[command(about = "List available attack suites")]
    List(ListCommand),

    #[command(about = "Generate or view reports from previous runs")]
    Report(ReportCommand),
}

impl Cli {
    pub async fn execute(self) -> Result<()> {
        match self.command {
            Commands::Run(cmd) => cmd.execute().await,
            Commands::List(cmd) => cmd.execute().await,
            Commands::Report(cmd) => cmd.execute().await,
        }
    }
}