From 39a97d4a89d858c956dd215fd8a4f108ca335947 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Wed, 8 Oct 2025 10:41:46 -0700 Subject: [PATCH] add log formatting --- .../rag_agent/src/rag_agent/content_builder_agent.py | 5 ++++- .../rag_agent/src/rag_agent/query_rewriter_agent.py | 5 ++++- .../rag_agent/src/rag_agent/response_generator_agent.py | 5 ++++- 3 files changed, 12 insertions(+), 3 deletions(-) diff --git a/demos/use_cases/rag_agent/src/rag_agent/content_builder_agent.py b/demos/use_cases/rag_agent/src/rag_agent/content_builder_agent.py index a9ec53f4..dd36be2d 100644 --- a/demos/use_cases/rag_agent/src/rag_agent/content_builder_agent.py +++ b/demos/use_cases/rag_agent/src/rag_agent/content_builder_agent.py @@ -13,7 +13,10 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse # Set up logging -logging.basicConfig(level=logging.INFO) +logging.basicConfig( + level=logging.INFO, + format="%(asctime)s - [CONTEXT_BUILDER] - %(levelname)s - %(message)s", +) logger = logging.getLogger(__name__) 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 9820a274..056df952 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 @@ -11,7 +11,10 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse # Set up logging -logging.basicConfig(level=logging.INFO) +logging.basicConfig( + level=logging.INFO, + format="%(asctime)s - [QUERY_REWRITER] - %(levelname)s - %(message)s", +) logger = logging.getLogger(__name__) 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 c3d1ef53..6b1c0561 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 @@ -16,7 +16,10 @@ from .api import ( ) # Set up logging -logging.basicConfig(level=logging.INFO) +logging.basicConfig( + level=logging.INFO, + format="%(asctime)s - [RESPONSE_GENERATOR] - %(levelname)s - %(message)s", +) logger = logging.getLogger(__name__) # Configuration for archgw LLM gateway