From f60bad92f0d72bbf74e8b4ec688ab72008bb6f6f Mon Sep 17 00:00:00 2001 From: yangdx Date: Mon, 23 Jun 2025 19:20:10 +0800 Subject: [PATCH] Fix: Delete relations from vector DB in both directions --- lightrag/utils_graph.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/lightrag/utils_graph.py b/lightrag/utils_graph.py index 54876fa7..ce837dc0 100644 --- a/lightrag/utils_graph.py +++ b/lightrag/utils_graph.py @@ -84,10 +84,12 @@ async def adelete_by_relation( return # Delete relation from vector database - relation_id = compute_mdhash_id( - source_entity + target_entity, prefix="rel-" - ) - await relationships_vdb.delete([relation_id]) + rel_ids_to_delete = [ + compute_mdhash_id(source_entity + target_entity, prefix="rel-"), + compute_mdhash_id(target_entity + source_entity, prefix="rel-"), + ] + + await relationships_vdb.delete(rel_ids_to_delete) # Delete relation from knowledge graph await chunk_entity_relation_graph.remove_edges(