terraphim_middleware 1.16.33

Terraphim middleware for searching haystacks
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

use terraphim_config::{ConfigBuilder, Haystack, Role, ServiceType};
use terraphim_middleware::indexer::search_haystacks;
use terraphim_types::{RelevanceFunction, SearchQuery};

/// Live MCP haystack test using SSE server-everything
///
/// Requires MCP_SERVER_URL (e.g., http://127.0.0.1:8001)
#[tokio::test]
#[ignore]
async fn mcp_live_haystack_smoke() {
    dotenvy::dotenv().ok();
    let base_url = match std::env::var("MCP_SERVER_URL") {
        Ok(v) if !v.trim().is_empty() => v,
        _ => {
            eprintln!("MCP_SERVER_URL not set; skipping live MCP test");
            return;
        }
    };

    let role = Role {
        shortname: Some("MCP".to_string()),
        name: "MCP".into(),
        relevance_function: RelevanceFunction::TitleScorer,
        terraphim_it: false,
        theme: "lumen".to_string(),
        kg: None,
        haystacks: vec![Haystack::new(base_url.clone(), ServiceType::Mcp, true)
            .with_extra_parameter("base_url".into(), base_url.clone())
            .with_extra_parameter("transport".into(), "sse".into())],
        llm_enabled: false,
        llm_api_key: None,
        llm_model: None,
        llm_auto_summarize: false,
        llm_chat_enabled: false,
        llm_chat_system_prompt: None,
        llm_chat_model: None,
        llm_context_window: None,
        llm_router_enabled: false,
        llm_router_config: None,
        extra: ahash::AHashMap::new(),
    };

    let mut config = ConfigBuilder::new()
        .add_role("MCP", role)
        .default_role("MCP")
        .unwrap()
        .build()
        .unwrap();

    let config_state = terraphim_config::ConfigState::new(&mut config)
        .await
        .expect("config state");

    // Perform a search (current indexer returns empty Index until full integration)
    let query = SearchQuery {
        search_term: "work".into(),
        skip: Some(0),
        limit: Some(10),
        role: Some("MCP".into()),
        operator: None,
        search_terms: None,
        layer: Default::default(),
        include_pinned: false,
    };

    let result = search_haystacks(config_state, query)
        .await
        .expect("search ok");
    // If server-everything is running and exposes search/list, we expect results
    // Otherwise, the index may be empty and this test remains a smoke check
    if !result.is_empty() {
        assert!(
            result.values().next().is_some(),
            "expect at least one document"
        );
    }
}