gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: get_knowledge_graph return null when query an isolated node

Browse Source
This commit is contained in:
yangdx
2025-04-16 11:14:48 +08:00
parent ccd67cdff7
commit a3de6b75a8
1 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lightrag/kg/postgres_impl.py
View File

@@ -1720,8 +1720,8 @@ class PGGraphStorage(BaseGraphStorage):
strip_label = node_label.strip('"')
count_query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (n:base {{entity_id: "{strip_label}"}})
OPTIONAL MATCH p = (n:base)-[*..{max_depth}]->(m:base)
RETURN count(distinct m) AS total_nodes
OPTIONAL MATCH p = (n)-[*..{max_depth}]-()
RETURN count(nodes(p)) AS total_nodes
$$) AS (total_nodes bigint)"""
count_result = await self._query(count_query)
@@ -1731,19 +1731,25 @@ class PGGraphStorage(BaseGraphStorage):
# Now get the actual data with limit
if node_label == "*":
query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (n:base)
OPTIONAL MATCH (n:base)-[r]->(target:base)
RETURN collect(distinct n) AS n, collect(distinct r) AS r
MATCH (node:base)
OPTIONAL MATCH (node)-[r]->()
RETURN collect(distinct node) AS n, collect(distinct r) AS r
LIMIT {max_nodes}
$$) AS (n agtype, r agtype)"""
else:
strip_label = node_label.strip('"')
query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (n:base {{entity_id: "{strip_label}"}})
OPTIONAL MATCH p = (n:base)-[*..{max_depth}]->(m:base)
RETURN nodes(p) AS n, relationships(p) AS r
LIMIT {max_nodes}
$$) AS (n agtype, r agtype)"""
if total_nodes > 0:
query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (node:base {{entity_id: "{strip_label}"}})
OPTIONAL MATCH p = (node)-[*..{max_depth}]-()
RETURN nodes(p) AS n, relationships(p) AS r
LIMIT {max_nodes}
$$) AS (n agtype, r agtype)"""
else:
query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (node:base {{entity_id: "{strip_label}"}})
RETURN node AS n
$$) AS (n agtype)"""
results = await self._query(query)