agent-diva-providers 0.5.0

LLM provider integrations for agent-diva
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

//! Shared `reqwest::Client` options for provider HTTP calls.

use reqwest::Client;
use std::time::Duration;

/// Local gateways and plain `http://` bases are often happier with HTTP/1.1 only.
/// Remote `https://` APIs (e.g. DeepSeek) should use normal ALPN so the peer does not RST during TLS.
pub(crate) fn should_force_http1_only_for_api_base(api_base: &str) -> bool {
    let Ok(url) = reqwest::Url::parse(api_base.trim()) else {
        return true;
    };
    match url.scheme() {
        "http" => true,
        "https" => url
            .host_str()
            .map(|host| {
                let h = host.to_ascii_lowercase();
                h == "localhost" || h == "127.0.0.1" || h == "::1" || h.ends_with(".local")
            })
            .unwrap_or(true),
        _ => true,
    }
}

pub(crate) fn build_api_http_client(
    api_base: &str,
    request_timeout: Duration,
) -> reqwest::Result<Client> {
    let mut builder = Client::builder()
        .connect_timeout(Duration::from_secs(45))
        .timeout(request_timeout);
    if should_force_http1_only_for_api_base(api_base) {
        builder = builder.http1_only();
    }
    builder.build()
}

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

    #[test]
    fn deepseek_https_public_no_force_http1() {
        assert!(!should_force_http1_only_for_api_base(
            "https://api.deepseek.com/v1"
        ));
    }

    #[test]
    fn local_litellm_https_forces_http1() {
        assert!(should_force_http1_only_for_api_base(
            "https://127.0.0.1:4000/v1"
        ));
    }

    #[test]
    fn plain_http_forces_http1() {
        assert!(should_force_http1_only_for_api_base(
            "http://localhost:4000"
        ));
    }
}