ruchat 0.1.2

ollama/chroma command-line AI chat tool
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
61
62

use crate::chroma::create_client;
use crate::error::RuChatError;
use anyhow::Result;
use chromadb::collection::ChromaCollection;
use clap::Parser;

/// Command-line arguments for listing Chroma database collections.
///
/// This struct defines the arguments required to list collections
/// in a Chroma database, including the collection name, server address,
/// database name, and an optional authentication token.
#[derive(Parser, Debug, Clone, PartialEq)]
pub struct ChromaLsArgs {
    /// Chroma database collection name.
    #[clap(short, long, default_value = "default")]
    pub(crate) collection: String,

    /// Chroma database server address and port.
    #[clap(short = 'C', long, default_value = "http://localhost:8000")]
    pub(crate) chroma_server: String,

    /// Chroma database name.
    #[clap(short = 'd', long, default_value = "default")]
    pub(crate) chroma_database: String,

    /// Chroma token for authentication.
    #[clap(short = 't', long)]
    pub(crate) chroma_token: Option<String>,
}

/// Lists collections in a Chroma database.
///
/// This function connects to a Chroma database using the provided
/// arguments and lists the details of the specified collection.
///
/// # Parameters
///
/// - `args`: The command-line arguments for listing collections.
///
/// # Returns
///
/// A `Result` indicating success or failure.
pub(crate) async fn chroma_ls(args: &ChromaLsArgs) -> Result<(), RuChatError> {
    // Instantiate a ChromaClient to connect to the Chroma database
    let client = create_client(
        args.chroma_token.as_deref(),
        &args.chroma_server,
        &args.chroma_database,
    )
    .await?;

    // Instantiate a ChromaCollection to perform operations on a collection
    let collection: ChromaCollection = client
        .get_or_create_collection(&args.collection, None)
        .await?;
    eprintln!("Collection Name: {}", collection.name());
    eprintln!("Collection ID: {}", collection.id());
    eprintln!("Collection Metadata: {:?}", collection.metadata());
    eprintln!("Collection Count: {}", collection.count().await?);

    Ok(())
}