x

Author: eyurtsev

langchain_postgres/vectorstores.py

@@ -847,7 +847,11 @@ async def aadd_texts(
         await self.__apost_init__()  # Lazy async init
         embeddings = await self.embedding_function.aembed_documents(list(texts))
         return await self.aadd_embeddings(
-            texts=texts, embeddings=embeddings, metadatas=metadatas, ids=ids, **kwargs
+            texts=list(texts),
+            embeddings=embeddings,
+            metadatas=metadatas,
+            ids=ids,
+            **kwargs,
         )
 
     def similarity_search(
@@ -2253,7 +2257,7 @@ async def aget_by_ids(self, ids: Sequence[str], /) -> List[Document]:
                 .filter(*filter_by)
             )
 
-            results: List[Any] = (await session.execute(stmt)).scalars().all()
+            results: Sequence[Any] = (await session.execute(stmt)).scalars().all()
 
             for result in results:
                 documents.append(

tests/unit_tests/test_vectorstore_standard_tests.py

@@ -5,26 +5,26 @@
     ReadWriteTestSuite,
 )
 
-from tests.unit_tests.test_vectorstore import aget_vectorstore, get_vectorstore
+from tests.unit_tests.test_vectorstore import get_vectorstore, aget_vectorstore
 
 
 class TestSync(ReadWriteTestSuite):
     @pytest.fixture()
     def vectorstore(self) -> VectorStore:
         """Get an empty vectorstore for unit tests."""
-        with get_vectorstore(embedding=self.get_embeddings()) as vectorstore:
-            vectorstore.drop_tables()
-            vectorstore.create_tables_if_not_exists()
-            vectorstore.create_collection()
-            yield vectorstore
+        with get_vectorstore(embedding=self.get_embeddings()) as vstore:
+            vstore.drop_tables()
+            vstore.create_tables_if_not_exists()
+            vstore.create_collection()
+            yield vstore
 
 
 class TestAsync(AsyncReadWriteTestSuite):
     @pytest.fixture()
     async def vectorstore(self) -> VectorStore:
         """Get an empty vectorstore for unit tests."""
-        async with aget_vectorstore(embedding=self.get_embeddings()) as vectorstore:
-            await vectorstore.adrop_tables()
-            await vectorstore.acreate_tables_if_not_exists()
-            await vectorstore.acreate_collection()
-            yield vectorstore
+        async with aget_vectorstore(embedding=self.get_embeddings()) as vstore:
+            await vstore.adrop_tables()
+            await vstore.acreate_tables_if_not_exists()
+            await vstore.acreate_collection()
+            yield vstore