From 2de75d18db48bd615230547d4b9a5c224ad2fb60 Mon Sep 17 00:00:00 2001
From: Salman Paracha <salmanparacha@MacBook-Pro-41.local>
Date: Fri, 5 Sep 2025 21:52:46 -0700
Subject: [PATCH] fixing logs

---
 crates/brightstaff/src/handlers/chat_completions.rs | 9 ++++-----
 crates/llm_gateway/src/stream_context.rs            | 2 +-
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/crates/brightstaff/src/handlers/chat_completions.rs b/crates/brightstaff/src/handlers/chat_completions.rs
index afabb64f..4c415b80 100644
--- a/crates/brightstaff/src/handlers/chat_completions.rs
+++ b/crates/brightstaff/src/handlers/chat_completions.rs
@@ -64,7 +64,6 @@ pub async fn chat(
     // remove metadata from the request
     let mut chat_request_user_preferences_removed = chat_request_parsed;
     if let Some(metadata) = chat_request_user_preferences_removed.get_mut("metadata") {
-        debug!("Removing metadata from request");
         if let Some(m) = metadata.as_object_mut() {
             m.remove("archgw_preference_config");
             debug!("Removed archgw_preference_config from metadata");
@@ -72,15 +71,15 @@ pub async fn chat(
 
         // if metadata is empty, remove it
         if metadata.as_object().map_or(false, |m| m.is_empty()) {
-            debug!("Removing empty metadata from request");
             chat_request_user_preferences_removed
                 .as_object_mut()
                 .map(|m| m.remove("metadata"));
+            debug!("Removed empty metadata from request");
         }
     }
 
     debug!(
-        "arch-router request received: {}",
+        "[BRIGHTSTAFF -> ARCH_ROUTER] REQ: {}",
         &serde_json::to_string(&chat_completion_request).unwrap()
     );
 
@@ -151,8 +150,8 @@ pub async fn chat(
     };
 
     debug!(
-        "sending request to llm provider: {}, with model hint: {}",
-            full_qualified_llm_provider_url, model_name
+        "[BRIGHTSTAFF -> ARCH_ROUTER] URL: {}, Model Hint: {}",
+        full_qualified_llm_provider_url, model_name
     );
 
     request_headers.insert(
diff --git a/crates/llm_gateway/src/stream_context.rs b/crates/llm_gateway/src/stream_context.rs
index 46ef1ecf..b216e9c8 100644
--- a/crates/llm_gateway/src/stream_context.rs
+++ b/crates/llm_gateway/src/stream_context.rs
@@ -120,7 +120,7 @@ impl StreamContext {
         ));
 
         info!(
-            "[ARCHGW_REQ_ID:{}] PROVIDER_SELECTION: hint='{}' -> selected='{}'",
+            "[ARCHGW_REQ_ID:{}] PROVIDER_SELECTION: Hint='{}' -> Selected='{}'",
             self.request_identifier(),
             self.get_http_request_header(ARCH_PROVIDER_HINT_HEADER)
                 .unwrap_or("none".to_string()),