From b3a19b1df0cf33029172ee144523669ee83e5f67 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Wed, 8 Oct 2025 11:20:13 -0700
Subject: [PATCH] fix logs

---
 .../rag_agent/src/rag_agent/context_builder_agent.py         | 5 +++--
 .../rag_agent/src/rag_agent/query_rewriter_agent.py          | 4 ++--
 .../rag_agent/src/rag_agent/response_generator_agent.py      | 2 +-
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py b/demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py
index ea21fa3f..b225aeed 100644
--- a/demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py
+++ b/demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py
@@ -15,7 +15,7 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse
 # Set up logging
 logging.basicConfig(
     level=logging.INFO,
-    format="%(asctime)s - [CONTEXT_BUILDER] - %(levelname)s - %(message)s",
+    format="%(asctime)s - [CONTEXT_BUILDER]    - %(levelname)s - %(message)s",
 )
 logger = logging.getLogger(__name__)
 
@@ -252,6 +252,7 @@ if __name__ == "__main__":
 def start_server(host: str = "localhost", port: int = 8000):
     """Start the REST server."""
     load_knowledge_base()
+    # Rename the uvicorn.error logger
     uvicorn.run(
         app,
         host=host,
@@ -261,7 +262,7 @@ def start_server(host: str = "localhost", port: int = 8000):
             "disable_existing_loggers": False,
             "formatters": {
                 "default": {
-                    "format": "%(asctime)s - [CONTEXT_BUILDER] - %(name)s - %(levelname)s - %(message)s",
+                    "format": "%(asctime)s - [CONTEXT_BUILDER]    - %(levelname)s - %(message)s",
                 },
             },
             "handlers": {
diff --git a/demos/use_cases/rag_agent/src/rag_agent/query_rewriter_agent.py b/demos/use_cases/rag_agent/src/rag_agent/query_rewriter_agent.py
index 446227fe..6b2447dd 100644
--- a/demos/use_cases/rag_agent/src/rag_agent/query_rewriter_agent.py
+++ b/demos/use_cases/rag_agent/src/rag_agent/query_rewriter_agent.py
@@ -13,7 +13,7 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse
 # Set up logging
 logging.basicConfig(
     level=logging.INFO,
-    format="%(asctime)s - [QUERY_REWRITER] - %(levelname)s - %(message)s",
+    format="%(asctime)s - [QUERY_REWRITER]     - %(levelname)s - %(message)s",
 )
 logger = logging.getLogger(__name__)
 
@@ -170,7 +170,7 @@ def start_server(host: str = "localhost", port: int = 8000):
             "disable_existing_loggers": False,
             "formatters": {
                 "default": {
-                    "format": "%(asctime)s - [QUERY_REWRITER] - %(name)s - %(levelname)s - %(message)s",
+                    "format": "%(asctime)s - [QUERY_REWRITER]     - %(levelname)s - %(message)s",
                 },
             },
             "handlers": {
diff --git a/demos/use_cases/rag_agent/src/rag_agent/response_generator_agent.py b/demos/use_cases/rag_agent/src/rag_agent/response_generator_agent.py
index 2e430f49..735f11da 100644
--- a/demos/use_cases/rag_agent/src/rag_agent/response_generator_agent.py
+++ b/demos/use_cases/rag_agent/src/rag_agent/response_generator_agent.py
@@ -284,7 +284,7 @@ def start_server(host: str = "localhost", port: int = 8000):
             "disable_existing_loggers": False,
             "formatters": {
                 "default": {
-                    "format": "%(asctime)s - [RESPONSE_GENERATOR] - %(name)s - %(levelname)s - %(message)s",
+                    "format": "%(asctime)s - [RESPONSE_GENERATOR] - %(levelname)s - %(message)s",
                 },
             },
             "handlers": {