diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index db61788a..e5c67df4 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -263,9 +263,8 @@ class LightRAG: _storages_status: StoragesStatus = field(default=StoragesStatus.NOT_CREATED) def __post_init__(self): - logger.setLevel(self.log_level) os.makedirs(os.path.dirname(self.log_file_path), exist_ok=True) - set_logger(self.log_file_path) + set_logger(self.log_file_path, self.log_level) logger.info(f"Logger initialized for working directory: {self.working_dir}") if not os.path.exists(self.working_dir): diff --git a/lightrag/utils.py b/lightrag/utils.py index d402d14c..ae7e8dce 100644 --- a/lightrag/utils.py +++ b/lightrag/utils.py @@ -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"