gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add entity vector database upsert for newly added entities by edges upserts

Browse source
This commit is contained in:
yangdx 2025-09-14 05:04:45 +08:00
parent 7cc2b69bcf
commit 4f5ad76c2c
1 changed files with 29 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
lightrag/operate.py
View file

@ -1813,6 +1813,35 @@ async def merge_nodes_and_edges(
retry_delay=0.1, retry_delay=0.1,
) )
# Update added_entities to entity vector database using safe operation wrapper
if added_entities and entity_vdb is not None:
for entity_data in added_entities:
entity_vdb_id = compute_mdhash_id(
entity_data["entity_name"], prefix="ent-"
)
entity_content = f"{entity_data['entity_name']}\n{entity_data['description']}"
vdb_data = {
entity_vdb_id: {
"content": entity_content,
"entity_name": entity_data["entity_name"],
"source_id": entity_data["source_id"],
"entity_type": entity_data["entity_type"],
"file_path": entity_data.get(
"file_path", "unknown_source"
),
}
}
# Use safe operation wrapper - VDB failure must throw exception
await safe_vdb_operation_with_exception(
operation=lambda data=vdb_data: entity_vdb.upsert(data),
operation_name="added_entity_upsert",
entity_name=entity_data["entity_name"],
max_retries=3,
retry_delay=0.1,
)
return edge_data, added_entities return edge_data, added_entities
except Exception as e: except Exception as e: