terraphim_middleware 1.16.34

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
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use terraphim_config::{ConfigBuilder, Haystack, Role, ServiceType};
use terraphim_middleware::{
    haystack::ClickUpHaystackIndexer,
    indexer::{search_haystacks, IndexMiddleware},
};
use terraphim_test_utils::EnvVarGuard;
use terraphim_types::{RelevanceFunction, SearchQuery};

#[tokio::test]
async fn clickup_mapping_handles_missing_token() {
    // Ensure no token is present
    let _guard = EnvVarGuard::remove("CLICKUP_API_TOKEN");

    let haystack = Haystack::new("clickup".to_string(), ServiceType::ClickUp, true);
    let indexer = ClickUpHaystackIndexer::default();
    let index = indexer
        .index("test", &haystack)
        .await
        .expect("indexing should not error");
    assert!(index.is_empty());
}

#[tokio::test]
#[ignore]
async fn clickup_live_search_returns_documents() {
    // Requires CLICKUP_API_TOKEN and CLICKUP_TEAM_ID set
    dotenvy::dotenv().ok();
    if std::env::var("CLICKUP_API_TOKEN").is_err() || std::env::var("CLICKUP_TEAM_ID").is_err() {
        eprintln!("CLICKUP_API_TOKEN or CLICKUP_TEAM_ID not set; skipping");
        return;
    }

    let role = Role {
        shortname: Some("ClickUp".to_string()),
        name: "ClickUp".into(),
        relevance_function: RelevanceFunction::TitleScorer,
        terraphim_it: false,
        theme: "lumen".to_string(),
        kg: None,
        haystacks: vec![
            Haystack::new("clickup".to_string(), ServiceType::ClickUp, true)
                .with_extra_parameter("team_id".into(), std::env::var("CLICKUP_TEAM_ID").unwrap())
                .with_extra_parameter("include_closed".into(), "true".into())
                .with_extra_parameter("subtasks".into(), "true".into()),
        ],
        extra: ahash::AHashMap::new(),
        llm_router_enabled: false,
        llm_router_config: None,
        ..Default::default()
    };

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

    let config_state = terraphim_config::ConfigState::new(&mut config)
        .await
        .unwrap();
    let query = SearchQuery {
        search_term: "meeting".into(),
        skip: Some(0),
        limit: Some(10),
        role: Some("ClickUp".into()),
        operator: None,
        search_terms: None,
        layer: Default::default(),
        include_pinned: false,
    };
    let _results = search_haystacks(config_state, query).await.unwrap();
    // Results should be empty or non-empty (both are valid for this test)
}

#[tokio::test]
#[ignore]
async fn clickup_live_search_work_term() {
    // Requires CLICKUP_API_TOKEN and CLICKUP_TEAM_ID set
    dotenvy::dotenv().ok();
    if std::env::var("CLICKUP_API_TOKEN").is_err() || std::env::var("CLICKUP_TEAM_ID").is_err() {
        eprintln!("CLICKUP_API_TOKEN or CLICKUP_TEAM_ID not set; skipping");
        return;
    }

    let role = Role {
        shortname: Some("ClickUp".to_string()),
        name: "ClickUp".into(),
        relevance_function: RelevanceFunction::TitleScorer,
        terraphim_it: false,
        theme: "lumen".to_string(),
        kg: None,
        haystacks: vec![
            Haystack::new("clickup".to_string(), ServiceType::ClickUp, true)
                .with_extra_parameter("team_id".into(), std::env::var("CLICKUP_TEAM_ID").unwrap())
                .with_extra_parameter("include_closed".into(), "true".into())
                .with_extra_parameter("subtasks".into(), "true".into()),
        ],
        extra: ahash::AHashMap::new(),
        llm_router_enabled: false,
        llm_router_config: None,
        ..Default::default()
    };

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

    let config_state = terraphim_config::ConfigState::new(&mut config)
        .await
        .unwrap();
    let query = SearchQuery {
        search_term: "work".into(),
        skip: Some(0),
        limit: Some(10),
        role: Some("ClickUp".into()),
        operator: None,
        search_terms: None,
        layer: Default::default(),
        include_pinned: false,
    };
    let results = search_haystacks(config_state, query).await.unwrap();
    assert!(
        !results.is_empty(),
        "expected some ClickUp results for term 'work'"
    );
}