Skip to main content

mentedb_extraction/
error.rs

1use thiserror::Error;
2
3/// Errors that can occur during memory extraction.
4#[derive(Debug, Error)]
5pub enum ExtractionError {
6    /// The LLM provider returned an error or was unreachable.
7    #[error("provider error: {0}")]
8    ProviderError(String),
9
10    /// Authentication failed — wrong or missing API key.
11    #[error("authentication failed: {0}")]
12    AuthError(String),
13
14    /// The requested model was not found.
15    #[error("model not found: {0}")]
16    ModelNotFound(String),
17
18    /// The LLM response could not be parsed as valid extraction output.
19    #[error("parse error: {0}")]
20    ParseError(String),
21
22    /// The extraction configuration is invalid.
23    #[error("config error: {0}")]
24    ConfigError(String),
25
26    /// An HTTP request failed.
27    #[error("http error: {0}")]
28    HttpError(String),
29
30    /// An embedding operation failed.
31    #[error("embedding error: {0}")]
32    EmbeddingError(String),
33
34    /// Rate limit exceeded after all retry attempts.
35    #[error("rate limit exceeded after {attempts} attempts")]
36    RateLimitExceeded { attempts: usize },
37}
38
39impl From<reqwest::Error> for ExtractionError {
40    fn from(e: reqwest::Error) -> Self {
41        ExtractionError::HttpError(e.to_string())
42    }
43}
44
45impl From<serde_json::Error> for ExtractionError {
46    fn from(e: serde_json::Error) -> Self {
47        ExtractionError::ParseError(e.to_string())
48    }
49}