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

Add null check for edge data

Browse Source
This commit is contained in:
yangdx
2025-04-21 18:32:33 +08:00
parent bfce14d41f
commit ff65cba544
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lightrag/operate.py
View File

@@ -400,6 +400,16 @@ async def _merge_edges_then_upsert(
for need_insert_id in [src_id, tgt_id]: for need_insert_id in [src_id, tgt_id]:
if not (await knowledge_graph_inst.has_node(need_insert_id)): if not (await knowledge_graph_inst.has_node(need_insert_id)):
# # Discard this edge if the node does not exist
# if need_insert_id == src_id:
# logger.warning(
# f"Discard edge: {src_id} - {tgt_id} | Source node missing"
# )
# else:
# logger.warning(
# f"Discard edge: {src_id} - {tgt_id} | Target node missing"
# )
# return None
await knowledge_graph_inst.upsert_node( await knowledge_graph_inst.upsert_node(
need_insert_id, need_insert_id,
node_data={ node_data={
@@ -738,7 +748,8 @@ async def extract_entities(
pipeline_status_lock, pipeline_status_lock,
llm_response_cache, llm_response_cache,
) )
relationships_data.append(edge_data) if edge_data is not None:
relationships_data.append(edge_data)
# Update vector databases with all collected data # Update vector databases with all collected data
if entity_vdb is not None and entities_data: if entity_vdb is not None and entities_data: