fix(lightrag): 过滤掉引用不存在节点的边以确保查询结果的有效性 #477
This commit is contained in:
parent
f686c2873c
commit
67bcd26565
@ -317,4 +317,12 @@ class LightRAGGraphAdapter(GraphAdapter):
|
||||
nodes.append(self.normalize_node(item))
|
||||
node_ids.add(item.element_id)
|
||||
|
||||
return {"nodes": nodes, "edges": edges}
|
||||
# 过滤掉引用不存在节点的边
|
||||
valid_edges = []
|
||||
for edge in edges:
|
||||
source_id = edge.get("source_id")
|
||||
target_id = edge.get("target_id")
|
||||
if source_id in node_ids and target_id in node_ids:
|
||||
valid_edges.append(edge)
|
||||
|
||||
return {"nodes": nodes, "edges": valid_edges}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user