From 44daf515016d30824179ef07b68bf0ad758c3e88 Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 27 Jun 2025 01:37:10 +0800 Subject: [PATCH] Fix edge_degree to handle non-existent nodes --- lightrag/kg/networkx_impl.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py index dc161359..bb7233b4 100644 --- a/lightrag/kg/networkx_impl.py +++ b/lightrag/kg/networkx_impl.py @@ -106,7 +106,9 @@ class NetworkXStorage(BaseGraphStorage): async def edge_degree(self, src_id: str, tgt_id: str) -> int: graph = await self._get_graph() - return graph.degree(src_id) + graph.degree(tgt_id) + src_degree = graph.degree(src_id) if graph.has_node(src_id) else 0 + tgt_degree = graph.degree(tgt_id) if graph.has_node(tgt_id) else 0 + return src_degree + tgt_degree async def get_edge( self, source_node_id: str, target_node_id: str