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

Fix edge_degree to handle non-existent nodes

Browse source
This commit is contained in:
yangdx 2025-06-27 01:37:10 +08:00
parent c740401b7f
commit 44daf51501
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lightrag/kg/networkx_impl.py
View file

@ -106,7 +106,9 @@ class NetworkXStorage(BaseGraphStorage):
async def edge_degree(self, src_id: str, tgt_id: str) -> int: async def edge_degree(self, src_id: str, tgt_id: str) -> int:
graph = await self._get_graph() 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( async def get_edge(
self, source_node_id: str, target_node_id: str self, source_node_id: str, target_node_id: str