ai-lib 0.4.0

A unified AI SDK for Rust providing a single interface for multiple AI providers with hybrid architecture
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

//! Error context and suggested actions

use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};

/// Context information for error tracking
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ErrorContext {
    /// AI provider that generated the error
    pub provider: String,
    /// API endpoint that was called
    pub endpoint: String,
    /// Request ID for tracking
    pub request_id: Option<String>,
    /// Timestamp when the error occurred
    pub timestamp: DateTime<Utc>,
    /// Number of retry attempts
    pub retry_count: u32,
    /// Suggested action to take
    pub suggested_action: SuggestedAction,
}

/// Suggested actions for error recovery
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum SuggestedAction {
    /// Retry the request with specified delay and max attempts
    Retry { delay_ms: u64, max_attempts: u32 },
    /// Switch to alternative providers
    SwitchProvider { alternative_providers: Vec<String> },
    /// Reduce request size or complexity
    ReduceRequestSize { max_tokens: Option<u32> },
    /// Check API credentials
    CheckCredentials,
    /// Contact support with specific reason
    ContactSupport { reason: String },
    /// No specific action recommended
    NoAction,
}

impl ErrorContext {
    /// Create a new error context
    pub fn new(provider: String, endpoint: String) -> Self {
        Self {
            provider,
            endpoint,
            request_id: None,
            timestamp: Utc::now(),
            retry_count: 0,
            suggested_action: SuggestedAction::NoAction,
        }
    }

    /// Create error context with retry information
    pub fn with_retry(mut self, retry_count: u32) -> Self {
        self.retry_count = retry_count;
        self
    }

    /// Create error context with request ID
    pub fn with_request_id(mut self, request_id: String) -> Self {
        self.request_id = Some(request_id);
        self
    }
}