gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename is_managed_by_server to auto_manage_storages_states

Browse Source
This commit is contained in:
ArnoChen
2025-02-19 05:27:38 +08:00
parent bfc548edf2
commit ca170ce783
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/api/lightrag_server.py
View File

@@ -1095,7 +1095,7 @@ def create_app(args):
}, },
log_level=args.log_level, log_level=args.log_level,
namespace_prefix=args.namespace_prefix, namespace_prefix=args.namespace_prefix,
is_managed_by_server=True, auto_manage_storages_states=False,
) )
else: else:
rag = LightRAG( rag = LightRAG(
@@ -1127,7 +1127,7 @@ def create_app(args):
}, },
log_level=args.log_level, log_level=args.log_level,
namespace_prefix=args.namespace_prefix, namespace_prefix=args.namespace_prefix,
is_managed_by_server=True, auto_manage_storages_states=False,
) )
async def pipeline_enqueue_file(file_path: Path) -> bool: async def pipeline_enqueue_file(file_path: Path) -> bool:

9
lightrag/lightrag.py
View File

@@ -349,8 +349,9 @@ class LightRAG:
# Extensions # Extensions
addon_params: dict[str, Any] = field(default_factory=dict) addon_params: dict[str, Any] = field(default_factory=dict)
# Ownership # Storages Management
is_managed_by_server: bool = False 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.""" """Dictionary for additional parameters and extensions."""
convert_response_to_json_func: Callable[[str], dict[str, Any]] = ( convert_response_to_json_func: Callable[[str], dict[str, Any]] = (
@@ -557,13 +558,13 @@ class LightRAG:
self.storages_status = StoragesStatus.CREATED self.storages_status = StoragesStatus.CREATED
# Initialize storages # Initialize storages
if not self.is_managed_by_server: if self.auto_manage_storages_states:
loop = always_get_an_event_loop() loop = always_get_an_event_loop()
loop.run_until_complete(self.initialize_storages()) loop.run_until_complete(self.initialize_storages())
def __del__(self): def __del__(self):
# Finalize storages # Finalize storages
if not self.is_managed_by_server: if self.auto_manage_storages_states:
loop = always_get_an_event_loop() loop = always_get_an_event_loop()
loop.run_until_complete(self.finalize_storages()) loop.run_until_complete(self.finalize_storages())