leindex 1.9.0

LeIndex MCP and semantic code search engine for AI tools and large codebases
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

//! Deterministic routing for exact and semantic queries.

/// The caller's explicit search intent.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RequestedMode {
    /// Infer intent from the query text.
    Auto,
    /// Search literal text without semantic expansion.
    Exact,
    /// Search semantic similarity.
    Semantic,
    /// Run PDG-aware deep analysis.
    Deep,
}

/// The inexpensive path selected before search state is touched.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum QueryRoute {
    /// A single identifier or qualified symbol path.
    ExactSymbol,
    /// Literal text matching.
    ExactText,
    /// Mandatory TF-IDF plus configured neural search after the provider is ready.
    Semantic,
    /// PDG-aware analysis.
    DeepPdg,
}

/// Classify a query without loading search, PDG, or embedding state.
pub fn classify(query: &str, requested: RequestedMode) -> QueryRoute {
    match requested {
        RequestedMode::Exact => return QueryRoute::ExactText,
        RequestedMode::Semantic => return QueryRoute::Semantic,
        RequestedMode::Deep => return QueryRoute::DeepPdg,
        RequestedMode::Auto => {}
    }

    let query = query.trim_matches(|c| matches!(c, '`' | '"' | '\''));
    let identifier = !query.is_empty()
        && !query.chars().any(char::is_whitespace)
        && query
            .chars()
            .all(|c| c.is_ascii_alphanumeric() || matches!(c, '_' | ':' | '.'));

    if identifier {
        QueryRoute::ExactSymbol
    } else {
        QueryRoute::Semantic
    }
}

#[cfg(test)]
mod tests {
    use super::{QueryRoute, RequestedMode, classify};

    #[test]
    fn routes_identifier_and_natural_language_queries() {
        assert_eq!(
            classify("`Askpass::new`", RequestedMode::Auto),
            QueryRoute::ExactSymbol
        );
        assert_eq!(
            classify("registry_record", RequestedMode::Exact),
            QueryRoute::ExactText
        );
        assert_eq!(
            classify("how are sudo credentials propagated", RequestedMode::Auto),
            QueryRoute::Semantic
        );
        assert_eq!(
            classify("sudo credential flow", RequestedMode::Deep),
            QueryRoute::DeepPdg
        );
    }
}