From 460d978f8a126da1544a1386c279a8a5eb053b03 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Thu, 7 Nov 2024 14:43:48 -0800
Subject: [PATCH] rename trace_arch => trace_arch_internal

---
 crates/common/src/configuration.rs          |  2 +-
 crates/prompt_gateway/src/stream_context.rs | 16 ++++++++--------
 demos/weather_forecast/arch_config.yaml     |  2 +-
 3 files changed, 10 insertions(+), 10 deletions(-)

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<f64>,
-    pub trace_arch: Option<bool>,
+    pub trace_arch_internal: Option<bool>,
 }
 
 #[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