fix tests

Author: dishaprakash

docs/vector_store.ipynb

@@ -653,7 +653,7 @@
     "await custom_store.aadd_texts(all_texts, metadatas=metadatas, ids=ids)\n",
     "\n",
     "# Use filter on search\n",
-    "docs = await custom_store.asimilarity_search(query, filter=\"len >= 6\")\n",
+    "docs = await custom_store.asimilarity_search(query, filter={\"len\": {\"$gte\": 6}})\n",
     "\n",
     "print(docs)"
    ]
@@ -774,7 +774,7 @@
    "source": [
     "import uuid\n",
     "\n",
-    "docs = await custom_store.asimilarity_search(query, filter=\"price_usd > 100\")\n",
+    "docs = await custom_store.asimilarity_search(query, filter={\"price_usd\": {\"$gte\": 100}})\n",
     "\n",
     "print(docs)"
    ]

tests/test_async_vectorstore_search.py

@@ -195,7 +195,7 @@ async def test_asimilarity_search_scann(self, vs_custom_scann_query_option):
         assert len(results) == 1
         assert results == [Document(page_content="foo", id=ids[0])]
         results = await vs_custom_scann_query_option.asimilarity_search(
-            "foo", k=1, filter="mycontent = 'bar'"
+            "foo", k=1, filter={'mycontent': 'bar'}
         )
         assert results == [Document(page_content="bar", id=ids[1])]
 

tests/test_vectorstore.py

@@ -222,19 +222,19 @@ async def test_aadd_images_store_uri_only(self, engine_sync, image_uris):
         assert len(results) == len(image_uris)
         for i, result_row in enumerate(results):
             assert (
-                result_row[vs._AlloyDBVectorStore__vs.content_column] == image_uris[i]
+                result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
             )
             uri_embedding = embeddings_service.embed_query(image_uris[i])
             image_embedding = image_embedding_service.embed_image([image_uris[i]])[0]
             actual_embedding = json.loads(
-                result_row[vs._AlloyDBVectorStore__vs.embedding_column]
+                result_row[vs._PGVectorStore__vs.embedding_column]
             )
             assert actual_embedding != pytest.approx(uri_embedding)
             assert actual_embedding == pytest.approx(image_embedding)
             assert result_row["image_id"] == str(i)
             assert result_row["source"] == "google.com"
             assert (
-                result_row[vs._AlloyDBVectorStore__vs.metadata_json_column]["image_uri"]
+                result_row[vs._PGVectorStore__vs.metadata_json_column]["image_uri"]
                 == image_uris[i]
             )
         await aexecute(engine_sync, f'DROP TABLE IF EXISTS "{table_name}"')
@@ -279,19 +279,19 @@ async def test_add_images_store_uri_only(self, engine_sync, image_uris):
         assert len(results) == len(image_uris)
         for i, result_row in enumerate(results):
             assert (
-                result_row[vs._AlloyDBVectorStore__vs.content_column] == image_uris[i]
+                result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
             )
             uri_embedding = embeddings_service.embed_query(image_uris[i])
             image_embedding = image_embedding_service.embed_image([image_uris[i]])[0]
             actual_embedding = json.loads(
-                result_row[vs._AlloyDBVectorStore__vs.embedding_column]
+                result_row[vs._PGVectorStore__vs.embedding_column]
             )
             assert actual_embedding != pytest.approx(uri_embedding)
             assert actual_embedding == pytest.approx(image_embedding)
             assert result_row["image_id"] == str(i)
             assert result_row["source"] == "google.com"
             assert (
-                result_row[vs._AlloyDBVectorStore__vs.metadata_json_column]["image_uri"]
+                result_row[vs._PGVectorStore__vs.metadata_json_column]["image_uri"]
                 == image_uris[i]
             )
         await vs.adelete(ids)