diff --git a/lightrag/operate.py b/lightrag/operate.py index 2a64a791..5d67b21b 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -615,10 +615,14 @@ async def extract_entities( # Merge results - only add entities and edges with new names for entity_name, entities in glean_nodes.items(): - if entity_name not in maybe_nodes: # Only accetp entities with new name in gleaning stage + 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(): - if edge_key not in maybe_edges: # Only accetp edges with new name in gleaning stage + 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: