bigrag 0.1.1

Rust client for bigRAG — a self-hostable RAG platform
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
63
64
65
66
67
68
69

//! File upload patterns: path, bytes, batch upload.
//!
//! Run with: `cargo run --example file_upload`
//! Requires a running bigRAG server at localhost:6100.

use bigrag::types::CreateCollectionBody;
use bigrag::{BigRag, FileInput};

#[tokio::main]
async fn main() -> Result<(), bigrag::BigRagError> {
    let client = BigRag::from_env()?;

    // Create a collection for testing
    client
        .collections()
        .create(CreateCollectionBody {
            name: "upload_test".into(),
            ..Default::default()
        })
        .await?;

    // Upload from a file path (simplest)
    let doc = client
        .documents()
        .upload("upload_test", "README.md", None)
        .await?;
    println!("From path: {} ({})", doc.filename, doc.status);

    // Upload from bytes in memory
    let content = b"# Hello\n\nThis is a test document.".to_vec();
    let doc = client
        .documents()
        .upload("upload_test", (content, "hello.md"), None)
        .await?;
    println!("From bytes: {} ({})", doc.filename, doc.status);

    // Upload with metadata
    let meta = serde_json::json!({"department": "engineering", "priority": "high"});
    let doc = client
        .documents()
        .upload("upload_test", "Cargo.toml", Some(meta))
        .await?;
    println!("With metadata: {} ({})", doc.filename, doc.status);

    // Batch upload
    let files = vec![
        FileInput::from("README.md"),
        FileInput::Bytes {
            data: b"Another doc".to_vec(),
            name: "another.txt".into(),
        },
    ];
    let batch = client
        .documents()
        .batch_upload("upload_test", files, None)
        .await?;
    println!("Batch uploaded: {} documents", batch.total);

    // Using the scoped collection client
    let col = client.collection("upload_test");
    let docs = col.list_documents(None).await?;
    println!("\nTotal documents in collection: {}", docs.total);

    // Cleanup
    client.collections().delete("upload_test").await?;
    println!("Cleaned up.");

    Ok(())
}