ai_client 0.1.1

A Rust crate for interacting with AI language model APIs
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

use std::error::Error;
use std::fmt;

/// Custom error type for the AI client
#[derive(Debug)]
pub enum LlmClientError {
    EnvVarMissing(String),
    HttpError(reqwest::Error),
    ValidationError(String),
    ApiError(String),
    SerializationError(serde_json::Error),
    RateLimitExceeded(String),
    CacheError(String),
}

impl fmt::Display for LlmClientError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            LlmClientError::EnvVarMissing(var) => write!(f, "Environment variable {} not set", var),
            LlmClientError::HttpError(err) => write!(f, "HTTP error: {}", err),
            LlmClientError::ValidationError(msg) => write!(f, "Validation error: {}", msg),
            LlmClientError::ApiError(msg) => write!(f, "API error: {}", msg),
            LlmClientError::SerializationError(err) => write!(f, "Serialization error: {}", err),
            LlmClientError::RateLimitExceeded(msg) => write!(f, "Rate limit exceeded: {}", msg),
            LlmClientError::CacheError(msg) => write!(f, "Cache error: {}", msg),
        }
    }
}

impl Error for LlmClientError {}

impl From<reqwest::Error> for LlmClientError {
    fn from(err: reqwest::Error) -> Self {
        LlmClientError::HttpError(err)
    }
}

impl From<serde_json::Error> for LlmClientError {
    fn from(err: serde_json::Error) -> Self {
        LlmClientError::SerializationError(err)
    }
}