diff --git a/dsRagAnything/Test.py b/dsRagAnything/Test.py index 444f385d..2627191b 100644 --- a/dsRagAnything/Test.py +++ b/dsRagAnything/Test.py @@ -8,11 +8,31 @@ from lightrag.utils import EmbeddingFunc import Config.Config # 控制日志输出 +# 设置根日志记录器的级别为INFO,这样所有子记录器的日志都会被捕获 +root_logger = logging.getLogger() +root_logger.setLevel(logging.INFO) + +# 确保根日志记录器有处理器 +if not root_logger.handlers: + handler = logging.StreamHandler() + 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) -handler = logging.StreamHandler() -handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) -logger.addHandler(handler) +if not logger.handlers: + handler = logging.StreamHandler() + handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) + logger.addHandler(handler) + +# 另外,尝试捕获raganything库可能使用的特定记录器 +raganything_logger = logging.getLogger('raganything') +raganything_logger.setLevel(logging.INFO) +if not raganything_logger.handlers: + handler = logging.StreamHandler() + handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) + raganything_logger.addHandler(handler) async def main(): # 设置 API 配置