initial commit

src/cli.rs

@@ -0,0 +1,62 @@
+//! Command-line interface definitions.
+
+use std::path::PathBuf;
+
+use clap::{Args, Parser, Subcommand};
+
+/// Main CLI configuration.
+#[derive(Parser)]
+#[command(version, about, long_about = None)]
+#[command(propagate_version = true)]
+pub struct Cli {
+    /// Where to read the database connection string from ('-' is stdin).
+    #[arg(short, long, default_value = "-")]
+    pub db_url: String,
+
+    /// Path to content tokenizer.
+    #[arg(short, long, default_value = "./tokenizer.json")]
+    pub tokenizer_path: PathBuf,
+
+    /// Path to ONNX model used for embeddings..
+    #[arg(short, long, default_value = "./model.onnx")]
+    pub model_path: PathBuf,
+
+    /// Maximum text chunk size when splitting up content. Documents are not updated if chunk size
+    /// changes.
+    #[arg(short, long, default_value = "500")]
+    pub chunk_size: usize,
+
+    #[command(subcommand)]
+    pub command: Commands,
+}
+
+/// Configuration for the ask command.
+#[derive(Args, Clone)]
+pub struct AskConfig {
+    /// Query to run.
+    #[arg(short, long)]
+    pub query: String,
+
+    /// Maximum number of results.
+    #[arg(short, long, default_value = "5")]
+    pub limit: usize,
+
+    /// Path to reranking model.
+    #[arg(short, long, default_value = "./reranking.onnx")]
+    pub reranking_path: PathBuf,
+}
+
+/// Configuration for the scan command.
+#[derive(Args, Clone)]
+pub struct ScanConfig {
+    /// Root directory of calibre library.
+    #[arg(short, long)]
+    pub library_path: PathBuf,
+}
+
+/// Available CLI commands.
+#[derive(Subcommand)]
+pub enum Commands {
+    Ask(AskConfig),
+    Scan(ScanConfig),
+}