yangdx
commited on
Commit
·
1fcec3e
1
Parent(s):
8fe10bd
Fix linting
Browse files- lightrag/operate.py +6 -2
lightrag/operate.py
CHANGED
@@ -615,10 +615,14 @@ async def extract_entities(
|
|
615 |
|
616 |
# Merge results - only add entities and edges with new names
|
617 |
for entity_name, entities in glean_nodes.items():
|
618 |
-
if
|
|
|
|
|
619 |
maybe_nodes[entity_name].extend(entities)
|
620 |
for edge_key, edges in glean_edges.items():
|
621 |
-
if
|
|
|
|
|
622 |
maybe_edges[edge_key].extend(edges)
|
623 |
|
624 |
if now_glean_index == entity_extract_max_gleaning - 1:
|
|
|
615 |
|
616 |
# Merge results - only add entities and edges with new names
|
617 |
for entity_name, entities in glean_nodes.items():
|
618 |
+
if (
|
619 |
+
entity_name not in maybe_nodes
|
620 |
+
): # Only accetp entities with new name in gleaning stage
|
621 |
maybe_nodes[entity_name].extend(entities)
|
622 |
for edge_key, edges in glean_edges.items():
|
623 |
+
if (
|
624 |
+
edge_key not in maybe_edges
|
625 |
+
): # Only accetp edges with new name in gleaning stage
|
626 |
maybe_edges[edge_key].extend(edges)
|
627 |
|
628 |
if now_glean_index == entity_extract_max_gleaning - 1:
|