Add Weaviate documentation

Author: willtai

docs/source/api.rst

@@ -42,6 +42,21 @@ HybridCypherRetriever
    :members:
 
 
+
+*******************
+External Retrievers
+*******************
+
+This section includes retrievers that integrate with databases external to Neo4j.
+
+
+WeaviateNeo4jRetriever
+======================
+
+.. autoclass:: neo4j_genai.retrievers.external.weaviate.WeaviateNeo4jRetriever
+   :members:
+
+
 ******
 Errors
 ******

src/neo4j_genai/__init__.py

@@ -16,11 +16,13 @@
 from .retrievers.hybrid import HybridCypherRetriever, HybridRetriever
 from .retrievers.text2cypher import Text2CypherRetriever
 from .retrievers.vector import VectorCypherRetriever, VectorRetriever
+from .retrievers.external.weaviate import WeaviateNeo4jRetriever
 
 __all__ = [
     "VectorRetriever",
     "VectorCypherRetriever",
     "HybridRetriever",
     "HybridCypherRetriever",
     "Text2CypherRetriever",
+    "WeaviateNeo4jRetriever",
 ]

src/neo4j_genai/retrievers/external/weaviate/weaviate.py

@@ -94,14 +94,37 @@ def search(
         If no embedder is provided, then it will assume that the vectorizer is used in Weaviate.
 
         See the following documentation for more details:
-        - [Query a vector index](https://neo4j.com/docs/cypher-manual/current/indexes-for-vector-search/#indexes-vector-query)
-        - [db.index.vector.queryNodes()](https://neo4j.com/docs/operations-manual/5/reference/procedures/#procedure_db_index_vector_queryNodes)
-        - [db.index.fulltext.queryNodes()](https://neo4j.com/docs/operations-manual/5/reference/procedures/#procedure_db_index_fulltext_querynodes)
+        - `Query a vector index <https://neo4j.com/docs/cypher-manual/current/indexes-for-vector-search/#indexes-vector-query>`_
+        - `db.index.vector.queryNodes() <https://neo4j.com/docs/operations-manual/5/reference/procedures/#procedure_db_index_vector_queryNodes>`_
+        - `db.index.fulltext.queryNodes() <https://neo4j.com/docs/operations-manual/5/reference/procedures/#procedure_db_index_fulltext_querynodes>`_
+
+
+        Example:
+
+        .. code-block:: python
+
+          import neo4j
+          from neo4j_genai.retrievers import WeaviateNeo4jRetriever
+
+          driver = neo4j.GraphDatabase.driver(URI, auth=AUTH)
+
+          retriever = WeaviateNeo4jRetriever(
+              driver=driver,
+              client=weaviate_client,
+              collection="Jeopardy",
+              id_property_external="neo4j_id",
+              id_property_neo4j="id",
+          )
+
+          biology_embedding = ...
+          retriever.search(query_vector=biology_embedding, top_k=2)
+
+
         Args:
-            query_text (str): The text to get the closest neighbors of.
-            query_vector (Optional[list[float]], optional): The vector embeddings to get the closest neighbors of. Defaults to None.
-            top_k (int, optional): The number of neighbors to return. Defaults to 5.
-            weaviate_filters (Optional[_Filters], optional): The filters to apply to the search query in Weaviate. Defaults to None.
+            query_text (Optional[str]): The text to get the closest neighbors of.
+            query_vector (Optional[list[float]]): The vector embeddings to get the closest neighbors of. Defaults to None.
+            top_k (int): The number of neighbors to return. Defaults to 5.
+            weaviate_filters (Optional[_Filters]): The filters to apply to the search query in Weaviate. Defaults to None.
         Raises:
             SearchValidationError: If validation of the input arguments fail.
         Returns:

tests/e2e/weaviate_e2e/test_weaviate_e2e.py

@@ -16,7 +16,7 @@
 import pytest
 from neo4j import Record
 import weaviate
-from neo4j_genai.retrievers.external.weaviate import WeaviateNeo4jRetriever
+from neo4j_genai import WeaviateNeo4jRetriever
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from .utils import EMBEDDING_BIOLOGY
 from .populate_dbs import populate_dbs