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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use gproxy_protocol::openai;

use crate::TransformError;

use super::super::claude_to_openai::{OutputEvent, State};

pub(in crate::common::stream) enum ResponseDelta {
    OutputText,
    ReasoningText,
    FunctionArguments,
}

impl State {
    pub(in crate::common::stream) fn response_delta(
        &mut self,
        kind: ResponseDelta,
        item_id: String,
        index: u64,
        delta: String,
    ) -> Result<OutputEvent, TransformError> {
        let sequence_number = Some(self.next_sequence());
        let output_index = index as u32;
        let event = match kind {
            ResponseDelta::OutputText => openai::KnownResponseStreamEvent::ResponseOutputTextDelta(
                crate::wire!(openai::ResponseOutputTextDeltaEvent {
                    content_index: Some(0),
                    delta,
                    item_id,
                    logprobs: None,
                    output_index,
                    sequence_number,
                    rest: Default::default(),
                }),
            ),
            ResponseDelta::ReasoningText => {
                openai::KnownResponseStreamEvent::ResponseReasoningTextDelta(crate::wire!(
                    openai::ResponseContentDeltaEvent {
                        content_index: 0,
                        delta,
                        item_id,
                        output_index,
                        sequence_number,
                        rest: Default::default(),
                    }
                ))
            }
            ResponseDelta::FunctionArguments => {
                openai::KnownResponseStreamEvent::ResponseFunctionCallArgumentsDelta(crate::wire!(
                    openai::ResponseItemStringDeltaEvent {
                        delta,
                        item_id,
                        output_index,
                        sequence_number,
                        rest: Default::default(),
                    }
                ))
            }
        };
        typed_response_event(event)
    }

    pub(in crate::common::stream) fn response_content_part_added(
        &mut self,
        item_id: String,
        output_index: u32,
        part: openai::ResponseContentPart,
    ) -> Result<OutputEvent, TransformError> {
        let sequence_number = Some(self.next_sequence());
        typed_response_event(openai::KnownResponseStreamEvent::ResponseContentPartAdded(
            crate::wire!(openai::ResponseContentPartEvent {
                content_index: 0,
                item_id,
                output_index,
                part,
                sequence_number,
                rest: Default::default(),
            }),
        ))
    }

    pub(in crate::common::stream) fn response_created(
        &mut self,
        response: openai::ResponseObject,
    ) -> Result<OutputEvent, TransformError> {
        let sequence_number = Some(self.next_sequence());
        typed_response_event(openai::KnownResponseStreamEvent::ResponseCreated(
            crate::wire!(openai::ResponseLifecycleEvent {
                response: Box::new(response),
                sequence_number,
                rest: Default::default(),
            }),
        ))
    }

    pub(in crate::common::stream) fn response_terminal(
        &mut self,
        incomplete: bool,
        response: openai::ResponseObject,
    ) -> Result<OutputEvent, TransformError> {
        let payload = crate::wire!(openai::ResponseLifecycleEvent {
            response: Box::new(response),
            sequence_number: Some(self.next_sequence()),
            rest: Default::default(),
        });
        typed_response_event(if incomplete {
            openai::KnownResponseStreamEvent::ResponseIncomplete(payload)
        } else {
            openai::KnownResponseStreamEvent::ResponseCompleted(payload)
        })
    }

    pub(in crate::common::stream) fn response_output_item_added(
        &mut self,
        item: openai::ResponseItem,
        output_index: u32,
    ) -> Result<OutputEvent, TransformError> {
        let sequence_number = Some(self.next_sequence());
        typed_response_event(openai::KnownResponseStreamEvent::ResponseOutputItemAdded(
            crate::wire!(openai::ResponseOutputItemEvent {
                item: Box::new(item),
                output_index,
                sequence_number,
                rest: Default::default(),
            }),
        ))
    }

    pub(in crate::common::stream) fn response_output_item_done(
        &mut self,
        item: openai::ResponseItem,
        output_index: u32,
    ) -> Result<OutputEvent, TransformError> {
        let sequence_number = Some(self.next_sequence());
        typed_response_event(openai::KnownResponseStreamEvent::ResponseOutputItemDone(
            crate::wire!(openai::ResponseOutputItemEvent {
                item: Box::new(item),
                output_index,
                sequence_number,
                rest: Default::default(),
            }),
        ))
    }
}

fn typed_response_event(
    event: openai::KnownResponseStreamEvent,
) -> Result<OutputEvent, TransformError> {
    Ok(OutputEvent::Responses(openai::ResponseStreamEvent::Known(
        Box::new(event),
    )))
}