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

Fix get_knowledge_graph query slow problem for PostgreSQL AGE

Browse Source
This commit is contained in:
yangdx
2025-04-24 08:40:06 +08:00
parent e19d5fa9be
commit ffbe2a0c5d
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lightrag/kg/postgres_impl.py
View File

@@ -1849,9 +1849,8 @@ class PGGraphStorage(BaseGraphStorage):
else: else:
strip_label = self._normalize_node_id(node_label) strip_label = self._normalize_node_id(node_label)
count_query = f"""SELECT * FROM cypher('{self.graph_name}', $$ count_query = f"""SELECT * FROM cypher('{self.graph_name}', $$
MATCH (n:base {{entity_id: "{strip_label}"}}) MATCH (n:base {{entity_id: "{strip_label}"}})-[r]-()
OPTIONAL MATCH p = (n)-[*..{max_depth}]-() RETURN count(r) AS total_nodes
RETURN count(nodes(p)) AS total_nodes
$$) AS (total_nodes bigint)""" $$) AS (total_nodes bigint)"""
count_result = await self._query(count_query) count_result = await self._query(count_query)