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

Fix node_degree error for PostgreSQL graph storage

Browse Source
This commit is contained in:
yangdx
2025-04-04 03:40:17 +08:00
parent 689cf44a5d
commit 0947feff2d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/kg/postgres_impl.py
View File

@@ -1232,13 +1232,15 @@ class PGGraphStorage(BaseGraphStorage):
label = node_id.strip('"') label = node_id.strip('"')
query = """SELECT * FROM cypher('%s', $$ query = """SELECT * FROM cypher('%s', $$
MATCH (n:base {entity_id: "%s"})-[]->(x) MATCH (n:base {entity_id: "%s"})-[]-(x)
RETURN count(x) AS total_edge_count RETURN count(x) AS total_edge_count
$$) AS (total_edge_count integer)""" % (self.graph_name, label) $$) AS (total_edge_count integer)""" % (self.graph_name, label)
record = (await self._query(query))[0] record = (await self._query(query))[0]
if record: if record:
edge_count = int(record["total_edge_count"]) edge_count = int(record["total_edge_count"])
print(f"edge_count: {edge_count} / {node_id}")
return edge_count return edge_count
async def edge_degree(self, src_id: str, tgt_id: str) -> int: async def edge_degree(self, src_id: str, tgt_id: str) -> int: