gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix lint

Browse Source
This commit is contained in:
zrguo
2025-03-20 22:13:31 +08:00
parent d817defbc4
commit eb1e954ced
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/lightrag.py
View File

@@ -2589,6 +2589,12 @@ class LightRAG:
f"Created or updated relationship: {rel_data['src']} -> {rel_data['tgt']}" f"Created or updated relationship: {rel_data['src']} -> {rel_data['tgt']}"
) )
# Delete relationships records from vector database
await self.relationships_vdb.delete(relations_to_delete)
logger.info(
f"Deleted {len(relations_to_delete)} relation records for entity '{entity_name}' from vector database"
)
# 7. Update entity vector representation # 7. Update entity vector representation
description = merged_entity_data.get("description", "") description = merged_entity_data.get("description", "")
source_id = merged_entity_data.get("source_id", "") source_id = merged_entity_data.get("source_id", "")
@@ -2651,12 +2657,6 @@ class LightRAG:
entity_id = compute_mdhash_id(entity_name, prefix="ent-") entity_id = compute_mdhash_id(entity_name, prefix="ent-")
await self.entities_vdb.delete([entity_id]) await self.entities_vdb.delete([entity_id])
# Delete relationships records from vector database
await self.relationships_vdb.delete(relations_to_delete)
logger.info(
f"Deleted {len(relations_to_delete)} relation records for entity '{entity_name}' from vector database"
)
logger.info( logger.info(
f"Deleted source entity '{entity_name}' and its vector embedding from database" f"Deleted source entity '{entity_name}' and its vector embedding from database"
) )