rag-module 0.6.7

Enterprise RAG module with chat context storage, vector search, session management, and model downloading. Rust implementation with Node.js compatibility.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use anyhow::Result;
use rag_module::RagModule;

#[tokio::main]
async fn main() -> Result<()> {
    println!("Testing get_all_contexts_for_user function\n");

    // Initialize RAG module
    let mut rag_module = RagModule::new("/Users/aparnapitchikala/Library/Application Support/Escher").await?;
    rag_module.initialize().await?;

    // Get all contexts
    let user_id = "148854b8-9091-70e0-ed07-40553c514535";
    println!("Fetching all contexts for user: {}...", user_id);
    let contexts = rag_module.get_all_contexts(user_id).await?;

    // Display results

    // Pretty print as JSON
    println!("\nJSON Output:");
    println!("{}", serde_json::to_string_pretty(&contexts)?);

    Ok(())
}