From fc6dff04fff09f1caf843c6007d9f31c54e6da20 Mon Sep 17 00:00:00 2001 From: zrguo <49157727+LarFii@users.noreply.github.com> Date: Mon, 17 Feb 2025 15:10:15 +0800 Subject: [PATCH] fix logger level --- .gitignore | 1 + lightrag/api/lightrag_server.py | 5 ++++- lightrag/operate.py | 2 +- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 58c9e17e..e8130e18 100644 --- a/.gitignore +++ b/.gitignore @@ -31,6 +31,7 @@ log/ .pytest_cache/ .ruff_cache/ .gradio/ +.history/ temp/ # IDE / Editor Files diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index b7a83e04..3928d065 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -63,7 +63,10 @@ from ..kg.tidb_impl import ( ) # Load environment variables -load_dotenv(override=True) +try: + load_dotenv(override=True) +except Exception as e: + logger.warning(f"Failed to load .env file: {e}") # Initialize config parser config = configparser.ConfigParser() diff --git a/lightrag/operate.py b/lightrag/operate.py index cc5dffe7..678194bd 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -1662,7 +1662,7 @@ async def naive_query( return sys_prompt len_of_prompts = len(encode_string_by_tiktoken(query + sys_prompt)) - logger.info(f"[naive_query]Prompt Tokens: {len_of_prompts}") + logger.debug(f"[naive_query]Prompt Tokens: {len_of_prompts}") response = await use_model_func( query,