fix test_search

weni-ai · Jun 4, 2024 · 1d7a0e0 · 1d7a0e0
1 parent 2373b85
commit 1d7a0e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/tests/test_elasticsearch_vector_store.py b/app/tests/test_elasticsearch_vector_store.py
@@ -126,7 +126,7 @@ def test_search(self):
             search="test", filter=query_filter
         )
         self.vectorstore.similarity_search_with_score.assert_called_once_with(
-            query="test", k=5, filter=query_filter
+            query="test", k=5, filter={'match': {'metadata.content_base_uuid': 'dfff32e7-dce6-40f7-a86e-8f9618887977'}}
         )
         self.assertEqual(1, len(results))
         self.assertEqual(results[0].page_content, "test doc")