ultrageopro commited on
Commit
403f026
·
unverified ·
1 Parent(s): a8fc721

feat: custom log dir

Browse files
Files changed (1) hide show
  1. lightrag/lightrag.py +5 -2
lightrag/lightrag.py CHANGED
@@ -126,8 +126,10 @@ class LightRAG:
126
  vector_storage: str = field(default="NanoVectorDBStorage")
127
  graph_storage: str = field(default="NetworkXStorage")
128
 
 
129
  current_log_level = logger.level
130
  log_level: str = field(default=current_log_level)
 
131
 
132
  # text chunking
133
  chunk_token_size: int = 1200
@@ -182,10 +184,11 @@ class LightRAG:
182
  chunking_func_kwargs: dict = field(default_factory=dict)
183
 
184
  def __post_init__(self):
185
- log_file = os.path.join("lightrag.log")
 
186
  set_logger(log_file)
187
- logger.setLevel(self.log_level)
188
 
 
189
  logger.info(f"Logger initialized for working directory: {self.working_dir}")
190
  if not os.path.exists(self.working_dir):
191
  logger.info(f"Creating working directory {self.working_dir}")
 
126
  vector_storage: str = field(default="NanoVectorDBStorage")
127
  graph_storage: str = field(default="NetworkXStorage")
128
 
129
+ # logging
130
  current_log_level = logger.level
131
  log_level: str = field(default=current_log_level)
132
+ logging_dir: str = field(default="logs")
133
 
134
  # text chunking
135
  chunk_token_size: int = 1200
 
184
  chunking_func_kwargs: dict = field(default_factory=dict)
185
 
186
  def __post_init__(self):
187
+ os.makedirs(self.logging_dir, exist_ok=True)
188
+ log_file = os.path.join(self.logging_dir, "lightrag.log")
189
  set_logger(log_file)
 
190
 
191
+ logger.setLevel(self.log_level)
192
  logger.info(f"Logger initialized for working directory: {self.working_dir}")
193
  if not os.path.exists(self.working_dir):
194
  logger.info(f"Creating working directory {self.working_dir}")