gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Rename allow_create to first_initialization for clarity

Browse source
This commit is contained in:
yangdx 2025-08-23 02:34:39 +08:00
parent 3fca3be09b
commit 059003c906
2 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/kg/shared_storage.py
View file

@ -1059,7 +1059,7 @@ async def initialize_pipeline_status():
Initialize pipeline namespace with default values. Initialize pipeline namespace with default values.
This function is called during FASTAPI lifespan for each worker. This function is called during FASTAPI lifespan for each worker.
""" """
pipeline_namespace = await get_namespace_data("pipeline_status", allow_create=True) pipeline_namespace = await get_namespace_data("pipeline_status", first_init=True)
async with get_internal_lock(): async with get_internal_lock():
# Check if already initialized by checking for required fields # Check if already initialized by checking for required fields
@ -1194,7 +1194,9 @@ async def try_initialize_namespace(namespace: str) -> bool:
return False return False
async def get_namespace_data(namespace: str, allow_create: bool = False) -> Dict[str, Any]: async def get_namespace_data(
namespace: str, first_init: bool = False
) -> Dict[str, Any]:
"""get the shared data reference for specific namespace """get the shared data reference for specific namespace
Args: Args:
@ -1212,7 +1214,7 @@ async def get_namespace_data(namespace: str, allow_create: bool = False) -> Dict
async with get_internal_lock(): async with get_internal_lock():
if namespace not in _shared_dicts: if namespace not in _shared_dicts:
# Special handling for pipeline_status namespace # Special handling for pipeline_status namespace
if namespace == "pipeline_status" and not allow_create: if namespace == "pipeline_status" and not first_init:
# Check if pipeline_status should have been initialized but wasn't # Check if pipeline_status should have been initialized but wasn't
# This helps users understand they need to call initialize_pipeline_status() # This helps users understand they need to call initialize_pipeline_status()
raise PipelineNotInitializedError(namespace) raise PipelineNotInitializedError(namespace)