ai 0.4.0

Simple to use LLM library for Rust with streaming, tool calling, OAuth helpers, and a lightweight agent loop
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

use reqwest::StatusCode;

#[derive(thiserror::Error, Debug)]
pub enum Error {
    #[error(transparent)]
    Http(#[from] reqwest::Error),

    #[error(transparent)]
    Json(#[from] serde_json::Error),

    #[error(transparent)]
    Io(#[from] std::io::Error),

    #[error("invalid header value for {0}: {1}")]
    InvalidHeaderValue(String, reqwest::header::InvalidHeaderValue),

    #[error("No API key for provider: {0}")]
    MissingApiKey(String),

    #[error("unsupported api: {0}")]
    UnsupportedApi(String),

    #[error("provider {provider} does not support {capability}")]
    UnsupportedCapability {
        provider: String,
        capability: &'static str,
    },

    #[error("provider returned HTTP {status}: {body}")]
    ApiStatus { status: StatusCode, body: String },

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

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

    #[error("request was cancelled")]
    Cancelled,

    #[error("stream ended before producing a final assistant message")]
    StreamClosed,
}

pub type Result<T> = std::result::Result<T, Error>;

impl Error {
    pub fn unsupported_capability(provider: impl Into<String>, capability: &'static str) -> Self {
        Self::UnsupportedCapability {
            provider: provider.into(),
            capability,
        }
    }
}