rename is_managed_by_server to auto_manage_storages_states

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

View File

@@ -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:

View File

@@ -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())