gproxy-transform 3.0.0-alpha.0

GPROXY v3 pairwise protocol transforms
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

//! Typed OpenAI and Gemini embedding pairs.

use gproxy_protocol::gemini;
use gproxy_protocol::openai::embeddings as openai;

use super::RequestContext;
use crate::TransformError;

pub mod openai_to_gemini {
    use super::*;

    pub fn request(
        input: openai::CreateEmbeddingRequest,
        context: RequestContext<'_>,
    ) -> Result<gemini::EmbedContentRequest, TransformError> {
        crate::embeddings::openai_to_gemini_single_typed(input, context.upstream_model)
    }

    pub fn batch_request(
        input: openai::CreateEmbeddingRequest,
        context: RequestContext<'_>,
    ) -> Result<gemini::BatchEmbedContentsRequest, TransformError> {
        crate::embeddings::openai_to_gemini_batch_typed(input, context.upstream_model)
    }

    pub fn response(input: gemini::EmbedContentResponse) -> openai::CreateEmbeddingResponse {
        crate::embeddings::gemini_single_response_to_openai_typed(input)
    }

    pub fn batch_response(
        input: gemini::BatchEmbedContentsResponse,
    ) -> openai::CreateEmbeddingResponse {
        crate::embeddings::gemini_batch_response_to_openai_typed(input)
    }
}

pub mod gemini_to_openai {
    use super::*;

    pub fn request(
        input: gemini::EmbedContentRequest,
        context: RequestContext<'_>,
    ) -> Result<openai::CreateEmbeddingRequest, TransformError> {
        crate::embeddings::gemini_single_to_openai_typed(input, context.upstream_model)
    }

    pub fn batch_request(
        input: gemini::BatchEmbedContentsRequest,
        context: RequestContext<'_>,
    ) -> Result<openai::CreateEmbeddingRequest, TransformError> {
        crate::embeddings::gemini_batch_to_openai_typed(input, context.upstream_model)
    }

    pub fn response(
        input: openai::CreateEmbeddingResponse,
    ) -> Result<gemini::EmbedContentResponse, TransformError> {
        crate::embeddings::openai_response_to_gemini_single_typed(input)
    }

    pub fn batch_response(
        input: openai::CreateEmbeddingResponse,
    ) -> Result<gemini::BatchEmbedContentsResponse, TransformError> {
        crate::embeddings::openai_response_to_gemini_batch_typed(input)
    }
}