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.
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():
# Check if already initialized by checking for required fields
@ -1194,7 +1194,9 @@ async def try_initialize_namespace(namespace: str) -> bool:
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
Args:
@ -1212,7 +1214,7 @@ async def get_namespace_data(namespace: str, allow_create: bool = False) -> Dict
async with get_internal_lock():
if namespace not in _shared_dicts:
# 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
# This helps users understand they need to call initialize_pipeline_status()
raise PipelineNotInitializedError(namespace)