diff --git a/dsRagAnything/Test.py b/dsRagAnything/Test.py index 09d5c448..9382f74e 100644 --- a/dsRagAnything/Test.py +++ b/dsRagAnything/Test.py @@ -4,10 +4,9 @@ import logging from raganything import RAGAnything, RAGAnythingConfig from lightrag.llm.openai import openai_complete_if_cache, openai_embed from lightrag.utils import EmbeddingFunc - +from logging.handlers import RotatingFileHandler# 导入RotatingFileHandler用于日志轮转 import Config.Config -# 控制日志输出 # 设置根日志记录器的级别为INFO,这样所有子记录器的日志都会被捕获 root_logger = logging.getLogger() root_logger.setLevel(logging.INFO) @@ -18,13 +17,25 @@ if not root_logger.handlers: handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) root_logger.addHandler(handler) + # 同时保持原有的ragAnything日志记录器配置 logger = logging.getLogger('ragAnything') logger.setLevel(logging.INFO) if not logger.handlers: - handler = logging.StreamHandler() - handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) - logger.addHandler(handler) + # 控制台输出处理器 + console_handler = logging.StreamHandler() + console_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) + logger.addHandler(console_handler) + # 循环滚动文件处理器(控制在200K左右) + file_handler = RotatingFileHandler( + 'raganything.log', + maxBytes=200*1024, # 200KB + backupCount=5, # 最多保留5个备份文件 + encoding='utf-8', + delay=True # 延迟创建文件,直到有日志输出 + ) + file_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) + logger.addHandler(file_handler) async def main():