theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-03-15 00:07:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Logging: Add newlines to Prompt and Response

Browse Source 
Makes things clearer rather than adding an extra space.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-12-22 23:55:22 -05:00
parent 80ef379721
commit 98a7b951b9
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gen_logging.py
View File

@@ -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")