From cbacb615f35a3ecde5f2406b72a00993cdfdbcf4 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Tue, 11 Mar 2025 18:33:26 +0800
Subject: [PATCH] Use COALESCE to handle null degree counts in Neo4J queries.

- Ensures degree count defaults to 0
- Prevents null-related errors in queries
---
 lightrag/kg/neo4j_impl.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py
index ef77cd06..d2d3459e 100644
--- a/lightrag/kg/neo4j_impl.py
+++ b/lightrag/kg/neo4j_impl.py
@@ -666,7 +666,7 @@ class Neo4JStorage(BaseGraphStorage):
                     main_query = """
                     MATCH (n)
                     OPTIONAL MATCH (n)-[r]-()
-                    WITH n, count(r) AS degree
+                    WITH n, COALESCE(count(r), 0) AS degree
                     WHERE degree >= $min_degree
                     ORDER BY degree DESC
                     LIMIT $max_nodes
@@ -703,7 +703,7 @@ class Neo4JStorage(BaseGraphStorage):
                     WITH start, nodes, relationships
                     UNWIND nodes AS node
                     OPTIONAL MATCH (node)-[r]-()
-                    WITH node, count(r) AS degree, start, nodes, relationships
+                    WITH node, COALESCE(count(r), 0) AS degree, start, nodes, relationships
                     WHERE node = start OR EXISTS((start)--(node)) OR degree >= $min_degree
                     ORDER BY
                         CASE