jammi-cli 0.5.5

Command-line interface for Jammi AI
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

use clap::Subcommand;
use jammi_ai::session::InferenceSession;
use jammi_engine::config::JammiConfig;
use jammi_engine::source::{FileFormat, SourceConnection, SourceType};

#[derive(Subcommand)]
pub enum SourceAction {
    /// List registered sources
    List,
    /// Add a file-shaped data source. `--url` accepts a local path
    /// (parsed into `file://...`) or any storage URL the build was
    /// compiled with: `s3://bucket/key`, `gs://bucket/key`,
    /// `azure://container/blob`.
    Add {
        /// Source name
        name: String,
        /// Storage URL or local path
        #[arg(long)]
        url: String,
        /// File format (parquet, csv, json)
        #[arg(long)]
        format: String,
    },
}

pub async fn run(
    config: JammiConfig,
    tenant: Option<jammi_engine::TenantId>,
    action: SourceAction,
) -> Result<(), Box<dyn std::error::Error>> {
    match action {
        SourceAction::List => {
            let session = InferenceSession::new(config).await?;
            if let Some(t) = tenant {
                session.bind_tenant(t);
            }
            let sources = session.catalog().list_sources().await?;
            if sources.is_empty() {
                println!("No sources registered.");
            } else {
                println!("{:<20} {:<15} Created", "Name", "Type");
                println!("{}", "-".repeat(55));
                for s in sources {
                    println!(
                        "{:<20} {:<15} {}",
                        s.source_id,
                        format!("{:?}", s.source_type),
                        s.created_at
                    );
                }
            }
        }
        SourceAction::Add { name, url, format } => {
            let file_format: FileFormat = format
                .parse()
                .map_err(|e: jammi_engine::error::JammiError| e.to_string())?;
            let connection = SourceConnection::parse(&url, file_format)?;
            let session = InferenceSession::new(config).await?;
            if let Some(t) = tenant {
                session.bind_tenant(t);
            }
            session
                .add_source(&name, SourceType::File, connection)
                .await?;
            println!("Source '{name}' added.");
        }
    }
    Ok(())
}