Update embeddings_function -> embeddings (#21)

API update for vectorstore

Author: eyurtsev

langchain_postgres/vectorstores.py

@@ -237,7 +237,7 @@ class PGVector(VectorStore):
 
     def __init__(
         self,
-        embedding_function: Embeddings,
+        embeddings: Embeddings,
         *,
         connection: Optional[Connection] = None,
         embedding_length: Optional[int] = None,
@@ -255,7 +255,7 @@ def __init__(
 
         Args:
             connection: Postgres connection string.
-            embedding_function: Any embedding function implementing
+            embeddings: Any embedding function implementing
                 `langchain.embeddings.base.Embeddings` interface.
             embedding_length: The length of the embedding vector. (default: None)
                 NOTE: This is not mandatory. Defining it will prevent vectors of
@@ -278,7 +278,7 @@ def __init__(
                 doesn't exist. disabling creation is useful when using ReadOnly
                 Databases.
         """
-        self.embedding_function = embedding_function
+        self.embedding_function = embeddings
         self._embedding_length = embedding_length
         self.collection_name = collection_name
         self.collection_metadata = collection_metadata
@@ -441,7 +441,7 @@ def __from(
         store = cls(
             connection=connection,
             collection_name=collection_name,
-            embedding_function=embedding,
+            embeddings=embedding,
             distance_strategy=distance_strategy,
             pre_delete_collection=pre_delete_collection,
             use_jsonb=use_jsonb,
@@ -1068,7 +1068,7 @@ def from_existing_index(
         store = cls(
             connection=connection,
             collection_name=collection_name,
-            embedding_function=embedding,
+            embeddings=embedding,
             distance_strategy=distance_strategy,
             pre_delete_collection=pre_delete_collection,
         )

tests/unit_tests/test_vectorstore.py

@@ -155,7 +155,7 @@ def test_pgvector_collection_with_metadata() -> None:
     pgvector = PGVector(
         collection_name="test_collection",
         collection_metadata={"foo": "bar"},
-        embedding_function=FakeEmbeddingsWithAdaDimension(),
+        embeddings=FakeEmbeddingsWithAdaDimension(),
         connection=CONNECTION_STRING,
         pre_delete_collection=True,
     )