Small bug fix

Author: alexthomas93

src/neo4j_graphrag/neo4j_queries.py

@@ -110,15 +110,15 @@
     "MATCH (n) "
     "WHERE elementId(n) = $node_element_id "
     "WITH n "
-    "CALL db.create.setNodeVectorProperty(n, $embedding_node_property, $vector) "
+    "CALL db.create.setNodeVectorProperty(n, $embedding_property, $vector) "
     "RETURN n"
 )
 
 UPSERT_VECTOR_ON_RELATIONSHIP_QUERY = (
     "MATCH ()-[r]->() "
     "WHERE elementId(r) = $rel_element_id "
     "WITH r "
-    "CALL db.create.setRelationshipVectorProperty(r, $embedding_node_property, $vector) "
+    "CALL db.create.setRelationshipVectorProperty(r, $embedding_property, $vector) "
     "RETURN r"
 )