Cleaning + update example

Author: stellasia

examples/customize/build_graph/components/pruners/graph_pruner.py

@@ -41,7 +41,7 @@
         Neo4jNode(
             id="Organization/Corp1",
             label="Organization",
-            properties={"name": "CorpA"},
+            properties={"name": "Corp1"},
         ),
     ],
     relationships=[
@@ -51,7 +51,7 @@
             type="KNOWS",
         ),
         Neo4jRelationship(
-            start_node_id="Organization/CorpA",
+            start_node_id="Organization/Corp2",
             end_node_id="Person/Jack",
             type="WORKS_FOR",
         ),
@@ -80,12 +80,14 @@
                 PropertyType(name="name", type="STRING", required=True),
                 PropertyType(name="address", type="STRING"),
             ],
+            additional_properties=True,
         ),
     ),
     relationship_types=(
         RelationshipType(
             label="WORKS_FOR",
             properties=[PropertyType(name="since", type="LOCAL_DATETIME")],
+            additional_properties=True,
         ),
         RelationshipType(
             label="KNOWS",

src/neo4j_graphrag/experimental/components/schema.py

@@ -106,7 +106,8 @@ def validate_additional_properties(self) -> Self:
         if len(self.properties) == 0 and not self.additional_properties:
             raise ValueError(
                 "Using `additional_properties=False` with no defined "
-                "properties will cause the model to be pruned during graph cleaning.",
+                "properties will cause the model to be pruned during graph cleaning. "
+                f"Define some properties or remove this NodeType: {self}"
             )
         return self
 
@@ -135,7 +136,8 @@ def validate_additional_properties(self) -> Self:
         if len(self.properties) == 0 and not self.additional_properties:
             raise ValueError(
                 "Using `additional_properties=False` with no defined "
-                "properties will cause the model to be pruned during graph cleaning.",
+                "properties will cause the model to be pruned during graph cleaning. "
+                f"Define some properties or remove this RelationshipType: {self}"
             )
         return self
 
@@ -218,20 +220,6 @@ def validate_additional_parameters(self) -> Self:
             )
         return self
 
-    @classmethod
-    def default_additional_node_types(cls, validated_data: dict[str, Any]) -> bool:
-        return len(validated_data["node_types"]) == 0
-
-    @classmethod
-    def default_additional_relationship_types(
-        cls, validated_data: dict[str, Any]
-    ) -> bool:
-        return len(validated_data["relationship_types"]) == 0
-
-    @classmethod
-    def default_additional_patterns(cls, validated_data: dict[str, Any]) -> bool:
-        return len(validated_data["patterns"]) == 0
-
     def node_type_from_label(self, label: str) -> Optional[NodeType]:
         return self._node_type_index.get(label)