Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
File size: 1,060 Bytes
f730cdd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | """
Logging utilities
"""
import logging
import sys
from pathlib import Path
from typing import Optional
def setup_logger(
name: str = "hf_agent", level: int = logging.INFO, log_file: Optional[Path] = None
) -> logging.Logger:
"""Setup and configure logger"""
logger = logging.getLogger(name)
logger.setLevel(level)
# Remove existing handlers
logger.handlers = []
# Console handler
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(level)
console_format = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
)
console_handler.setFormatter(console_format)
logger.addHandler(console_handler)
# File handler if log_file specified
if log_file:
log_file.parent.mkdir(parents=True, exist_ok=True)
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(level)
file_handler.setFormatter(console_format)
logger.addHandler(file_handler)
return logger
|