gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: custom log dir

Browse source
This commit is contained in:
ultrageopro 2025-02-02 14:04:24 +03:00
parent c07b5522fe
commit 35c4115441
No known key found for this signature in database
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lightrag/lightrag.py
View file

@ -126,8 +126,10 @@ class LightRAG:
vector_storage: str = field(default="NanoVectorDBStorage") vector_storage: str = field(default="NanoVectorDBStorage")
graph_storage: str = field(default="NetworkXStorage") graph_storage: str = field(default="NetworkXStorage")
# logging
current_log_level = logger.level current_log_level = logger.level
log_level: str = field(default=current_log_level) log_level: str = field(default=current_log_level)
logging_dir: str = field(default="logs")
# text chunking # text chunking
chunk_token_size: int = 1200 chunk_token_size: int = 1200
@ -182,10 +184,11 @@ class LightRAG:
chunking_func_kwargs: dict = field(default_factory=dict) chunking_func_kwargs: dict = field(default_factory=dict)
def __post_init__(self): def __post_init__(self):
log_file = os.path.join("lightrag.log") os.makedirs(self.logging_dir, exist_ok=True)
log_file = os.path.join(self.logging_dir, "lightrag.log")
set_logger(log_file) set_logger(log_file)
logger.setLevel(self.log_level)
logger.setLevel(self.log_level)
logger.info(f"Logger initialized for working directory: {self.working_dir}") logger.info(f"Logger initialized for working directory: {self.working_dir}")
if not os.path.exists(self.working_dir): if not os.path.exists(self.working_dir):
logger.info(f"Creating working directory {self.working_dir}") logger.info(f"Creating working directory {self.working_dir}")