Moved schema to components/

examples/pipeline/kg_builder_with_schema.py

@@ -18,7 +18,7 @@
 import logging
 from typing import Any
 
-from neo4j_genai.kg_construction.schema import (
+from neo4j_genai.components.schema import (
     SchemaBuilder,
     SchemaEntity,
     SchemaRelation,

src/neo4j_genai/components/__init__.py

@@ -0,0 +1,14 @@
+#  Copyright (c) "Neo4j"
+#  Neo4j Sweden AB [https://neo4j.com]
+#  #
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#  #
+#      https://www.apache.org/licenses/LICENSE-2.0
+#  #
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.

src/neo4j_genai/components/kg_writer.py

@@ -18,11 +18,12 @@
 from typing import Literal, Optional
 
 import neo4j
+from pydantic import validate_call
+
 from neo4j_genai.components.types import Neo4jGraph, Neo4jNode, Neo4jRelationship
 from neo4j_genai.indexes import upsert_vector, upsert_vector_on_relationship
 from neo4j_genai.neo4j_queries import UPSERT_NODE_QUERY, UPSERT_RELATIONSHIP_QUERY
 from neo4j_genai.pipeline.component import Component, DataModel
-from pydantic import validate_call
 
 
 class KGWriterModel(DataModel):

src/neo4j_genai/kg_construction/schema.py → src/neo4j_genai/components/schema.py

@@ -30,9 +30,10 @@
 
 from typing import Any, Dict, List, Literal, Tuple
 
+from pydantic import BaseModel, ValidationError, model_validator, validate_call
+
 from neo4j_genai.exceptions import SchemaValidationError
 from neo4j_genai.pipeline import Component, DataModel
-from pydantic import BaseModel, ValidationError, model_validator, validate_call
 
 
 class SchemaProperty(BaseModel):

src/neo4j_genai/components/text_splitters/langchain.py

@@ -15,6 +15,7 @@
 from __future__ import annotations
 
 from langchain_text_splitters import TextSplitter as LangChainTextSplitter
+
 from neo4j_genai.components.text_splitters.base import TextSplitter
 from neo4j_genai.components.types import TextChunk, TextChunks
 

src/neo4j_genai/components/text_splitters/llamaindex.py

@@ -15,6 +15,7 @@
 from __future__ import annotations
 
 from llama_index.core.node_parser import TextSplitter as LlamaIndexTextSplitter
+
 from neo4j_genai.components.text_splitters.base import TextSplitter
 from neo4j_genai.components.types import TextChunk, TextChunks
 

src/neo4j_genai/components/types.py

@@ -16,9 +16,10 @@
 
 from typing import Any, Optional
 
-from neo4j_genai.pipeline.component import DataModel
 from pydantic import BaseModel, field_validator
 
+from neo4j_genai.pipeline.component import DataModel
+
 
 class TextChunk(BaseModel):
     """A chunk of text split from a document by a text splitter.

tests/unit/components/__init__.py

@@ -0,0 +1,14 @@
+#  Copyright (c) "Neo4j"
+#  Neo4j Sweden AB [https://neo4j.com]
+#  #
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#  #
+#      https://www.apache.org/licenses/LICENSE-2.0
+#  #
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.

tests/unit/kg_construction/test_schema.py → tests/unit/components/test_schema.py

@@ -15,13 +15,13 @@
 from __future__ import annotations
 
 import pytest
-from neo4j_genai.exceptions import SchemaValidationError
-from neo4j_genai.kg_construction.schema import (
+from neo4j_genai.components.schema import (
     SchemaBuilder,
     SchemaEntity,
     SchemaProperty,
     SchemaRelation,
 )
+from neo4j_genai.exceptions import SchemaValidationError
 from pydantic import ValidationError