zrguo commited on
Commit
4eec805
·
unverified ·
2 Parent(s): f2caa47 036b520

Merge pull request #1053 from HKUDS/dev3

Browse files
Files changed (1) hide show
  1. lightrag/lightrag.py +6 -0
lightrag/lightrag.py CHANGED
@@ -2576,6 +2576,12 @@ class LightRAG:
2576
 
2577
  # 9. Delete source entities
2578
  for entity_name in source_entities:
 
 
 
 
 
 
2579
  # Delete entity node from knowledge graph
2580
  await self.chunk_entity_relation_graph.delete_node(entity_name)
2581
 
 
2576
 
2577
  # 9. Delete source entities
2578
  for entity_name in source_entities:
2579
+ if entity_name == target_entity:
2580
+ logger.info(
2581
+ f"Skipping deletion of '{entity_name}' as it's also the target entity"
2582
+ )
2583
+ continue
2584
+
2585
  # Delete entity node from knowledge graph
2586
  await self.chunk_entity_relation_graph.delete_node(entity_name)
2587