toolhub-recommender 0.1.0

Hybrid FTS + sqlite-vec cosine recommender with success-rate reranker for toolhub
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

//! Shared recommend-pipeline tuning constants + helpers.
//!
//! Used by both the CLI `recommend` command and the MCP server so changes
//! land in one place.

pub const VEC_CANDIDATES: usize = 50;
pub const FTS_CANDIDATES: usize = 50;
pub const COS_WEIGHT: f32 = 0.6;
pub const FTS_WEIGHT: f32 = 0.4;

/// Tokenise on whitespace, double-quote each token (escaping internal quotes),
/// and OR-join. OR keeps recall when only some words match.
pub fn build_fts_query(task: &str) -> String {
    let toks: Vec<String> = task
        .split_whitespace()
        .filter(|t| !t.is_empty())
        .map(|t| {
            let cleaned = t.replace('"', "");
            format!("\"{cleaned}\"")
        })
        .collect();
    toks.join(" OR ")
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn empty_input_yields_empty_query() {
        assert_eq!(build_fts_query(""), "");
        assert_eq!(build_fts_query("   "), "");
    }

    #[test]
    fn quotes_each_token_and_or_joins() {
        assert_eq!(build_fts_query("design tokens"), "\"design\" OR \"tokens\"");
    }

    #[test]
    fn strips_embedded_quotes() {
        assert_eq!(build_fts_query("ab\"cd"), "\"abcd\"");
    }
}