Fix test

Author: dishaprakash

langchain_postgres/vectorstore/async_vectorstore.py

@@ -849,8 +849,6 @@ async def is_valid_index(
     async def aget_by_ids(self, ids: Sequence[str]) -> list[Document]:
         """Get documents by ids."""
 
-        quoted_ids = [f"'{id_val}'" for id_val in ids]
-
         columns = self.metadata_columns + [
             self.id_column,
             self.content_column,
@@ -861,7 +859,7 @@ async def aget_by_ids(self, ids: Sequence[str]) -> list[Document]:
         column_names = ", ".join(f'"{col}"' for col in columns)
 
         placeholders = ", ".join(f":id_{i}" for i in range(len(ids)))
-        param_dict = {f"id_{i}": id for i, id in enumerate(quoted_ids)}
+        param_dict = {f"id_{i}": id for i, id in enumerate(ids)}
 
         query = f'SELECT {column_names} FROM "{self.schema_name}"."{self.table_name}" WHERE "{self.id_column}" IN ({placeholders});'
 

tests/unit_tests/v2/test_async_pg_vectorstore_index.py

@@ -19,7 +19,7 @@
 from tests.utils import VECTORSTORE_CONNECTION_STRING_ASYNCPG as CONNECTION_STRING
 
 DEFAULT_TABLE = "default" + str(uuid.uuid4()).replace("-", "_")
-DEFAULT_INDEX_NAME = DEFAULT_TABLE + DEFAULT_INDEX_NAME_SUFFIX
+DEFAULT_INDEX_NAME = DEFAULT_TABLE + "index"
 VECTOR_SIZE = 768
 
 embeddings_service = DeterministicFakeEmbedding(size=VECTOR_SIZE)