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

use async_openai::error::OpenAIError;
#[cfg(feature = "mistralai")]
use mistralai_client::v1::error::{ApiError, ClientError};
#[cfg(feature = "ollama")]
use ollama_rs::error::OllamaError;
use reqwest::{Error as ReqwestError, StatusCode};
use thiserror::Error;

#[derive(Error, Debug)]
pub enum EmbedderError {
    #[error("Network request failed: {0}")]
    RequestError(#[from] ReqwestError),

    #[error("OpenAI error: {0}")]
    OpenAIError(#[from] OpenAIError),

    #[error("URL parsing error: {0}")]
    UrlParseError(#[from] url::ParseError),

    #[error("HTTP error: {status_code} {error_message}")]
    HttpError {
        status_code: StatusCode,
        error_message: String,
    },

    #[error("FastEmbed error: {0}")]
    FastEmbedError(String),

    #[cfg(feature = "ollama")]
    #[error("Ollama error: {0}")]
    OllamaError(#[from] OllamaError),

    #[cfg(feature = "mistralai")]
    #[error("MistralAI Client error: {0}")]
    MistralAIClientError(#[from] ClientError),

    #[cfg(feature = "mistralai")]
    #[error("MistralAI API error: {0}")]
    MistralAIApiError(#[from] ApiError),
}