openrouter-rs 0.5.2

A type-safe OpenRouter Rust SDK
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

use serde::Deserialize;
use serde_json::Value;
use surf::StatusCode;

use crate::error::{ApiErrorContext, ApiErrorKind, OpenRouterError};

#[derive(Deserialize, Debug)]
struct ApiErrorResponse {
    error: ApiError,
}

#[derive(Deserialize, Debug)]
struct ApiError {
    code: Option<i64>,
    message: String,
    metadata: Option<ApiErrorMetadata>,
}

#[derive(Deserialize, Debug)]
#[serde(untagged)]
enum ApiErrorMetadata {
    ModerationError(ModerationErrorMetadata),
    ProviderError(ProviderErrorMetadata),
    Raw(Value),
}

#[derive(Deserialize, Debug)]
struct ModerationErrorMetadata {
    reasons: Vec<String>,
    flagged_input: String,
    provider_name: String,
    model_slug: String,
}

#[derive(Deserialize, Debug)]
struct ProviderErrorMetadata {
    provider_name: String,
    raw: Value,
}

fn build_api_error(
    status: StatusCode,
    request_id: Option<String>,
    api_error: ApiError,
) -> OpenRouterError {
    let ApiError {
        code,
        message,
        metadata,
    } = api_error;

    let (kind, normalized_metadata) = match metadata {
        Some(ApiErrorMetadata::ModerationError(moderation)) => (
            ApiErrorKind::Moderation {
                reasons: moderation.reasons.clone(),
                flagged_input: moderation.flagged_input.clone(),
                provider_name: moderation.provider_name.clone(),
                model_slug: moderation.model_slug.clone(),
            },
            Some(serde_json::json!({
                "reasons": moderation.reasons,
                "flagged_input": moderation.flagged_input,
                "provider_name": moderation.provider_name,
                "model_slug": moderation.model_slug,
            })),
        ),
        Some(ApiErrorMetadata::ProviderError(provider)) => (
            ApiErrorKind::Provider {
                provider_name: provider.provider_name,
                raw: provider.raw.clone(),
            },
            Some(provider.raw),
        ),
        Some(ApiErrorMetadata::Raw(raw)) => (ApiErrorKind::Generic, Some(raw)),
        None => (ApiErrorKind::Generic, None),
    };

    OpenRouterError::Api(Box::new(ApiErrorContext {
        status,
        api_code: code,
        message,
        request_id,
        metadata: normalized_metadata,
        kind,
    }))
}

pub fn parse_api_error(
    status: StatusCode,
    request_id: Option<String>,
    text: &str,
) -> OpenRouterError {
    match serde_json::from_str::<ApiErrorResponse>(text) {
        Ok(payload) => build_api_error(status, request_id, payload.error),
        Err(_) => OpenRouterError::Api(Box::new(ApiErrorContext {
            status,
            api_code: Some(i64::from(u16::from(status))),
            message: text.to_string(),
            request_id,
            metadata: None,
            kind: ApiErrorKind::Generic,
        })),
    }
}