yangdx commited on
Commit
d2b7a97
·
1 Parent(s): bf7cb9a

feat: automatically initialize API manager in single process mode

Browse files

- Add manager init check in __post_init__
- Call initialize_manager if needed
- Add info log message for init
- Ensure API manager is ready for use

Files changed (1) hide show
  1. lightrag/lightrag.py +6 -0
lightrag/lightrag.py CHANGED
@@ -267,6 +267,12 @@ class LightRAG:
267
  _storages_status: StoragesStatus = field(default=StoragesStatus.NOT_CREATED)
268
 
269
  def __post_init__(self):
 
 
 
 
 
 
270
  os.makedirs(os.path.dirname(self.log_file_path), exist_ok=True)
271
  set_logger(self.log_file_path, self.log_level)
272
  logger.info(f"Logger initialized for working directory: {self.working_dir}")
 
267
  _storages_status: StoragesStatus = field(default=StoragesStatus.NOT_CREATED)
268
 
269
  def __post_init__(self):
270
+ # Initialize manager if needed
271
+ from lightrag.api.utils_api import manager, initialize_manager
272
+ if manager is None:
273
+ initialize_manager()
274
+ logger.info("Initialized manager for single process mode")
275
+
276
  os.makedirs(os.path.dirname(self.log_file_path), exist_ok=True)
277
  set_logger(self.log_file_path, self.log_level)
278
  logger.info(f"Logger initialized for working directory: {self.working_dir}")