gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-03-09 01:00:42 +08:00
parent 3cf4268e7a
commit c5d0962872
3 changed files with 120 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/operate.py
View File

@@ -522,8 +522,9 @@ async def extract_entities(
maybe_edges[tuple(sorted(k))].extend(v)
from .kg.shared_storage import get_graph_db_lock
graph_db_lock = get_graph_db_lock(enable_logging = True)
graph_db_lock = get_graph_db_lock(enable_logging=True)
# Ensure that nodes and edges are merged and upserted atomically
async with graph_db_lock:
all_entities_data = await asyncio.gather(
@@ -535,7 +536,9 @@ async def extract_entities(
all_relationships_data = await asyncio.gather(
*[
_merge_edges_then_upsert(k[0], k[1], v, knowledge_graph_inst, global_config)
_merge_edges_then_upsert(
k[0], k[1], v, knowledge_graph_inst, global_config
)
for k, v in maybe_edges.items()
]
)