openai/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-11-08 20:40:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(app-server, core): add more spans

Browse Source
This commit is contained in:
Owen Lin
2026-03-12 15:28:35 -07:00
parent a30b807efe
commit b1e3b1d08d
15 changed files with 434 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files
codex-rs/codex-client/src/default_client.rs
+10 -9
View File
@@ -1,12 +1,12 @@
use http::Error as HttpError;
use http::HeaderMap;
use http::HeaderName;
use http::HeaderValue;
use opentelemetry::global;
use opentelemetry::propagation::Injector;
use reqwest::IntoUrl;
use reqwest::Method;
use reqwest::Response;
use reqwest::header::HeaderMap;
use reqwest::header::HeaderName;
use reqwest::header::HeaderValue;
use serde::Serialize;
use std::fmt::Display;
use std::time::Duration;
@@ -154,14 +154,15 @@ impl<'a> Injector for HeaderMapInjector<'a> {
}
}
pub fn inject_current_span_trace_headers(headers: &mut HeaderMap) {
global::get_text_map_propagator(|prop| {
prop.inject_context(&Span::current().context(), &mut HeaderMapInjector(headers));
});
}
fn trace_headers() -> HeaderMap {
let mut headers = HeaderMap::new();
global::get_text_map_propagator(|prop| {
prop.inject_context(
&Span::current().context(),
&mut HeaderMapInjector(&mut headers),
);
});
inject_current_span_trace_headers(&mut headers);
headers
}
codex-rs/codex-client/src/lib.rs
+1
View File
@@ -8,6 +8,7 @@ mod transport;
pub use crate::default_client::CodexHttpClient;
pub use crate::default_client::CodexRequestBuilder;
pub use crate::default_client::inject_current_span_trace_headers;
pub use crate::error::StreamError;
pub use crate::error::TransportError;
pub use crate::request::Request;