Skip to main content

polars_ai/
error.rs

1use std::error::Error;
2use std::fmt;
3
4/// Custom error type for OpenAI response errors.
5#[derive(Debug)]
6pub struct OpenAIResponseError {
7    /// Error message.
8    message: String,
9}
10
11impl OpenAIResponseError {
12    /// Creates a new instance of `OpenAIResponseError`.
13    ///
14    /// # Arguments
15    ///
16    /// * `message` - The error message.
17    ///
18    /// # Returns
19    ///
20    /// A new instance of `OpenAIResponseError`.
21    pub fn new(message: &str) -> Self {
22        OpenAIResponseError {
23            message: message.to_string(),
24        }
25    }
26}
27
28impl Error for OpenAIResponseError {}
29
30impl fmt::Display for OpenAIResponseError {
31    /// Formats the error for display.
32    ///
33    /// # Arguments
34    ///
35    /// * `f` - The formatter.
36    ///
37    /// # Returns
38    ///
39    /// A `fmt::Result`.
40    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
41        write!(f, "OpenAI Response Error: {}", self.message)
42    }
43}