fix tests

Author: dishaprakash

src/langchain_google_alloydb_pg/vectorstore.py

@@ -20,16 +20,16 @@
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VectorStore
-
-from .async_vectorstore import AsyncAlloyDBVectorStore
-from .engine import AlloyDBEngine
 from langchain_postgres.v2.indexes import (
     DEFAULT_DISTANCE_STRATEGY,
     BaseIndex,
     DistanceStrategy,
     QueryOptions,
 )
 
+from .async_vectorstore import AsyncAlloyDBVectorStore
+from .engine import AlloyDBEngine
+
 
 class AlloyDBVectorStore(VectorStore):
     """Google AlloyDB Vector Store class"""

tests/test_async_vectorstore_index.py

@@ -21,16 +21,16 @@
 import pytest_asyncio
 from langchain_core.documents import Document
 from langchain_core.embeddings import DeterministicFakeEmbedding
-from sqlalchemy import text
-
-from langchain_google_alloydb_pg import AlloyDBEngine
-from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
-from langchain_google_alloydb_pg.indexes import (
+from langchain_postgres.v2.indexes import (
     DEFAULT_INDEX_NAME_SUFFIX,
     DistanceStrategy,
     HNSWIndex,
     IVFFlatIndex,
 )
+from sqlalchemy import text
+
+from langchain_google_alloydb_pg import AlloyDBEngine
+from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
 
 DEFAULT_TABLE = "test_table" + str(uuid.uuid4()).replace("-", "_")
 DEFAULT_INDEX_NAME = DEFAULT_TABLE + DEFAULT_INDEX_NAME_SUFFIX

tests/test_async_vectorstore_search.py

@@ -19,15 +19,17 @@
 import pytest_asyncio
 from langchain_core.documents import Document
 from langchain_core.embeddings import DeterministicFakeEmbedding
+from langchain_postgres.v2.indexes import (
+    DistanceStrategy,
+    HNSWQueryOptions,
+)
 from metadata_filtering_data import FILTERING_TEST_CASES, METADATAS
 from PIL import Image
 from sqlalchemy import text
 
 from langchain_google_alloydb_pg import AlloyDBEngine, Column
 from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
 from langchain_google_alloydb_pg.indexes import (
-    DistanceStrategy,
-    HNSWQueryOptions,
     ScaNNQueryOptions,
 )
 

tests/test_vectorstore_index.py

@@ -22,14 +22,16 @@
 import sqlalchemy
 from langchain_core.documents import Document
 from langchain_core.embeddings import DeterministicFakeEmbedding
-from sqlalchemy import text
-
-from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore
-from langchain_google_alloydb_pg.indexes import (
+from langchain_postgres.v2.indexes import (
     DEFAULT_INDEX_NAME_SUFFIX,
     DistanceStrategy,
     HNSWIndex,
     IVFFlatIndex,
+)
+from sqlalchemy import text
+
+from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore
+from langchain_google_alloydb_pg.indexes import (
     IVFIndex,
     ScaNNIndex,
 )