From 488f67e5b27d8fd6b956fdf9f73822e26c025d1b 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 (cherry picked from commit 29bf59366317333923634420452801c96bbfd68d) --- lightrag/lightrag.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 1f32da50..e4b02519 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -3144,6 +3144,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) @@ -3173,6 +3177,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)