mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-11-18 02:53:46 +00:00
fix: Implement PipelineNotInitializedError usage in get_namespace_data
- Add PipelineNotInitializedError import to shared_storage.py - Raise PipelineNotInitializedError when accessing uninitialized pipeline_status namespace - This provides clear error messages to users about initialization requirements - Other namespaces continue to be created dynamically as before Addresses review feedback from PR #1978 about unused exception class
This commit is contained in:
parent
3a64b267cb
commit
c66fc3483a
@ -8,6 +8,8 @@ import time
|
|||||||
import logging
|
import logging
|
||||||
from typing import Any, Dict, List, Optional, Union, TypeVar, Generic
|
from typing import Any, Dict, List, Optional, Union, TypeVar, Generic
|
||||||
|
|
||||||
|
from lightrag.exceptions import PipelineNotInitializedError
|
||||||
|
|
||||||
|
|
||||||
# Define a direct print function for critical logs that must be visible in all processes
|
# Define a direct print function for critical logs that must be visible in all processes
|
||||||
def direct_log(message, enable_output: bool = False, level: str = "DEBUG"):
|
def direct_log(message, enable_output: bool = False, level: str = "DEBUG"):
|
||||||
@ -1203,6 +1205,13 @@ async def get_namespace_data(namespace: str) -> Dict[str, Any]:
|
|||||||
|
|
||||||
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
|
||||||
|
if namespace == "pipeline_status":
|
||||||
|
# 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)
|
||||||
|
|
||||||
|
# For other namespaces, create them dynamically as before
|
||||||
if _is_multiprocess and _manager is not None:
|
if _is_multiprocess and _manager is not None:
|
||||||
_shared_dicts[namespace] = _manager.dict()
|
_shared_dicts[namespace] = _manager.dict()
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user