Skip to main content

chat_core/
error.rs

1use crate::types::metadata::Metadata;
2use thiserror::Error;
3
4#[derive(Clone, Debug, Error)]
5pub enum ChatError {
6    #[error("network error: {0}")]
7    Network(String),
8
9    #[error("provider error: {0}")]
10    Provider(String),
11
12    #[error("rate limited")]
13    RateLimited,
14
15    #[error("exceeded maximum steps")]
16    MaxStepsExceeded,
17
18    #[error("invalid response: {0}")]
19    InvalidResponse(String),
20
21    #[error("error in callback function: {0}")]
22    Callback(String),
23
24    #[error("unknown error: {0}")]
25    Other(String),
26}
27
28#[derive(Clone, Debug, Error)]
29#[error("Chat engine failed: {err}")]
30pub struct ChatFailure {
31    pub metadata: Option<Metadata>,
32    #[source]
33    pub err: ChatError,
34}
35
36impl ChatFailure {
37    pub fn from_err<E: Into<ChatError>>(err: E) -> Self {
38        Self {
39            metadata: None,
40            err: err.into(),
41        }
42    }
43}