diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index b3a72d4d..a0b03129 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -1095,7 +1095,7 @@ def create_app(args): }, log_level=args.log_level, namespace_prefix=args.namespace_prefix, - is_managed_by_server=True, + auto_manage_storages_states=False, ) else: rag = LightRAG( @@ -1127,7 +1127,7 @@ def create_app(args): }, log_level=args.log_level, namespace_prefix=args.namespace_prefix, - is_managed_by_server=True, + auto_manage_storages_states=False, ) async def pipeline_enqueue_file(file_path: Path) -> bool: diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index ca1ac7ce..23bf1c8d 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -349,8 +349,9 @@ class LightRAG: # Extensions addon_params: dict[str, Any] = field(default_factory=dict) - # Ownership - is_managed_by_server: bool = False + # Storages Management + auto_manage_storages_states: bool = True + """If True, lightrag will automatically calls initialize_storages and finalize_storages at the appropriate times.""" """Dictionary for additional parameters and extensions.""" convert_response_to_json_func: Callable[[str], dict[str, Any]] = ( @@ -557,13 +558,13 @@ class LightRAG: self.storages_status = StoragesStatus.CREATED # Initialize storages - if not self.is_managed_by_server: + if self.auto_manage_storages_states: loop = always_get_an_event_loop() loop.run_until_complete(self.initialize_storages()) def __del__(self): # Finalize storages - if not self.is_managed_by_server: + if self.auto_manage_storages_states: loop = always_get_an_event_loop() loop.run_until_complete(self.finalize_storages())