fix: respect user-specified log level in set_logger

Previously, the set_logger function would always set the log level to DEBUG, overriding any user-specified log level.
This commit is contained in:
yangdx
2025-02-21 14:46:27 +08:00
parent 1749678384
commit cff229a806
2 changed files with 10 additions and 5 deletions

View File

@@ -57,11 +57,17 @@ logger = logging.getLogger("lightrag")
logging.getLogger("httpx").setLevel(logging.WARNING)
def set_logger(log_file: str):
logger.setLevel(logging.DEBUG)
def set_logger(log_file: str, level: int = logging.DEBUG):
"""Set up file logging with the specified level.
Args:
log_file: Path to the log file
level: Logging level (e.g. logging.DEBUG, logging.INFO)
"""
logger.setLevel(level)
file_handler = logging.FileHandler(log_file, encoding="utf-8")
file_handler.setLevel(logging.DEBUG)
file_handler.setLevel(level)
formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"