rust-genai 0.3.1

Rust SDK for the Google Gemini API and Vertex AI
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

use std::collections::HashMap;

use reqwest::header::HeaderMap;

use rust_genai_types::http::HttpResponse;

pub(crate) fn sdk_http_response_from_headers(headers: &HeaderMap) -> HttpResponse {
    let mut map: HashMap<String, String> = HashMap::new();
    for (name, value) in headers.iter() {
        let Ok(value_str) = value.to_str() else {
            continue;
        };
        let key = name.as_str().to_string();
        map.entry(key)
            .and_modify(|existing| {
                if !existing.is_empty() {
                    existing.push_str(", ");
                }
                existing.push_str(value_str);
            })
            .or_insert_with(|| value_str.to_string());
    }

    HttpResponse {
        headers: Some(map),
        body: None,
    }
}

pub(crate) fn sdk_http_response_from_headers_and_body(
    headers: &HeaderMap,
    body: String,
) -> HttpResponse {
    let mut response = sdk_http_response_from_headers(headers);
    response.body = Some(body);
    response
}