ragcli 0.1.0

CLI for local RAG
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

mod agent;
mod app;
mod citation;
mod cli;
mod commands;
mod config;
mod fsutil;
mod graph;
mod ingest;
mod jsonutil;
mod models;
mod query;
mod retrieval;
mod rewrite;
mod source_kind;
mod store;
mod telemetry;
#[cfg(test)]
mod test_support;
mod ui;

use anyhow::Result;
use clap::Parser;
use cli::{Cli, Command};

fn main() -> Result<()> {
    let cli = Cli::parse();
    let runtime = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()?;

    let mut telemetry = {
        let _guard = runtime.enter();
        match telemetry::init() {
            Ok(telemetry) => telemetry,
            Err(err) if matches!(cli.command, Command::Doctor { .. }) => {
                eprintln!("telemetry init error: {err}");
                telemetry::TelemetryGuard::disabled()
            }
            Err(err) => return Err(err),
        }
    };

    let result = runtime.block_on(app::run(cli));

    if let Err(err) = {
        let _guard = runtime.enter();
        telemetry.shutdown()
    } {
        eprintln!("telemetry shutdown error: {err}");
    }

    drop(runtime);
    result
}