yangdx commited on
Commit
09b84d6
·
1 Parent(s): 3f54db9

Use COALESCE to handle null degree counts in Neo4J queries.

Browse files

- Ensures degree count defaults to 0
- Prevents null-related errors in queries

Files changed (1) hide show
  1. lightrag/kg/neo4j_impl.py +2 -2
lightrag/kg/neo4j_impl.py CHANGED
@@ -666,7 +666,7 @@ class Neo4JStorage(BaseGraphStorage):
666
  main_query = """
667
  MATCH (n)
668
  OPTIONAL MATCH (n)-[r]-()
669
- WITH n, count(r) AS degree
670
  WHERE degree >= $min_degree
671
  ORDER BY degree DESC
672
  LIMIT $max_nodes
@@ -703,7 +703,7 @@ class Neo4JStorage(BaseGraphStorage):
703
  WITH start, nodes, relationships
704
  UNWIND nodes AS node
705
  OPTIONAL MATCH (node)-[r]-()
706
- WITH node, count(r) AS degree, start, nodes, relationships
707
  WHERE node = start OR EXISTS((start)--(node)) OR degree >= $min_degree
708
  ORDER BY
709
  CASE
 
666
  main_query = """
667
  MATCH (n)
668
  OPTIONAL MATCH (n)-[r]-()
669
+ WITH n, COALESCE(count(r), 0) AS degree
670
  WHERE degree >= $min_degree
671
  ORDER BY degree DESC
672
  LIMIT $max_nodes
 
703
  WITH start, nodes, relationships
704
  UNWIND nodes AS node
705
  OPTIONAL MATCH (node)-[r]-()
706
+ WITH node, COALESCE(count(r), 0) AS degree, start, nodes, relationships
707
  WHERE node = start OR EXISTS((start)--(node)) OR degree >= $min_degree
708
  ORDER BY
709
  CASE