gproxy-protocol 4.1.2

GPROXY v4 connection modelling, operation declarations, vendor wire types and pairwise conversion
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

use super::*;
use crate::wire::{DeclaredFields, gemini as g};

impl<R: ResourceAccess> GenerationResources<'_, R> {
    pub async fn gemini(
        &self,
        input: g::GenerateContentRequestBody,
    ) -> Result<g::GenerateContentRequestBody, TransformError> {
        let mut input = input.into_declared();
        let mut budget = self.budget();
        for content in input
            .contents
            .iter_mut()
            .chain(input.system_instruction.iter_mut())
        {
            for part in content.parts.iter_mut().flatten() {
                if let Some(file) = &part.file_data {
                    if part.inline_data.is_some() {
                        return Err(TransformError::shape(
                            "part",
                            "ambiguous file and inline data",
                        ));
                    }
                    let media = budget
                        .read(ResourceReference::Url(file.file_uri.clone()), false)
                        .await?;
                    if file
                        .mime_type
                        .as_ref()
                        .is_some_and(|mime| mime != &media.mime)
                    {
                        return Err(TransformError::invalid_result(
                            "file_data.mime",
                            "resource MIME differs from request",
                        ));
                    }
                    part.inline_data =
                        Some(g::Blob::builder(media.mime, STANDARD.encode(media.bytes)).build());
                    part.file_data = None;
                }
            }
        }
        Ok(input)
    }
}