Merge pull request #1053 from HKUDS/dev3
Browse files- 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 |
|