Added comment about Embedder type in docstring

neo4j · Apr 22, 2024 · 643d2a3 · 643d2a3
1 parent 2a21e0d
commit 643d2a3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/examples/similarity_search_for_text.py b/examples/similarity_search_for_text.py
@@ -15,7 +15,7 @@
 driver = GraphDatabase.driver(URI, auth=AUTH)
 
 
-# Create Embedder object
+# Create CustomEmbedder object with the required Embedder type
 class CustomEmbedder(Embedder):
     def embed_query(self, text: str) -> list[float]:
         return [random() for _ in range(DIMENSION)]

diff --git a/src/neo4j_genai/retrievers.py b/src/neo4j_genai/retrievers.py
@@ -66,7 +66,8 @@ def search(self, *args, **kwargs) -> Any:
 
 class VectorRetriever(Retriever):
     """
-    Provides retrieval method using vector search over embeddings
+    Provides retrieval method using vector search over embeddings.
+    If an embedder is provided, it needs to have the required Embedder type.
     """
 
     def __init__(
@@ -160,7 +161,8 @@ def search(
 
 class VectorCypherRetriever(Retriever):
     """
-    Provides retrieval method using vector similarity and custom Cypher query
+    Provides retrieval method using vector similarity and custom Cypher query.
+    If an embedder is provided, it needs to have the required Embedder type.
     """
 
     def __init__(