From 649164c3e692659aeabc966e53e0af415ca15863 Mon Sep 17 00:00:00 2001
From: zrguo <49157727+LarFii@users.noreply.github.com>
Date: Wed, 5 Mar 2025 16:55:09 +0800
Subject: [PATCH 1/2] Update lightrag.py

---
 lightrag/lightrag.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index e8e468af..f81ade0b 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -1945,6 +1945,9 @@ class LightRAG:
                     new_entity_name, new_node_data
                 )
 
+                # Store relationships that need to be updated
+                relations_to_update = []
+
                 # Get all edges related to the original entity
                 edges = await self.chunk_entity_relation_graph.get_node_edges(
                     entity_name
@@ -1960,10 +1963,12 @@ class LightRAG:
                                 await self.chunk_entity_relation_graph.upsert_edge(
                                     new_entity_name, target, edge_data
                                 )
+                                relations_to_update.append((new_entity_name, target, edge_data))
                             else:  # target == entity_name
                                 await self.chunk_entity_relation_graph.upsert_edge(
                                     source, new_entity_name, edge_data
                                 )
+                                relations_to_update.append((source, new_entity_name, edge_data))
 
                 # Delete old entity
                 await self.chunk_entity_relation_graph.delete_node(entity_name)
@@ -1972,6 +1977,35 @@ class LightRAG:
                 old_entity_id = compute_mdhash_id(entity_name, prefix="ent-")
                 await self.entities_vdb.delete([old_entity_id])
 
+                # Update relationship vector representations
+                for src, tgt, edge_data in relations_to_update:
+                    description = edge_data.get("description", "")
+                    keywords = edge_data.get("keywords", "")
+                    source_id = edge_data.get("source_id", "")
+                    weight = float(edge_data.get("weight", 1.0))
+                    
+                    # Create new content for embedding
+                    content = f"{src}\t{tgt}\n{keywords}\n{description}"
+                    
+                    # Calculate relationship ID
+                    relation_id = compute_mdhash_id(src + tgt, prefix="rel-")
+                    
+                    # Prepare data for vector database update
+                    relation_data = {
+                        relation_id: {
+                            "content": content,
+                            "src_id": src,
+                            "tgt_id": tgt,
+                            "source_id": source_id,
+                            "description": description,
+                            "keywords": keywords,
+                            "weight": weight,
+                        }
+                    }
+                    
+                    # Update vector database
+                    await self.relationships_vdb.upsert(relation_data)
+
                 # Update working entity name to new name
                 entity_name = new_entity_name
             else:
@@ -2082,7 +2116,7 @@ class LightRAG:
             weight = float(new_edge_data.get("weight", 1.0))
 
             # Create content for embedding
-            content = f"{keywords}\t{source_entity}\n{target_entity}\n{description}"
+            content = f"{source_entity}\t{target_entity}\n{keywords}\n{description}"
 
             # Calculate relation ID
             relation_id = compute_mdhash_id(

From 917dc39334d99120f98ede2c7cb13c70bdf78e24 Mon Sep 17 00:00:00 2001
From: zrguo <49157727+LarFii@users.noreply.github.com>
Date: Wed, 5 Mar 2025 17:00:01 +0800
Subject: [PATCH 2/2] fix linting

---
 lightrag/lightrag.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index f81ade0b..5c060658 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -1963,12 +1963,16 @@ class LightRAG:
                                 await self.chunk_entity_relation_graph.upsert_edge(
                                     new_entity_name, target, edge_data
                                 )
-                                relations_to_update.append((new_entity_name, target, edge_data))
+                                relations_to_update.append(
+                                    (new_entity_name, target, edge_data)
+                                )
                             else:  # target == entity_name
                                 await self.chunk_entity_relation_graph.upsert_edge(
                                     source, new_entity_name, edge_data
                                 )
-                                relations_to_update.append((source, new_entity_name, edge_data))
+                                relations_to_update.append(
+                                    (source, new_entity_name, edge_data)
+                                )
 
                 # Delete old entity
                 await self.chunk_entity_relation_graph.delete_node(entity_name)
@@ -1983,13 +1987,13 @@ class LightRAG:
                     keywords = edge_data.get("keywords", "")
                     source_id = edge_data.get("source_id", "")
                     weight = float(edge_data.get("weight", 1.0))
-                    
+
                     # Create new content for embedding
                     content = f"{src}\t{tgt}\n{keywords}\n{description}"
-                    
+
                     # Calculate relationship ID
                     relation_id = compute_mdhash_id(src + tgt, prefix="rel-")
-                    
+
                     # Prepare data for vector database update
                     relation_data = {
                         relation_id: {
@@ -2002,7 +2006,7 @@ class LightRAG:
                             "weight": weight,
                         }
                     }
-                    
+
                     # Update vector database
                     await self.relationships_vdb.upsert(relation_data)