Removed sentence_transformers from init

Author: willtai

docs/source/user_guide.rst

@@ -266,7 +266,7 @@ The `OpenAIEmbedder` was illustrated previously. Here is how to use the `Sentenc
 
 .. code:: python
 
-    from neo4j_genai.embeddings import SentenceTransformerEmbeddings
+    from neo4j_genai.embeddings.sentence_transformers import SentenceTransformerEmbeddings
 
     embedder = SentenceTransformerEmbeddings(model="all-MiniLM-L6-v2")  # Note: this is the default model
 

examples/graphrag.py

@@ -19,7 +19,7 @@
 URI = "neo4j://localhost:7687"
 AUTH = ("neo4j", "password")
 DATABASE = "neo4j"
-INDEX = "vector-index-name"
+INDEX = "moviePlotsEmbedding"
 
 
 # setup logger config

src/neo4j_genai/embeddings/__init__.py

@@ -1,5 +1,14 @@
-from .sentence_transformers import SentenceTransformerEmbeddings
-
-__all__ = [
-    "SentenceTransformerEmbeddings",
-]
+#  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/embeddings/openai.py

@@ -1,3 +1,18 @@
+#  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.
+
 from __future__ import annotations
 
 from typing import Any

src/neo4j_genai/embeddings/sentence_transformers.py

@@ -1,3 +1,18 @@
+#  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.
+
 from typing import Any
 
 import numpy as np

tests/unit/embeddings/test_sentence_transformers.py

@@ -3,7 +3,7 @@
 import numpy as np
 import pytest
 from neo4j_genai.embedder import Embedder
-from neo4j_genai.embeddings import SentenceTransformerEmbeddings
+from neo4j_genai.embeddings.sentence_transformers import SentenceTransformerEmbeddings
 
 
 @patch("sentence_transformers.SentenceTransformer")