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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
use std::path::PathBuf;
use clap::{Parser, Subcommand};
#[derive(Parser, Debug, Clone)]
#[clap(author, about, version)]
pub struct Args {
/// Optional path to overwrite the config
#[arg(short, long, default_value = "kwaak.toml")]
pub config_path: PathBuf,
/// Skip initial indexing and splash screen
#[arg(short, long, default_value_t = false)]
pub skip_indexing: bool,
/// Allow running with a dirty git directory
#[arg(long, default_value_t = false)]
pub allow_dirty: bool,
/// Subcommands corresponding to each mode
#[clap(subcommand)]
pub command: Option<Commands>,
}
#[derive(Subcommand, Debug, Clone, Default)]
pub enum Commands {
/// Initializes a new kwaak project in the current directory
Init {
#[arg(long, default_value_t = false)]
dry_run: bool,
/// Output to a specific file
#[arg(long)]
file: Option<PathBuf>,
},
/// Start the TUI (default)
#[default]
Tui,
/// Query the indexed project
Query {
#[arg(short, long)]
query: String,
},
/// Run an agent directly
RunAgent {
#[arg(short, long)]
initial_message: String,
},
/// Index the current project
Index,
/// Tests a tool
TestTool {
tool_name: String,
tool_args: Option<String>,
},
/// Print the configuration and exit
PrintConfig,
/// Clear the index and cache for this project and exit
ClearCache,
/// Run evaluations
Eval {
#[command(subcommand)]
eval_type: EvalCommands,
},
}
#[derive(Subcommand, Debug, Clone)]
pub enum EvalCommands {
/// Run the patch evaluation
Patch {
/// Number of iterations to run
#[arg(short, long, default_value_t = 1)]
iterations: u32,
},
}