theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-29 18:51:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

make log level a global var

Browse Source
This commit is contained in:
AlpinDale
2024-04-01 23:07:30 +00:00
parent 031349133b
commit 6c4a1a9c70
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
common/logger.py
View File

@@ -19,7 +19,7 @@ from rich.progress import (
from common.utils import unwrap from common.utils import unwrap
RICH_CONSOLE = Console() RICH_CONSOLE = Console()
LOG_LEVEL = os.getenv("TABBY_LOG_LEVEL", "INFO")a
def get_loading_progress_bar(): def get_loading_progress_bar():
"""Gets a pre-made progress bar for loading tasks.""" """Gets a pre-made progress bar for loading tasks."""
@@ -88,7 +88,7 @@ UVICORN_LOG_CONFIG = {
"class": f"{UvicornLoggingHandler.__module__}.{UvicornLoggingHandler.__qualname__}", # noqa "class": f"{UvicornLoggingHandler.__module__}.{UvicornLoggingHandler.__qualname__}", # noqa
}, },
}, },
"root": {"handlers": ["uvicorn"], "propagate": False, "level": "INFO"}, "root": {"handlers": ["uvicorn"], "propagate": False, "level": LOG_LEVEL},
} }
@@ -96,11 +96,10 @@ def setup_logger():
"""Bootstrap the logger.""" """Bootstrap the logger."""
logger.remove() logger.remove()
log_level = os.getenv("TABBY_LOG_LEVEL", "INFO")
logger.add( logger.add(
RICH_CONSOLE.print, RICH_CONSOLE.print,
level=log_level, level=LOG_LEVEL,
format=_log_formatter, format=_log_formatter,
colorize=True, colorize=True,
) )