Bug fix

alexthomas93 · Jan 30, 2025 · cd27513 · cd27513
1 parent 240d946
commit cd27513
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/neo4j_graphrag/schema.py b/src/neo4j_graphrag/schema.py
@@ -727,7 +727,9 @@ def enhance_properties(
     excluded = EXCLUDED_RELS if is_relationship else EXCLUDED_LABELS
     if name in excluded:
         return
-    props = structured_schema["node_props"].get(name)
+    props = structured_schema["rel_props" if is_relationship else "node_props"].get(
+        name
+    )
     if not props:  # The node has no properties
         return
     enhanced_cypher = get_enhanced_schema_cypher(