yangdx commited on
Commit
1fcec3e
·
1 Parent(s): 8fe10bd

Fix linting

Browse files
Files changed (1) hide show
  1. 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 entity_name not in maybe_nodes: # Only accetp entities with new name in gleaning stage
 
 
619
  maybe_nodes[entity_name].extend(entities)
620
  for edge_key, edges in glean_edges.items():
621
- if edge_key not in maybe_edges: # Only accetp edges with new name in gleaning stage
 
 
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: