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
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
70
71
72
73
74

//! Debug script to examine the exact metadata structure after decryption

use anyhow::Result;
use rag_module::*;
use rag_module::services::search_service::EstateSearchOptions;

#[tokio::main]
async fn main() -> Result<()> {
    println!("🔍 DEBUGGING METADATA STRUCTURE AFTER DECRYPTION");
    println!("===============================================\n");
    
    let rag = create_rag_module("./test_data").await?;
    rag.initialize().await?;
    
    let user_id = "test_user_123";
    
    // Search for just 1 resource to debug its structure
    let results = rag.search_service.search_estate_resources(
        "aws resources",
        EstateSearchOptions {
            resource_types: None,
            account_ids: None,
            regions: None,
            services: None,  
            states: None,
            environment: None,
            application: None,
            synced_after: None,
            limit: Some(1), // Just get 1 for debugging
            score_threshold: Some(0.001), 
            include_metadata: true,
            use_anonymous_ids: false,
        },
        None,
        user_id,
    ).await?;
    
    if let Some(result) = results.first() {
        println!("📋 SINGLE RESULT STRUCTURE:");
        println!("==========================");
        println!("{}", serde_json::to_string_pretty(result)?);
        
        println!("\n📋 TOP-LEVEL KEYS:");
        if let Some(obj) = result.as_object() {
            for key in obj.keys() {
                println!("   - {}", key);
            }
        }
        
        // Check metadata structure specifically  
        if let Some(metadata) = result.get("metadata") {
            println!("\n📋 METADATA KEYS:");
            if let Some(meta_obj) = metadata.as_object() {
                for key in meta_obj.keys() {
                    println!("   - metadata.{}", key);
                }
                
                // Check if there's nested metadata
                if let Some(nested_metadata) = meta_obj.get("metadata") {
                    println!("\n📋 NESTED METADATA KEYS:");
                    if let Some(nested_obj) = nested_metadata.as_object() {
                        for key in nested_obj.keys() {
                            println!("   - metadata.metadata.{}", key);
                        }
                    }
                }
            }
        }
    } else {
        println!("❌ No results found!");
    }
    
    Ok(())
}