forge-guardrails 0.1.2

Foundation types for an LLM-agent workflow framework
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
38
39
40
41
42
43
44
45
46
47

//! Private dataset capture and review tool for Forge tool-call verifier rows.

mod agent_logs;
mod assemble;
mod capture;
mod cli;
mod prompts;
mod review;
mod schema;
mod split;
mod stub_tools;
mod validate;

use std::env;

use cli::{parse_args, print_help, Command};

#[tokio::main(flavor = "current_thread")]
async fn main() {
    let command = match parse_args(env::args().skip(1)) {
        Ok(command) => command,
        Err(message) if message == "__help__" => {
            print_help();
            return;
        }
        Err(message) => {
            eprintln!("error: {message}");
            eprintln!("run `forge-dataset --help` for usage");
            std::process::exit(2);
        }
    };

    let result = match command {
        Command::Prompts(cli) => prompts::run(cli),
        Command::Capture(cli) => capture::run(cli).await,
        Command::Review(cli) => review::run(*cli).await,
        Command::AgentLogs(cli) => agent_logs::run(*cli),
        Command::Assemble(cli) => assemble::run(cli),
        Command::Split(cli) => split::run(cli),
        Command::Validate(cli) => validate::run(cli),
    };

    if let Err(err) = result {
        eprintln!("error: {err}");
        std::process::exit(1);
    }
}