poemsforaphrodite's picture
Upload folder using huggingface_hub
b72ab63 verified
raw
history blame
664 Bytes
import logging
from typing import Union
from rich.console import Console
from rich.logging import RichHandler
def setup_logging(terminal_width: Union[int, None] = None) -> None:
logger = logging.getLogger("fastapi_cli")
console = Console(width=terminal_width) if terminal_width else None
rich_handler = RichHandler(
show_time=False,
rich_tracebacks=True,
tracebacks_show_locals=True,
markup=True,
show_path=False,
console=console,
)
rich_handler.setFormatter(logging.Formatter("%(message)s"))
logger.addHandler(rich_handler)
logger.setLevel(logging.INFO)
logger.propagate = False