Apply changes requested by the PR review

Author: NathalieCharbel

docs/source/user_guide_kg_builder.rst

@@ -125,8 +125,8 @@ This schema information can be provided to the `SimpleKGBuilder` as demonstrated
         # ...
     )
 
-Prompt Template, Lexical Graph Config, Schema Enforcement, and Error Behavior
---------------------------------------------------------
+Extra configurations
+--------------------
 
 These parameters are part of the `EntityAndRelationExtractor` component.
 For detailed information, refer to the section on :ref:`Entity and Relation Extractor`.
@@ -138,7 +138,7 @@ They are also accessible via the `SimpleKGPipeline` interface.
         # ...
         prompt_template="",
         lexical_graph_config=my_config,
-        enforce_schema=SchemaEnforcementMode.Strict
+        enforce_schema="STRICT"
         on_error="RAISE",
         # ...
     )

src/neo4j_graphrag/experimental/components/entity_relation_extractor.py

@@ -409,7 +409,7 @@ def _enforce_nodes(
             schema_entity = schema.entities.get(node.label)
             if not schema_entity:
                 continue
-            allowed_props = schema_entity.get("properties", {})
+            allowed_props = schema_entity.get("properties", [])
             filtered_props = self._enforce_properties(node.properties, allowed_props)
             if filtered_props:
                 valid_nodes.append(
@@ -461,7 +461,7 @@ def _enforce_relationships(
                     (start_label, rel.type, end_label) not in potential_schema):
                 continue
 
-            allowed_props = schema_relation.get("properties", {})
+            allowed_props = schema_relation.get("properties", [])
             filtered_props = self._enforce_properties(rel.properties, allowed_props)
 
             valid_rels.append(

src/neo4j_graphrag/experimental/components/types.py

@@ -174,5 +174,5 @@ class GraphResult(DataModel):
 
 
 class SchemaEnforcementMode(str, Enum):
-    NONE = "none"
-    STRICT = "strict"
+    NONE = "NONE"
+    STRICT = "STRICT"