swiftide-integrations 0.34.0

Composable LLM agents, typed task graphs, and streaming RAG pipelines in Rust
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

use derive_builder::Builder;
use reqwest::header::{AUTHORIZATION, HeaderMap};
use secrecy::{ExposeSecret as _, SecretString};
use serde::Deserialize;

const OPENROUTER_API_BASE: &str = "https://openrouter.ai/api/v1";

#[derive(Clone, Debug, Deserialize, Builder)]
#[serde(default)]
#[builder(setter(into, strip_option))]
pub struct OpenRouterConfig {
    #[builder(default = OPENROUTER_API_BASE.to_string())]
    api_base: String,
    api_key: SecretString,
    /// Sets the HTTP-Referer header (leaderbord)
    site_url: Option<String>,
    /// Sets the name (leaderbord)
    site_name: Option<String>,
}

impl OpenRouterConfig {
    pub fn builder() -> OpenRouterConfigBuilder {
        OpenRouterConfigBuilder::default()
    }
    pub fn with_api_base(&mut self, api_base: &str) -> &mut Self {
        self.api_base = api_base.to_string();

        self
    }

    pub fn with_api_key(&mut self, api_key: impl Into<SecretString>) -> &mut Self {
        self.api_key = api_key.into();

        self
    }
    pub fn with_site_url(&mut self, site_url: &str) -> &mut Self {
        self.site_url = Some(site_url.to_string());

        self
    }

    pub fn with_site_name(&mut self, site_name: &str) -> &mut Self {
        self.site_name = Some(site_name.to_string());

        self
    }
}

impl Default for OpenRouterConfig {
    fn default() -> Self {
        Self {
            api_base: OPENROUTER_API_BASE.to_string(),
            api_key: std::env::var("OPENROUTER_API_KEY")
                .unwrap_or_else(|_| String::new())
                .into(),
            site_url: None,
            site_name: None,
        }
    }
}

impl async_openai::config::Config for OpenRouterConfig {
    fn headers(&self) -> HeaderMap {
        let mut headers = HeaderMap::new();

        let api_key = self.api_key.expose_secret();
        assert!(!api_key.is_empty(), "API key for OpenRouter is required");

        headers.insert(
            AUTHORIZATION,
            format!("Bearer {}", self.api_key.expose_secret())
                .as_str()
                .parse()
                .unwrap(),
        );
        if let Ok(site_url) = self
            .site_url
            .as_deref()
            .unwrap_or("https://github.com/bosun-ai/swiftide")
            .parse()
        {
            headers.insert("HTTP-Referer", site_url);
        }

        if let Ok(site_name) = self.site_url.as_deref().unwrap_or("Swiftide").parse() {
            headers.insert("X-Title", site_name);
        }

        headers
    }

    fn url(&self, path: &str) -> String {
        format!("{}{}", self.api_base, path)
    }

    fn api_base(&self) -> &str {
        &self.api_base
    }

    fn api_key(&self) -> &SecretString {
        &self.api_key
    }

    fn query(&self) -> Vec<(&str, &str)> {
        vec![]
    }
}