yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-26 14:22:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Only merge new entities/edges during gleaning

Browse Source 
- Restrict gleaning to new entity names
- Only add edges with new keys
- Prevent similar decription of the same entity or edge
This commit is contained in:
yangdx 2025-04-10 20:31:52 +08:00
parent 32d5e3d908
commit 339bc99259
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/operate.py
View File

@ -613,11 +613,13 @@ async def extract_entities(
glean_result, chunk_key, file_path
)
# Merge results
# Merge results - only add entities and edges with new names
for entity_name, entities in glean_nodes.items():
maybe_nodes[entity_name].extend(entities)
if entity_name not in maybe_nodes: # Only accetp entities with new name in gleaning stage
maybe_nodes[entity_name].extend(entities)
for edge_key, edges in glean_edges.items():
maybe_edges[edge_key].extend(edges)
if edge_key not in maybe_edges: # Only accetp edges with new name in gleaning stage
maybe_edges[edge_key].extend(edges)
if now_glean_index == entity_extract_max_gleaning - 1:
break