feat: Update a test to set explain_analyze param

googleapis · Sep 23, 2024 · f5e7ac3 · f5e7ac3
1 parent 85b703d
commit f5e7ac3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_async_vectorstore_search.py b/tests/test_async_vectorstore_search.py
@@ -295,7 +295,9 @@ async def test_similarity_search_by_vector(self, vs_custom):
         results = await vs_custom.asimilarity_search_by_vector(embedding)
         assert len(results) == 4
         assert results[0] == Document(page_content="foo")
-        results = await vs_custom.asimilarity_search_with_score_by_vector(embedding)
+        results = await vs_custom.asimilarity_search_with_score_by_vector(
+            embedding, explain_analyze=True
+        )
         assert results[0][0] == Document(page_content="foo")
         assert results[0][1] == 0