diff --git a/crates/common/src/configuration.rs b/crates/common/src/configuration.rs index c5b2e9eb..af32a718 100644 --- a/crates/common/src/configuration.rs +++ b/crates/common/src/configuration.rs @@ -12,7 +12,7 @@ pub struct Overrides { #[derive(Debug, Clone, Serialize, Deserialize, Default)] pub struct Tracing { pub sampling_rate: Option, - pub trace_arch: Option, + pub trace_arch_internal: Option, } #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Hash, Default)] diff --git a/crates/prompt_gateway/src/stream_context.rs b/crates/prompt_gateway/src/stream_context.rs index dac50ad5..b3a60875 100644 --- a/crates/prompt_gateway/src/stream_context.rs +++ b/crates/prompt_gateway/src/stream_context.rs @@ -176,7 +176,7 @@ impl StreamContext { headers.push((REQUEST_ID_HEADER, self.request_id.as_ref().unwrap())); } - if self.trace_arch() && self.traceparent.is_some() { + if self.trace_arch_internal() && self.traceparent.is_some() { headers.push((TRACE_PARENT_HEADER, self.traceparent.as_ref().unwrap())); } @@ -305,7 +305,7 @@ impl StreamContext { headers.push((REQUEST_ID_HEADER, self.request_id.as_ref().unwrap())); } - if self.trace_arch() && self.traceparent.is_some() { + if self.trace_arch_internal() && self.traceparent.is_some() { headers.push((TRACE_PARENT_HEADER, self.traceparent.as_ref().unwrap())); } @@ -325,10 +325,10 @@ impl StreamContext { } } - fn trace_arch(&self) -> bool { + fn trace_arch_internal(&self) -> bool { match self.tracing.as_ref() { - Some(tracing) => match tracing.trace_arch.as_ref() { - Some(trace_arch) => *trace_arch, + Some(tracing) => match tracing.trace_arch_internal.as_ref() { + Some(trace_arch_internal) => *trace_arch_internal, None => false, }, None => false, @@ -515,7 +515,7 @@ impl StreamContext { headers.push((REQUEST_ID_HEADER, self.request_id.as_ref().unwrap())); } - if self.trace_arch() && self.traceparent.is_some() { + if self.trace_arch_internal() && self.traceparent.is_some() { headers.push((TRACE_PARENT_HEADER, self.traceparent.as_ref().unwrap())); } @@ -666,7 +666,7 @@ impl StreamContext { headers.push((REQUEST_ID_HEADER, self.request_id.as_ref().unwrap())); } - if self.trace_arch() && self.traceparent.is_some() { + if self.trace_arch_internal() && self.traceparent.is_some() { headers.push((TRACE_PARENT_HEADER, self.traceparent.as_ref().unwrap())); } @@ -846,7 +846,7 @@ impl StreamContext { headers.push((REQUEST_ID_HEADER, self.request_id.as_ref().unwrap())); } - if self.trace_arch() && self.traceparent.is_some() { + if self.trace_arch_internal() && self.traceparent.is_some() { headers.push((TRACE_PARENT_HEADER, self.traceparent.as_ref().unwrap())); } diff --git a/demos/weather_forecast/arch_config.yaml b/demos/weather_forecast/arch_config.yaml index 41af8338..574a9d35 100644 --- a/demos/weather_forecast/arch_config.yaml +++ b/demos/weather_forecast/arch_config.yaml @@ -74,4 +74,4 @@ prompt_targets: tracing: random_sampling: 100 - trace_arch: true + trace_arch_internal: true