Refactor shared storage module to improve async handling and naming consistency

• Add async support for get_namespace_data
• Rename get_update_flags to get_update_flag
• Rename set_update_flag to set_all_update_flags
• Update docstrings for clarity
• Fix typos in log messages
This commit is contained in:
yangdx
2025-03-01 05:01:26 +08:00
parent fd76e00c6a
commit d704512139
3 changed files with 11 additions and 13 deletions

View File

@@ -338,7 +338,7 @@ async def extract_entities(
) -> None:
from lightrag.kg.shared_storage import get_namespace_data
pipeline_status = get_namespace_data("pipeline_status")
pipeline_status = await get_namespace_data("pipeline_status")
use_llm_func: callable = global_config["llm_model_func"]
entity_extract_max_gleaning = global_config["entity_extract_max_gleaning"]
enable_llm_cache_for_entity_extract: bool = global_config[