fix(lightrag): 过滤掉引用不存在节点的边以确保查询结果的有效性 #477

Author: xerrors

src/knowledge/adapters/lightrag.py

@@ -317,4 +317,12 @@ def _process_query_result(self, result, limit: int = None) -> dict[str, list]:
                                 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}