synthclaw 0.1.3

Lightweight synthetic data generation library/CLI.
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
48
49
50
51
52
53
54
55
56
57
58
59
60

use clap::{Parser, Subcommand};
use tracing_subscriber::{fmt, prelude::*, EnvFilter};

mod cli;

use cli::{datasets, estimate, generate, resume};

#[derive(Parser)]
#[command(name = "synthclaw")]
#[command(author, version, about = "Lightweight synthetic data generation CLI")]
#[command(propagate_version = true)]
struct Cli {
    #[command(subcommand)]
    command: Commands,

    /// Enable verbose output
    #[arg(short, long, global = true)]
    verbose: bool,
}

#[derive(Subcommand)]
enum Commands {
    /// Generate synthetic data from scratch or augment existing datasets
    Generate(generate::GenerateArgs),
    /// Search, explore and preview HuggingFace datasets
    #[command(subcommand)]
    Datasets(datasets::DatasetsCommand),
    /// Resume an interrupted generation from checkpoint
    Resume(resume::ResumeArgs),
    /// Estimate cost for a generation job without running it
    Estimate(estimate::EstimateArgs),
}

fn setup_logging(verbose: bool) {
    let filter = if verbose {
        EnvFilter::new("synth_claw=debug,info")
    } else {
        EnvFilter::new("synth_claw=info,warn")
    };

    tracing_subscriber::registry()
        .with(fmt::layer().with_target(false))
        .with(filter)
        .init();
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let cli = Cli::parse();
    setup_logging(cli.verbose);

    match cli.command {
        Commands::Generate(args) => generate::run(args).await?,
        Commands::Datasets(cmd) => datasets::run(cmd).await?,
        Commands::Resume(args) => resume::run(args).await?,
        Commands::Estimate(args) => estimate::run(args).await?,
    }

    Ok(())
}