stakai 0.3.70

A provider-agnostic Rust SDK for AI completions with streaming support - Built by Stakpak
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163

//! HTTP headers support for custom provider headers

use serde::{Deserialize, Serialize};
use std::collections::HashMap;

/// HTTP headers map (single value per header name)
/// Headers can be layered; later values override earlier ones
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct Headers {
    inner: HashMap<String, String>,
}

impl Headers {
    /// Create a new empty headers map
    pub fn new() -> Self {
        Self::default()
    }

    /// Insert a header
    pub fn insert(&mut self, key: impl Into<String>, value: impl Into<String>) {
        self.inner.insert(key.into(), value.into());
    }

    /// Merge headers from overlay (consuming)
    pub fn merge(&mut self, overlay: Headers) {
        for (k, v) in overlay.inner {
            self.inner.insert(k, v);
        }
    }

    /// Merge headers from overlay (borrowing)
    pub fn merge_with(&mut self, overlay: &Headers) {
        for (k, v) in &overlay.inner {
            self.inner.insert(k.clone(), v.clone());
        }
    }

    /// Get a header value
    pub fn get(&self, key: &str) -> Option<&String> {
        self.inner.get(key)
    }

    /// Check if headers is empty
    pub fn is_empty(&self) -> bool {
        self.inner.is_empty()
    }

    /// Get number of headers
    pub fn len(&self) -> usize {
        self.inner.len()
    }

    /// Convert to reqwest HeaderMap
    pub fn to_reqwest_headers(&self) -> reqwest::header::HeaderMap {
        let mut headers = reqwest::header::HeaderMap::new();
        for (k, v) in &self.inner {
            if let (Ok(name), Ok(value)) = (
                reqwest::header::HeaderName::from_bytes(k.as_bytes()),
                reqwest::header::HeaderValue::from_str(v),
            ) {
                headers.insert(name, value);
            }
        }
        headers
    }

    /// Iterate over headers
    pub fn iter(&self) -> impl Iterator<Item = (&String, &String)> {
        self.inner.iter()
    }
}

// From implementations for ergonomics
impl From<(String, String)> for Headers {
    fn from((key, value): (String, String)) -> Self {
        let mut headers = Headers::new();
        headers.insert(key, value);
        headers
    }
}

impl From<(&str, &str)> for Headers {
    fn from((key, value): (&str, &str)) -> Self {
        let mut headers = Headers::new();
        headers.insert(key, value);
        headers
    }
}

impl From<Vec<(String, String)>> for Headers {
    fn from(vec: Vec<(String, String)>) -> Self {
        let mut headers = Headers::new();
        for (k, v) in vec {
            headers.insert(k, v);
        }
        headers
    }
}

impl<const N: usize> From<[(String, String); N]> for Headers {
    fn from(arr: [(String, String); N]) -> Self {
        let mut headers = Headers::new();
        for (k, v) in arr {
            headers.insert(k, v);
        }
        headers
    }
}

impl<const N: usize> From<[(&str, &str); N]> for Headers {
    fn from(arr: [(&str, &str); N]) -> Self {
        let mut headers = Headers::new();
        for (k, v) in arr {
            headers.insert(k, v);
        }
        headers
    }
}

impl IntoIterator for Headers {
    type Item = (String, String);
    type IntoIter = std::collections::hash_map::IntoIter<String, String>;

    fn into_iter(self) -> Self::IntoIter {
        self.inner.into_iter()
    }
}

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

    #[test]
    fn test_headers_basic() {
        let mut headers = Headers::new();
        headers.insert("x-api-key", "test-key");
        assert_eq!(headers.get("x-api-key"), Some(&"test-key".to_string()));
    }

    #[test]
    fn test_headers_merge() {
        let mut headers1 = Headers::new();
        headers1.insert("key1", "value1");

        let mut headers2 = Headers::new();
        headers2.insert("key2", "value2");

        headers1.merge(headers2);
        assert_eq!(headers1.len(), 2);
    }

    #[test]
    fn test_headers_from_tuple() {
        let headers: Headers = ("x-api-key", "test").into();
        assert_eq!(headers.get("x-api-key"), Some(&"test".to_string()));
    }

    #[test]
    fn test_headers_from_array() {
        let headers: Headers = [("key1", "val1"), ("key2", "val2")].into();
        assert_eq!(headers.len(), 2);
    }
}