reformat fixes

Author: dishaprakash

langchain_postgres/v2/vectorstores.py

@@ -850,7 +850,7 @@ async def aadd_images(
     ) -> list[str]:
         """Embed images and add to the table."""
         return await self._engine._run_as_async(
-            self._PGVectorStore__vs.aadd_images(uris, metadatas, ids, **kwargs)  # type: ignore
+            self.__vs.aadd_images(uris, metadatas, ids, **kwargs)  # type: ignore
         )
 
     def add_images(
@@ -862,7 +862,7 @@ def add_images(
     ) -> list[str]:
         """Embed images and add to the table."""
         return self._engine._run_as_sync(
-            self._PGVectorStore__vs.aadd_images(uris, metadatas, ids, **kwargs)  # type: ignore
+            self.__vs.aadd_images(uris, metadatas, ids, **kwargs)  # type: ignore
         )
 
     def similarity_search_image(
@@ -874,7 +874,7 @@ def similarity_search_image(
     ) -> list[Document]:
         """Return docs selected by similarity search on image."""
         return self._engine._run_as_sync(
-            self._PGVectorStore__vs.asimilarity_search_image(
+            self.__vs.asimilarity_search_image(
                 image_uri, k, filter, **kwargs
             )  # type: ignore
         )
@@ -888,7 +888,7 @@ async def asimilarity_search_image(
     ) -> list[Document]:
         """Return docs selected by similarity search on image_uri."""
         return await self._engine._run_as_async(
-            self._PGVectorStore__vs.asimilarity_search_image(
+            self.__vs.asimilarity_search_image(
                 image_uri, k, filter, **kwargs
             )  # type: ignore
         )

tests/unit_tests/v2/test_pg_vectorstore.py

@@ -225,7 +225,7 @@ async def test_aadd_docs(self, engine: PGEngine, vs: PGVectorStore) -> None:
         await aexecute(engine, f'TRUNCATE TABLE "{DEFAULT_TABLE}"')
 
     async def test_aadd_images(
-        self, engine_sync: PGVectorStore, image_uris: list[str]
+        self, engine_sync: PGEngine, image_uris: list[str]
     ) -> None:
         engine_sync.init_vectorstore_table(
             IMAGE_TABLE,
@@ -254,7 +254,7 @@ async def test_aadd_images(
         await aexecute(engine_sync, f'TRUNCATE TABLE "{IMAGE_TABLE}"')
 
     async def test_add_images(
-        self, engine_sync: PGVectorStore, image_uris: list[str]
+        self, engine_sync: PGEngine, image_uris: list[str]
     ) -> None:
         engine_sync.init_vectorstore_table(IMAGE_TABLE_SYNC, VECTOR_SIZE)
         vs = PGVectorStore.create_sync(

tests/unit_tests/v2/test_pg_vectorstore_search.py

@@ -220,7 +220,7 @@ async def test_asimilarity_search_by_vector(self, vs: PGVectorStore) -> None:
         assert result[0][0] == Document(page_content="foo", id=ids[0])
         assert result[0][1] == 0
 
-    async def test_asimilarity_search_image(self, image_vs, image_uris):
+    async def test_asimilarity_search_image(self, image_vs: PGVectorStore, image_uris: list[str]) -> None:
         results = await image_vs.asimilarity_search_image(image_uris[0], k=1)
         assert len(results) == 1
         assert results[0].metadata["image_uri"] == image_uris[0]
@@ -441,7 +441,7 @@ def test_similarity_search_by_vector(self, vs_custom: PGVectorStore) -> None:
         assert result[0][0] == Document(page_content="foo", id=ids[0])
         assert result[0][1] == 0
 
-    def test_similarity_search_image(self, image_vs, image_uris):
+    def test_similarity_search_image(self, image_vs: PGVectorStore, image_uris: list[str]) -> None:
         results = image_vs.similarity_search_image(image_uris[0], k=1)
         assert len(results) == 1
         assert results[0].metadata["image_uri"] == image_uris[0]