mirror of
https://github.com/openai/codex.git
synced 2026-09-09 15:58:47 +00:00
4861236f06
## What changed - Add optional W3C `traceparent` and `tracestate` fields to relay frames. - Copy trace context from JSON-RPC requests onto relay data frames. - For encrypted requests split across multiple Noise records, attach the context only to the first record while keeping the request payload encrypted. ## Testing - Cover trace propagation for both Noise relay paths, including fragmented encrypted requests. GitOrigin-RevId: a61bbbefef31e2e7e93a43f439c5f296700feb7b
66 lines
2.2 KiB
Rust
66 lines
2.2 KiB
Rust
// This file is @generated by prost-build.
|
|
#[derive(Clone, PartialEq, ::prost::Message)]
|
|
pub struct RelayMessageFrame {
|
|
#[prost(uint32, tag = "1")]
|
|
pub version: u32,
|
|
#[prost(string, tag = "2")]
|
|
pub stream_id: ::prost::alloc::string::String,
|
|
#[prost(uint32, tag = "3")]
|
|
pub ack: u32,
|
|
#[prost(uint32, tag = "4")]
|
|
pub ack_bits: u32,
|
|
#[prost(oneof = "relay_message_frame::Body", tags = "5, 6, 7, 8, 9, 10")]
|
|
pub body: ::core::option::Option<relay_message_frame::Body>,
|
|
#[prost(string, optional, tag = "11")]
|
|
pub traceparent: ::core::option::Option<::prost::alloc::string::String>,
|
|
#[prost(string, optional, tag = "12")]
|
|
pub tracestate: ::core::option::Option<::prost::alloc::string::String>,
|
|
}
|
|
pub mod relay_message_frame {
|
|
#[derive(Clone, PartialEq, ::prost::Oneof)]
|
|
pub enum Body {
|
|
#[prost(message, tag = "5")]
|
|
Data(super::RelayData),
|
|
#[prost(message, tag = "6")]
|
|
AckFrame(super::RelayAck),
|
|
#[prost(message, tag = "7")]
|
|
Resume(super::RelayResume),
|
|
#[prost(message, tag = "8")]
|
|
Reset(super::RelayReset),
|
|
#[prost(message, tag = "9")]
|
|
Heartbeat(super::RelayHeartbeat),
|
|
#[prost(message, tag = "10")]
|
|
Handshake(super::RelayHandshake),
|
|
}
|
|
}
|
|
#[derive(Clone, PartialEq, ::prost::Message)]
|
|
pub struct RelayData {
|
|
#[prost(uint32, tag = "1")]
|
|
pub seq: u32,
|
|
#[prost(uint32, tag = "2")]
|
|
pub segment_index: u32,
|
|
#[prost(uint32, tag = "3")]
|
|
pub segment_count: u32,
|
|
#[prost(bytes = "vec", tag = "4")]
|
|
pub payload: ::prost::alloc::vec::Vec<u8>,
|
|
}
|
|
#[derive(Clone, PartialEq, Eq, Hash, ::prost::Message)]
|
|
pub struct RelayAck {}
|
|
#[derive(Clone, Copy, PartialEq, Eq, Hash, ::prost::Message)]
|
|
pub struct RelayResume {
|
|
#[prost(uint32, tag = "1")]
|
|
pub next_seq: u32,
|
|
}
|
|
#[derive(Clone, PartialEq, Eq, Hash, ::prost::Message)]
|
|
pub struct RelayReset {
|
|
#[prost(string, tag = "1")]
|
|
pub reason: ::prost::alloc::string::String,
|
|
}
|
|
#[derive(Clone, PartialEq, Eq, Hash, ::prost::Message)]
|
|
pub struct RelayHeartbeat {}
|
|
#[derive(Clone, PartialEq, Eq, Hash, ::prost::Message)]
|
|
pub struct RelayHandshake {
|
|
#[prost(bytes = "vec", tag = "1")]
|
|
pub payload: ::prost::alloc::vec::Vec<u8>,
|
|
}
|