diff --git a/libs/community/langchain_community/graphs/age_graph.py b/libs/community/langchain_community/graphs/age_graph.py index 4fba85731ad..434491253c6 100644 --- a/libs/community/langchain_community/graphs/age_graph.py +++ b/libs/community/langchain_community/graphs/age_graph.py @@ -697,8 +697,9 @@ class AGEGraph(GraphStore): # query for inserting nodes node_insert_query = ( """ - MERGE (n:`{label}` {properties}) - """ + MERGE (n:`{label}` {{`id`: "{id}"}}) + SET n = {properties} + """ if not include_source else """ MERGE (n:`{label}` {properties}) @@ -735,6 +736,7 @@ class AGEGraph(GraphStore): query = node_insert_query.format( label=AGEGraph.clean_graph_labels(node.type), properties=self._format_properties(node.properties), + id=node.id, ) self.query(query) diff --git a/libs/community/tests/integration_tests/graphs/test_age_graph.py b/libs/community/tests/integration_tests/graphs/test_age_graph.py index 383ddb62f8d..32c2fab33cb 100644 --- a/libs/community/tests/integration_tests/graphs/test_age_graph.py +++ b/libs/community/tests/integration_tests/graphs/test_age_graph.py @@ -10,7 +10,11 @@ from langchain_community.graphs.graph_document import GraphDocument, Node, Relat test_data = [ GraphDocument( - nodes=[Node(id="foo", type="foo"), Node(id="bar", type="bar")], + nodes=[ + Node(id="foo", type="foo"), + Node(id="bar", type="bar"), + Node(id="foo", type="foo", properties={"property_a": "a"}), + ], relationships=[ Relationship( source=Node(id="foo", type="foo"),