Skip to content

Commit f7ca762

Browse files
Cleanup cypher deprecated function (#323)
1 parent 2d62e4c commit f7ca762

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/neo4j_graphrag/experimental/components/resolver.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ async def run(self) -> ResolutionStats:
221221
UNWIND labels(entity) AS lab
222222
WITH lab, entity
223223
WHERE NOT lab IN ['__Entity__', '__KGBuilder__']
224-
WITH lab, collect({{ id: id(entity), {props_map} }}) AS labelCluster
224+
WITH lab, collect({{ id: elementId(entity), {props_map} }}) AS labelCluster
225225
RETURN lab, labelCluster
226226
"""
227227

@@ -260,10 +260,10 @@ async def run(self) -> ResolutionStats:
260260
for node_id_set in merged_sets:
261261
if len(node_id_set) > 1:
262262
merge_query = (
263-
"MATCH (n) WHERE id(n) IN $ids "
263+
"MATCH (n) WHERE elementId(n) IN $ids "
264264
"WITH collect(n) AS nodes "
265265
"CALL apoc.refactor.mergeNodes(nodes, {properties: 'discard', mergeRels: true}) "
266-
"YIELD node RETURN id(node)"
266+
"YIELD node RETURN elementId(node)"
267267
)
268268
result, _, _ = self.driver.execute_query(
269269
merge_query,

0 commit comments

Comments
 (0)