artificial-openai 0.5.0

use std::str::Utf8Error;

use artificial_core::error::ArtificialError;
use reqwest::StatusCode;
use std::time::Duration;

/// Headers conveying rate limit information returned by OpenAI.
#[derive(Debug, Clone)]
pub struct OpenAiRateLimitHeaders {
    pub limit_requests: Option<u32>,
    pub remaining_requests: Option<u32>,
    pub reset_requests: Option<String>,
    pub limit_tokens: Option<u32>,
    pub remaining_tokens: Option<u32>,
    pub reset_tokens: Option<String>,
}

/// High-level error type covering every failure mode the client can hit.
#[derive(Debug, thiserror::Error)]
pub enum OpenAiError {
    #[error("request failed: {0}")]
    Http(#[from] reqwest::Error),

    #[error("couldn’t serialise body: {0}")]
    Serde(#[from] serde_json::Error),

    #[error("rate limited (status {status}), retry_after={retry_after:?}, reset_at={reset_at:?}")]
    RateLimited {
        status: StatusCode,
        body: String,
        retry_after: Option<Duration>,
        reset_at: Option<String>,
        headers: OpenAiRateLimitHeaders,
    },

    #[error("OpenAI returned non-success status {status}: {body}")]
    Api { status: StatusCode, body: String },

    #[error("OpenAI format error: {0}")]
    Format(String),

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

impl From<OpenAiError> for ArtificialError {
    fn from(value: OpenAiError) -> Self {
        ArtificialError::Backend(Box::new(value))
    }
}

impl From<Utf8Error> for OpenAiError {
    fn from(value: Utf8Error) -> Self {
        Self::Unknown(format!("UTF8 error: {value}"))
    }
}