gproxy-transform 2.6.3

Pairwise request/response/stream transforms between the OpenAI, Anthropic Claude, and Google Gemini APIs
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

//! Gemini -> OpenAI edit-image transforms.

use crate::protocol::{gemini, openai};
use crate::transform::{TransformContext, TransformError};

use super::super::common;

pub fn request(
    input: gemini::GenerateContentRequest,
    _: &TransformContext,
) -> Result<openai::ImageEditRequest, TransformError> {
    let images = common::gemini_request_image_references(&input);
    if images.is_empty() {
        return Err(TransformError::InvalidInput {
            reason: "Gemini edit-image request did not contain image parts".to_owned(),
        });
    }

    Ok(crate::protocol::wire!(openai::ImageEditRequest {
        images,
        prompt: common::gemini_request_prompt(&input),
        background: None,
        input_fidelity: None,
        mask: None,
        model: input.model.map(Into::into),
        moderation: None,
        n: input
            .generation_config
            .as_ref()
            .and_then(|config| config.candidate_count)
            .and_then(common::positive_i32_to_u32),
        output_compression: None,
        output_format: common::gemini_output_format(input.generation_config.as_ref()),
        partial_images: None,
        quality: None,
        size: common::gemini_to_openai_edit_size(input.generation_config.as_ref()),
        stream: None,
        user: None,
        extra: Default::default(),
    }))
}

pub fn response(
    input: gemini::GenerateContentResponse,
    _: &TransformContext,
) -> Result<openai::ImagesResponse, TransformError> {
    Ok(common::gemini_response_to_openai_images(input))
}

pub fn stream_event(
    input: gemini::GenerateContentResponse,
    _: &TransformContext,
) -> Result<openai::ImageEditStreamEvent, TransformError> {
    common::gemini_to_openai_edit_stream(input).ok_or_else(|| TransformError::InvalidInput {
        reason: "Gemini image stream chunk did not contain an inline image".to_owned(),
    })
}