agent-search 0.4.6

Unified multi-provider search CLI for AI agents — 12 providers, 14 modes, one binary
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

use crate::types::Mode;
use regex::Regex;
use std::sync::OnceLock;

fn social_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)(\btweet\b|\btweets\b|\bon twitter\b|\bon x\b|x\.com|twitter\.com|\btrending on\b|what.*\bsaying\b|@\w{1,15}\b)").unwrap())
}

fn news_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(latest|breaking|news|today|yesterday|this week|headlines|update|announced)\b").unwrap())
}

fn academic_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(paper|research|study|journal|pubmed|arxiv|doi|peer.?review|citation|thesis|dissertation)\b").unwrap())
}

fn scholar_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(scholar|google scholar|academic search|scientific literature)\b").unwrap())
}

fn patents_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(patent|patents|patent number|USPTO|EPO|invention|prior art)\b").unwrap())
}

fn people_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(who is|linkedin|profile|founder|ceo|cto|engineer at|works at|person)\b").unwrap())
}

fn extract_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(extract|scrape|read page|get content|full text|article text)\b").unwrap())
}

fn similar_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(similar to|like this|related to|find similar|pages like)\b").unwrap())
}

fn images_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(image|images|photo|picture|illustration|diagram)\b").unwrap())
}

fn places_re() -> &'static Regex {
    static RE: OnceLock<Regex> = OnceLock::new();
    RE.get_or_init(|| Regex::new(r"(?i)\b(near me|restaurant|hotel|directions|address|location|map|places)\b").unwrap())
}

pub fn classify_intent(query: &str) -> Mode {
    let checks: &[(Mode, &dyn Fn() -> &'static Regex)] = &[
        (Mode::Social, &social_re),
        (Mode::News, &news_re),
        (Mode::Academic, &academic_re),
        (Mode::Scholar, &scholar_re),
        (Mode::Patents, &patents_re),
        (Mode::People, &people_re),
        (Mode::Extract, &extract_re),
        (Mode::Similar, &similar_re),
        (Mode::Images, &images_re),
        (Mode::Places, &places_re),
    ];

    for (mode, re_fn) in checks {
        if re_fn().is_match(query) {
            return *mode;
        }
    }
    Mode::General
}