swink-agent-plugin-web 0.8.0

Web browsing plugin for swink-agent: fetch, search, screenshot, extract
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

use std::pin::Pin;

use serde::{Deserialize, Serialize};
use thiserror::Error;

#[cfg(feature = "brave")]
mod brave;
#[cfg(feature = "duckduckgo")]
mod duckduckgo;
#[cfg(feature = "tavily")]
mod tavily;

#[cfg(feature = "brave")]
pub use brave::BraveProvider;
#[cfg(feature = "duckduckgo")]
pub use duckduckgo::DuckDuckGoProvider;
#[cfg(feature = "tavily")]
pub use tavily::TavilyProvider;

/// A single search result.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SearchResult {
    pub title: String,
    pub url: String,
    pub snippet: String,
}

/// Errors from search providers.
#[derive(Debug, Error)]
pub enum SearchError {
    #[error("Network error: {0}")]
    NetworkError(String),
    #[error("Failed to parse response: {0}")]
    ParseError(String),
    #[error("Rate limited by search provider")]
    RateLimited,
    #[error("API key is required but not configured")]
    ApiKeyMissing,
    #[error("Search provider unavailable: {0}")]
    ProviderUnavailable(String),
}

/// Abstraction over different search backends.
pub trait SearchProvider: Send + Sync {
    fn name(&self) -> &str;
    fn search(
        &self,
        query: &str,
        max_results: usize,
    ) -> Pin<
        Box<dyn std::future::Future<Output = Result<Vec<SearchResult>, SearchError>> + Send + '_>,
    >;
}

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

    #[test]
    fn search_result_is_publicly_constructible() {
        let result = SearchResult {
            title: "Rust".to_string(),
            url: "https://www.rust-lang.org".to_string(),
            snippet: "Systems programming language".to_string(),
        };
        assert_eq!(result.title, "Rust");
    }
}