groqai 0.1.3

A modern, type-safe Rust SDK for the Groq AI API with enterprise-grade features
Documentation 
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

// src/error.rs

use thiserror::Error;
use serde::{Deserialize, Serialize};

/// Detailed error information from Groq API
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GroqApiErrorDetails {
    pub message: String,
    #[serde(rename = "type")]
    pub error_type: Option<String>,
    pub code: Option<String>,
    pub param: Option<String>,
}

/// Enhanced API error with structured details
#[derive(Debug, Clone)]
pub struct GroqApiError {
    pub status: reqwest::StatusCode,
    pub details: Option<GroqApiErrorDetails>,
    pub raw_response: String,
}

impl std::fmt::Display for GroqApiError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "API request failed with status code {}: {}", self.status, self.raw_response)?;
        if let Some(details) = &self.details {
            write!(f, " (Error: {}, Type: {})", 
                details.message, 
                details.error_type.as_deref().unwrap_or("unknown"))?;
        }
        Ok(())
    }
}

impl std::error::Error for GroqApiError {}

#[derive(Error, Debug)]
pub enum GroqError {
    #[error("{0}")]
    Api(#[from] GroqApiError),

    #[error("JSON serialization error: {0}")]
    Json(#[from] serde_json::Error),
    
    #[error("Reqwest error: {0}")]
    Reqwest(#[from] reqwest::Error),

    #[error("Invalid message content: {0}")]
    InvalidMessage(String),

    #[error("IO error: {0}")]
    Io(#[from] std::io::Error),

    #[error("Groq API error: {0}")]
    GroqApiError(String),

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

    #[error("Invalid API key: {0}")]
    InvalidApiKey(String),

    #[error("Stream parsing error: {0}")]
    StreamParsing(String),
}

impl GroqError {
    /// Create a new API error with structured details
    pub fn api_error(status: reqwest::StatusCode, response_text: String) -> Self {
        let details = serde_json::from_str::<serde_json::Value>(&response_text)
            .ok()
            .and_then(|json| {
                json.get("error")
                    .and_then(|error| serde_json::from_value::<GroqApiErrorDetails>(error.clone()).ok())
            });

        let api_error = GroqApiError {
            status,
            details,
            raw_response: response_text,
        };

        GroqError::Api(api_error)
    }

    /// Check if this is a rate limit error
    pub fn is_rate_limit(&self) -> bool {
        match self {
            GroqError::Api(api_error) => {
                api_error.status == reqwest::StatusCode::TOO_MANY_REQUESTS
                    || api_error.details.as_ref()
                        .map(|d| d.error_type.as_deref() == Some("rate_limit_exceeded"))
                        .unwrap_or(false)
            }
            _ => false,
        }
    }

    /// Check if this is an authentication error
    pub fn is_authentication_error(&self) -> bool {
        match self {
            GroqError::Api(api_error) => {
                api_error.status == reqwest::StatusCode::UNAUTHORIZED
                    || api_error.status == reqwest::StatusCode::FORBIDDEN
            }
            _ => false,
        }
    }
}