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

use crate::claude::count_tokens::response::ClaudeCountTokensResponse;
use crate::claude::count_tokens::types::BetaMessageTokensCount;
use crate::claude::types::ClaudeResponseHeaders;
use crate::gemini::count_tokens::response::GeminiCountTokensResponse;
use crate::transform::claude::utils::beta_error_response_from_status_message;
use crate::transform::utils::TransformError;

impl TryFrom<GeminiCountTokensResponse> for ClaudeCountTokensResponse {
    type Error = TransformError;

    fn try_from(value: GeminiCountTokensResponse) -> Result<Self, TransformError> {
        Ok(match value {
            GeminiCountTokensResponse::Success {
                stats_code,
                headers,
                body,
            } => {
                let input_tokens = body.total_tokens;

                ClaudeCountTokensResponse::Success {
                    stats_code,
                    headers: ClaudeResponseHeaders {
                        extra: headers.extra,
                    },
                    body: BetaMessageTokensCount {
                        context_management: None,
                        input_tokens,
                    },
                }
            }
            GeminiCountTokensResponse::Error {
                stats_code,
                headers,
                body,
            } => ClaudeCountTokensResponse::Error {
                stats_code,
                headers: ClaudeResponseHeaders {
                    extra: headers.extra,
                },
                body: beta_error_response_from_status_message(stats_code, body.error.message),
            },
        })
    }
}