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

Improve thread safety for relationship rebuilding

Browse source 
- Sort src and tgt for consistent lock keys
- Maintain order-independent locking
This commit is contained in:
yangdx 2025-07-19 10:25:48 +08:00
parent 05bc5cfb64
commit 6e1657a771
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/operate.py
View file

@ -429,8 +429,12 @@ async def _rebuild_knowledge_from_chunks(
async with semaphore: async with semaphore:
workspace = global_config.get("workspace", "") workspace = global_config.get("workspace", "")
namespace = f"{workspace}:GraphDB" if workspace else "GraphDB" namespace = f"{workspace}:GraphDB" if workspace else "GraphDB"
# Sort src and tgt to ensure order-independent lock key generation
sorted_key_parts = sorted([src, tgt])
async with get_storage_keyed_lock( async with get_storage_keyed_lock(
f"{src}-{tgt}", namespace=namespace, enable_logging=False f"{sorted_key_parts[0]}-{sorted_key_parts[1]}",
namespace=namespace,
enable_logging=False,
): ):
try: try:
await _rebuild_single_relationship( await _rebuild_single_relationship(