project-rag 0.1.0

RAG-based codebase indexing and semantic search - dual purpose library and MCP server
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

/// Simple integration tests for basic server functionality
use anyhow::Result;
use project_rag::config::Config;
use project_rag::mcp_server::RagMcpServer;
use tempfile::TempDir;

#[tokio::test]
async fn test_server_creation_with_config() -> Result<()> {
    let db_dir = TempDir::new()?;
    let cache_dir = TempDir::new()?;

    let mut config = Config::default();
    config.vector_db.lancedb_path = db_dir.path().to_path_buf();
    config.cache.hash_cache_path = cache_dir.path().join("hash_cache.json");
    config.cache.git_cache_path = cache_dir.path().join("git_cache.json");

    let server = RagMcpServer::with_config(config).await?;

    // Verify server was created successfully
    assert!(std::mem::size_of_val(&server) > 0);

    Ok(())
}

#[tokio::test]
async fn test_server_creation_with_defaults() -> Result<()> {
    // This should work with default configuration
    let server = RagMcpServer::new().await;

    // Server creation should succeed
    assert!(server.is_ok());

    Ok(())
}

#[tokio::test]
async fn test_path_normalization() -> Result<()> {
    // Test path normalization with current directory
    let normalized = RagMcpServer::normalize_path(".")?;
    assert!(normalized.len() > 1);
    assert!(normalized.starts_with('/') || normalized.chars().nth(1) == Some(':'));

    Ok(())
}

#[tokio::test]
async fn test_config_with_custom_batch_size() -> Result<()> {
    let db_dir = TempDir::new()?;
    let cache_dir = TempDir::new()?;

    let mut config = Config::default();
    config.vector_db.lancedb_path = db_dir.path().to_path_buf();
    config.cache.hash_cache_path = cache_dir.path().join("hash_cache.json");
    config.cache.git_cache_path = cache_dir.path().join("git_cache.json");
    config.embedding.batch_size = 64;
    config.embedding.timeout_secs = 60;

    let server = RagMcpServer::with_config(config).await?;

    // Verify server was created with custom config
    assert!(std::mem::size_of_val(&server) > 0);

    Ok(())
}

#[tokio::test]
async fn test_full_indexing_workflow() -> Result<()> {
    let codebase_dir = TempDir::new()?;
    let db_dir = TempDir::new()?;
    let cache_dir = TempDir::new()?;

    // Create a simple test file
    let src_dir = codebase_dir.path().join("src");
    std::fs::create_dir_all(&src_dir)?;
    std::fs::write(src_dir.join("test.rs"), "fn main() { println!(\"test\"); }")?;

    // Create server
    let mut config = Config::default();
    config.vector_db.lancedb_path = db_dir.path().to_path_buf();
    config.cache.hash_cache_path = cache_dir.path().join("hash_cache.json");
    config.cache.git_cache_path = cache_dir.path().join("git_cache.json");

    let server = RagMcpServer::with_config(config).await?;

    // Test path normalization
    let normalized_path = RagMcpServer::normalize_path(&codebase_dir.path().to_string_lossy())?;
    assert!(!normalized_path.is_empty());

    // Test indexing (using the public do_index method)
    let index_response = server
        .do_index(
            normalized_path,
            Some("test_project".to_string()),
            vec![],
            vec![],
            1_048_576,
            None,
            None,
            None, // cancel_token
        )
        .await?;

    // Verify basic indexing worked
    assert!(index_response.files_indexed > 0);
    assert!(index_response.chunks_created > 0);

    Ok(())
}