ai-lib 0.4.0

A unified AI SDK for Rust providing a single interface for multiple AI providers with hybrid architecture
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

use std::error::Error as StdError;
use thiserror::Error;

/// Transport layer error types, unified encapsulation of HTTP and JSON errors
///
/// Transport layer error types with unified encapsulation of HTTP and JSON errors
///
/// Unified encapsulation of all HTTP-level errors and JSON parsing errors
#[derive(Error, Debug, Clone)]
pub enum TransportError {
    #[error("HTTP request failed: {0}")]
    HttpError(String),

    #[error("JSON serialization/deserialization failed: {0}")]
    JsonError(String),

    #[error("Invalid URL: {0}")]
    InvalidUrl(String),

    #[error("Authentication failed: {0}")]
    AuthenticationError(String),

    #[error("Rate limit exceeded")]
    RateLimitExceeded,

    #[error("Server error: {status} - {message}")]
    ServerError { status: u16, message: String },

    #[error("Client error: {status} - {message}")]
    ClientError { status: u16, message: String },

    #[error("Timeout error: {0}")]
    Timeout(String),
}

impl TransportError {
    /// Create error from HTTP status code
    pub fn from_status(status: u16, message: String) -> Self {
        match status {
            400..=499 => Self::ClientError { status, message },
            500..=599 => Self::ServerError { status, message },
            _ => Self::InvalidUrl(format!("Unexpected status code: {}", status)),
        }
    }
}

impl From<reqwest::Error> for TransportError {
    fn from(err: reqwest::Error) -> Self {
        let mut msg = err.to_string();
        if let Some(source) = StdError::source(&err) {
            msg.push_str(": ");
            msg.push_str(&source.to_string());
        }
        Self::HttpError(msg)
    }
}

impl From<serde_json::Error> for TransportError {
    fn from(err: serde_json::Error) -> Self {
        Self::JsonError(err.to_string())
    }
}