jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Logging: Remove preferences global

Browse source 
This is no longer needed because config is a singleton.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-09-14 21:49:44 -04:00
parent a09dd802c2
commit 6f28cfe905
1 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
common/gen_logging.py
View file

@ -7,17 +7,14 @@ from typing import Optional
from common.tabby_config import config
# Global logging preferences constant
PREFERENCES = config.logging
def broadcast_status():
"""Broadcasts the current logging status"""
enabled = []
if PREFERENCES.log_prompt:
if config.logging.log_prompt:
enabled.append("prompts")
if PREFERENCES.log_generation_params:
if config.logging.log_generation_params:
enabled.append("generation params")
if len(enabled) > 0:
@ -28,13 +25,13 @@ def broadcast_status():
def log_generation_params(**kwargs):
"""Logs generation parameters to console."""
if PREFERENCES.log_generation_params:
if config.logging.log_generation_params:
logger.info(f"Generation options: {kwargs}\n")
def log_prompt(prompt: str, request_id: str, negative_prompt: Optional[str]):
"""Logs the prompt to console."""
if PREFERENCES.log_prompt:
if config.logging.log_prompt:
formatted_prompt = "\n" + prompt
logger.info(
f"Prompt (ID: {request_id}): {formatted_prompt if prompt else 'Empty'}\n"
@ -47,7 +44,7 @@ def log_prompt(prompt: str, request_id: str, negative_prompt: Optional[str]):
def log_response(request_id: str, response: str):
"""Logs the response to console."""
if PREFERENCES.log_prompt:
if config.logging.log_prompt:
formatted_response = "\n" + response
logger.info(
f"Response (ID: {request_id}): "