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

use crate::openai::create_response::response::OpenAiCreateResponseResponse;
use crate::openai::create_response::stream::ResponseStreamEvent;
use crate::openai::create_response::websocket::response::OpenAiCreateResponseWebSocketMessageResponse;
use crate::openai::create_response::websocket::types::{
    OpenAiCreateResponseWebSocketDoneMarker, OpenAiCreateResponseWebSocketServerMessage,
};
use crate::transform::openai::websocket::context::OpenAiWebsocketTransformContext;
use crate::transform::utils::TransformError;

impl TryFrom<OpenAiCreateResponseResponse> for Vec<OpenAiCreateResponseWebSocketMessageResponse> {
    type Error = TransformError;

    fn try_from(value: OpenAiCreateResponseResponse) -> Result<Self, TransformError> {
        Ok(openai_nonstream_response_to_websocket_messages_with_context(value)?.0)
    }
}

pub fn openai_nonstream_response_to_websocket_messages_with_context(
    value: OpenAiCreateResponseResponse,
) -> Result<
    (
        Vec<OpenAiCreateResponseWebSocketMessageResponse>,
        OpenAiWebsocketTransformContext,
    ),
    TransformError,
> {
    let events = Vec::<ResponseStreamEvent>::try_from(value)?;
    openai_stream_events_to_websocket_messages_with_context(&events)
}

pub fn openai_stream_events_to_websocket_messages_with_context(
    value: &[ResponseStreamEvent],
) -> Result<
    (
        Vec<OpenAiCreateResponseWebSocketMessageResponse>,
        OpenAiWebsocketTransformContext,
    ),
    TransformError,
> {
    let ctx = OpenAiWebsocketTransformContext::default();
    let mut messages = Vec::with_capacity(value.len() + 1);
    for event in value {
        messages.push(OpenAiCreateResponseWebSocketServerMessage::StreamEvent(
            event.clone(),
        ));
    }
    messages.push(OpenAiCreateResponseWebSocketServerMessage::Done(
        OpenAiCreateResponseWebSocketDoneMarker::Done,
    ));

    Ok((messages, ctx))
}