From 29bf59366317333923634420452801c96bbfd68d Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 25 Oct 2025 22:32:27 +0800 Subject: [PATCH] Fix entity and relation chunk cleanup in deletion pipeline MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit • Delete from entity_chunks storage • Delete from relation_chunks storage --- lightrag/lightrag.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 7af96237..ca972712 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -3302,6 +3302,10 @@ class LightRAG: list(entities_to_delete) ) + # Delete from entity_chunks storage + if self.entity_chunks: + await self.entity_chunks.delete(list(entities_to_delete)) + async with pipeline_status_lock: log_message = f"Successfully deleted {len(entities_to_delete)} entities" logger.info(log_message) @@ -3331,6 +3335,14 @@ class LightRAG: list(relationships_to_delete) ) + # Delete from relation_chunks storage + if self.relation_chunks: + relation_storage_keys = [ + make_relation_chunk_key(src, tgt) + for src, tgt in relationships_to_delete + ] + await self.relation_chunks.delete(relation_storage_keys) + async with pipeline_status_lock: log_message = f"Successfully deleted {len(relationships_to_delete)} relations" logger.info(log_message)