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

Fix linting

Browse source
This commit is contained in:
yangdx 2025-04-28 02:15:25 +08:00
parent d7552a2a87
commit 594e7b751a
3 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
lightrag/operate.py
View file

@ -503,6 +503,7 @@ async def merge_nodes_and_edges(
"""
# Get lock manager from shared storage
from .kg.shared_storage import get_graph_db_lock
graph_db_lock = get_graph_db_lock(enable_logging=False)
# Collect all nodes and edges from all chunks
@ -527,7 +528,9 @@ async def merge_nodes_and_edges(
# Use graph database lock to ensure atomic merges and updates
async with graph_db_lock:
async with pipeline_status_lock:
log_message = f"Merging nodes/edges {current_file_number}/{total_files}: {file_path}"
log_message = (
f"Merging nodes/edges {current_file_number}/{total_files}: {file_path}"
)
logger.info(log_message)
pipeline_status["latest_message"] = log_message
pipeline_status["history_messages"].append(log_message)
@ -585,9 +588,7 @@ async def merge_nodes_and_edges(
}
await entity_vdb.upsert(data_for_vdb)
log_message = (
f"Updating {total_relations_count} relations {current_file_number}/{total_files}: {file_path}"
)
log_message = f"Updating {total_relations_count} relations {current_file_number}/{total_files}: {file_path}"
logger.info(log_message)
if pipeline_status is not None:
async with pipeline_status_lock: