rucora-providers 0.1.5

LLM provider implementations for rucora (OpenAI, Anthropic, Gemini, Ollama, etc.)
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

//! HTTP 客户端配置的共享常量和工具函数。
//!
//! 该模块提供：
//! - 默认的请求超时配置
//! - 连接超时配置
//! - HTTP 客户端构建工具

use std::time::Duration;

use reqwest::header::HeaderMap;

/// 默认请求超时时间（秒）
pub const DEFAULT_REQUEST_TIMEOUT_SECS: u64 = 120;

/// 默认连接超时时间（秒）
pub const DEFAULT_CONNECT_TIMEOUT_SECS: u64 = 15;

/// 构建带有超时配置的 HTTP 客户端
///
/// # 参数
///
/// - `headers`: 默认的 HTTP 请求头
///
/// # 返回
///
/// 配置好超时的 `reqwest::Client` 实例
pub fn build_client(headers: HeaderMap) -> reqwest::Client {
    reqwest::Client::builder()
        .default_headers(headers)
        .timeout(Duration::from_secs(DEFAULT_REQUEST_TIMEOUT_SECS))
        .connect_timeout(Duration::from_secs(DEFAULT_CONNECT_TIMEOUT_SECS))
        .build()
        .expect("reqwest client build failed")
}

/// 构建带有自定义超时的 HTTP 客户端
///
/// # 参数
///
/// - `headers`: 默认的 HTTP 请求头
/// - `request_timeout_secs`: 请求超时时间（秒）
/// - `connect_timeout_secs`: 连接超时时间（秒）
///
/// # 返回
///
/// 配置好超时的 `reqwest::Client` 实例
pub fn build_client_with_timeout(
    headers: HeaderMap,
    request_timeout_secs: u64,
    connect_timeout_secs: u64,
) -> reqwest::Client {
    reqwest::Client::builder()
        .default_headers(headers)
        .timeout(Duration::from_secs(request_timeout_secs))
        .connect_timeout(Duration::from_secs(connect_timeout_secs))
        .build()
        .expect("reqwest client build failed")
}