ultrageopro
commited on
feat: custom log dir
Browse files- 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 |
-
|
|
|
|
| 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}")
|