rag-module 0.1.4

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
25
26
27
28

use rag_module::services::EmbeddingService;
use rag_module::types::EmbeddingConfig;
use std::path::PathBuf;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("🧪 Testing embedding generation directly...");
    
    let config = EmbeddingConfig::default();
    let models_path = PathBuf::from("./test-models");
    
    let service = EmbeddingService::new(&config, &models_path).await?;
    service.initialize().await?;
    
    println!("🔄 Generating test embedding...");
    let result = service.generate_embedding("test text").await;
    
    match result {
        Ok(embedding) => {
            println!("✅ Generated embedding with {} dimensions", embedding.len());
        }
        Err(e) => {
            println!("❌ Failed to generate embedding: {}", e);
        }
    }
    
    Ok(())
}