apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

rename trace_arch => trace_arch_internal

Browse source
This commit is contained in:
Adil Hafeez 2024-11-07 14:43:48 -08:00
parent 270ff1f5de
commit 460d978f8a
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
crates/common/src/configuration.rs
View file

@ -12,7 +12,7 @@ pub struct Overrides {
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct Tracing {
pub sampling_rate: Option<f64>,
pub trace_arch: Option<bool>,
pub trace_arch_internal: Option<bool>,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Hash, Default)]

16
crates/prompt_gateway/src/stream_context.rs
View file

@ -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()));
}

2
demos/weather_forecast/arch_config.yaml
View file

@ -74,4 +74,4 @@ prompt_targets:
tracing:
random_sampling: 100
trace_arch: true
trace_arch_internal: true