gproxy-protocol 1.0.20

Wire-format types and cross-protocol transforms for Claude, OpenAI, and Gemini LLM 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

use crate::gemini::generate_content::request::GeminiGenerateContentRequest;
use crate::gemini::generate_content::response::GeminiGenerateContentResponse;
use crate::gemini::stream_generate_content::request::GeminiStreamGenerateContentRequest;
use crate::gemini::stream_generate_content::response::GeminiStreamGenerateContentResponse;
use crate::transform::utils::TransformError;

impl TryFrom<&GeminiGenerateContentRequest> for GeminiStreamGenerateContentRequest {
    type Error = TransformError;

    fn try_from(value: &GeminiGenerateContentRequest) -> Result<Self, TransformError> {
        Ok(GeminiStreamGenerateContentRequest {
            method: value.method,
            path: crate::gemini::stream_generate_content::request::PathParameters {
                model: value.path.model.clone(),
            },
            query: crate::gemini::stream_generate_content::request::QueryParameters::default(),
            headers: crate::gemini::stream_generate_content::request::RequestHeaders::default(),
            body: value.body.clone(),
        })
    }
}

impl TryFrom<GeminiGenerateContentRequest> for GeminiStreamGenerateContentRequest {
    type Error = TransformError;

    fn try_from(value: GeminiGenerateContentRequest) -> Result<Self, TransformError> {
        GeminiStreamGenerateContentRequest::try_from(&value)
    }
}

impl TryFrom<GeminiGenerateContentResponse> for GeminiStreamGenerateContentResponse {
    type Error = TransformError;

    fn try_from(value: GeminiGenerateContentResponse) -> Result<Self, TransformError> {
        Ok(match value {
            GeminiGenerateContentResponse::Success {
                stats_code,
                headers,
                ..
            } => GeminiStreamGenerateContentResponse::Success {
                stats_code,
                headers,
            },
            GeminiGenerateContentResponse::Error {
                stats_code,
                headers,
                body,
            } => GeminiStreamGenerateContentResponse::Error {
                stats_code,
                headers,
                body,
            },
        })
    }
}