openrouter2 0.3.0

Typed async and blocking reqwest clients for the OpenRouter API.
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

use reqwest::header::{HeaderName, HeaderValue};

use crate::OpenRouterError;

#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct RequestOptions {
    pub http_referer: Option<String>,
    pub x_title: Option<String>,
    pub session_id: Option<String>,
    pub extra_headers: Vec<(String, String)>,
}

impl RequestOptions {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn with_http_referer(mut self, referer: impl Into<String>) -> Self {
        self.http_referer = Some(referer.into());
        self
    }

    pub fn with_x_title(mut self, title: impl Into<String>) -> Self {
        self.x_title = Some(title.into());
        self
    }

    pub fn with_session_id(mut self, session_id: impl Into<String>) -> Self {
        self.session_id = Some(session_id.into());
        self
    }

    pub fn with_header(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
        self.extra_headers.push((name.into(), value.into()));
        self
    }

    #[cfg(feature = "async")]
    pub(crate) fn apply_async(
        &self,
        mut builder: reqwest::RequestBuilder,
    ) -> Result<reqwest::RequestBuilder, OpenRouterError> {
        if let Some(value) = &self.http_referer {
            builder = builder.header("HTTP-Referer", header_value(value)?);
        }
        if let Some(value) = &self.x_title {
            builder = builder.header("X-Title", header_value(value)?);
        }
        if let Some(value) = &self.session_id {
            builder = builder.header("X-Session-Id", header_value(value)?);
        }
        for (name, value) in &self.extra_headers {
            builder = builder.header(header_name(name)?, header_value(value)?);
        }
        Ok(builder)
    }

    #[cfg(feature = "blocking")]
    pub(crate) fn apply_blocking(
        &self,
        mut builder: reqwest::blocking::RequestBuilder,
    ) -> Result<reqwest::blocking::RequestBuilder, OpenRouterError> {
        if let Some(value) = &self.http_referer {
            builder = builder.header("HTTP-Referer", header_value(value)?);
        }
        if let Some(value) = &self.x_title {
            builder = builder.header("X-Title", header_value(value)?);
        }
        if let Some(value) = &self.session_id {
            builder = builder.header("X-Session-Id", header_value(value)?);
        }
        for (name, value) in &self.extra_headers {
            builder = builder.header(header_name(name)?, header_value(value)?);
        }
        Ok(builder)
    }
}

fn header_name(value: &str) -> Result<HeaderName, OpenRouterError> {
    HeaderName::from_bytes(value.as_bytes())
        .map_err(|e| OpenRouterError::InvalidHeader(e.to_string()))
}

fn header_value(value: &str) -> Result<HeaderValue, OpenRouterError> {
    HeaderValue::from_str(value).map_err(|e| OpenRouterError::InvalidHeader(e.to_string()))
}