va-ai-api-bridge 0.1.7

Protocol translation primitives for AI API request and response shapes.
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use serde_json::{Map, Value};

use crate::Extensions;

pub(super) struct AnthropicImageSource {
    pub(super) media_type: Option<String>,
    pub(super) url: Option<String>,
    pub(super) data: Option<String>,
}

pub(super) struct AnthropicFileSource {
    pub(super) media_type: Option<String>,
    pub(super) url: Option<String>,
    pub(super) data: Option<String>,
    pub(super) file_id: Option<String>,
}

pub(super) fn anthropic_source_to_image(source: Option<&Value>) -> AnthropicImageSource {
    AnthropicImageSource {
        media_type: source
            .and_then(|source| source.get("media_type"))
            .and_then(Value::as_str)
            .map(ToString::to_string),
        url: source
            .and_then(|source| source.get("url").or_else(|| source.get("image_url")))
            .and_then(Value::as_str)
            .map(ToString::to_string),
        data: source
            .and_then(|source| source.get("data"))
            .and_then(Value::as_str)
            .map(ToString::to_string),
    }
}

pub(super) fn anthropic_source_to_file(source: Option<&Value>) -> AnthropicFileSource {
    AnthropicFileSource {
        media_type: source
            .and_then(|source| source.get("media_type"))
            .and_then(Value::as_str)
            .map(ToString::to_string),
        url: source
            .and_then(|source| source.get("url").or_else(|| source.get("file_url")))
            .and_then(Value::as_str)
            .map(ToString::to_string),
        data: source
            .and_then(|source| source.get("data"))
            .and_then(Value::as_str)
            .map(ToString::to_string),
        file_id: source
            .and_then(|source| source.get("file_id"))
            .and_then(Value::as_str)
            .map(ToString::to_string),
    }
}

pub(super) fn anthropic_image_source(
    media_type: Option<&str>,
    url: Option<&str>,
    data: Option<&str>,
    extensions: &Extensions,
) -> Value {
    let mut source = Map::new();
    if let Some(url) = url {
        source.insert("type".to_string(), Value::String("url".to_string()));
        source.insert("url".to_string(), Value::String(url.to_string()));
        return Value::Object(source);
    }

    source.insert("type".to_string(), Value::String("base64".to_string()));
    if let Some(media_type) = media_type {
        source.insert(
            "media_type".to_string(),
            Value::String(media_type.to_string()),
        );
    }
    if let Some(data) = data {
        source.insert("data".to_string(), Value::String(data.to_string()));
    }
    for (key, value) in extensions {
        source.insert(key.clone(), value.clone());
    }
    Value::Object(source)
}

pub(super) fn anthropic_file_source(
    media_type: Option<&str>,
    url: Option<&str>,
    data: Option<&str>,
    extensions: &Extensions,
) -> Value {
    let mut source = Map::new();
    if let Some(Value::String(file_id)) = extensions.get("file_id") {
        source.insert("type".to_string(), Value::String("file".to_string()));
        source.insert("file_id".to_string(), Value::String(file_id.clone()));
        return Value::Object(source);
    }
    if let Some(url) = url {
        source.insert("type".to_string(), Value::String("url".to_string()));
        source.insert("url".to_string(), Value::String(url.to_string()));
        return Value::Object(source);
    }

    let (data_url_media_type, data) = data.map(split_data_url).unwrap_or((None, ""));
    source.insert("type".to_string(), Value::String("base64".to_string()));
    if let Some(media_type) = media_type.or(data_url_media_type) {
        source.insert(
            "media_type".to_string(),
            Value::String(media_type.to_string()),
        );
    }
    source.insert("data".to_string(), Value::String(data.to_string()));
    Value::Object(source)
}

fn split_data_url(data: &str) -> (Option<&str>, &str) {
    let Some(rest) = data.strip_prefix("data:") else {
        return (None, data);
    };
    let Some((metadata, payload)) = rest.split_once(',') else {
        return (None, data);
    };
    let media_type = metadata
        .split(';')
        .next()
        .filter(|media_type| !media_type.is_empty());
    (media_type, payload)
}