Logging: Add newlines to Prompt and Response

Makes things clearer rather than adding an extra space. Signed-off-by: kingbri <[email protected]>
theroyallab · Dec 23, 2023 · 98a7b95 · 98a7b95
1 parent 80ef379
commit 98a7b95
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/gen_logging.py b/gen_logging.py
@@ -56,10 +56,12 @@ def log_generation_params(**kwargs):
 def log_prompt(prompt: str):
     """Logs the prompt to console."""
     if CONFIG.prompt:
-        logger.info(f"Prompt: {prompt if prompt else 'Empty'}\n")
+        formatted_prompt = "\n" + prompt
+        logger.info(f"Prompt: {formatted_prompt if prompt else 'Empty'}\n")
 
 
 def log_response(response: str):
     """Logs the response to console."""
     if CONFIG.prompt:
-        logger.info(f"Response: {response if response else 'Empty'}\n")
+        formatted_response = "\n" + response
+        logger.info(f"Response: {formatted_response if response else 'Empty'}\n")