From f512dabc87c9e7a9e2cfb0ccd4c50bd87a1f0247 Mon Sep 17 00:00:00 2001 From: nathaliecharbel Date: Fri, 4 Apr 2025 16:20:55 +0200 Subject: [PATCH] Use elementId() instead of id() in cypher queries --- src/neo4j_graphrag/experimental/components/resolver.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/neo4j_graphrag/experimental/components/resolver.py b/src/neo4j_graphrag/experimental/components/resolver.py index 1dce5302..ecff8cd0 100644 --- a/src/neo4j_graphrag/experimental/components/resolver.py +++ b/src/neo4j_graphrag/experimental/components/resolver.py @@ -221,7 +221,7 @@ async def run(self) -> ResolutionStats: UNWIND labels(entity) AS lab WITH lab, entity WHERE NOT lab IN ['__Entity__', '__KGBuilder__'] - WITH lab, collect({{ id: id(entity), {props_map} }}) AS labelCluster + WITH lab, collect({{ id: elementId(entity), {props_map} }}) AS labelCluster RETURN lab, labelCluster """ @@ -260,10 +260,10 @@ async def run(self) -> ResolutionStats: for node_id_set in merged_sets: if len(node_id_set) > 1: merge_query = ( - "MATCH (n) WHERE id(n) IN $ids " + "MATCH (n) WHERE elementId(n) IN $ids " "WITH collect(n) AS nodes " "CALL apoc.refactor.mergeNodes(nodes, {properties: 'discard', mergeRels: true}) " - "YIELD node RETURN id(node)" + "YIELD node RETURN elementId(node)" ) result, _, _ = self.driver.execute_query( merge_query,