Fix pytests (#2157)

tests/vector_stores/test_elasticsearch.py

@@ -208,15 +208,19 @@ class TestElasticsearchDB(unittest.TestCase):
         # Verify search parameters
         self.assertEqual(search_args["index"], "test_collection")
         body = search_args["body"]
-        self.assertIn("script_score", body["query"])
-        self.assertEqual(
-            body["query"]["script_score"]["script"]["params"]["query_vector"],
-            query_vector
-        )
+        
+        # Verify KNN query structure
+        self.assertIn("query", body)
+        self.assertIn("bool", body["query"])
+        self.assertIn("must", body["query"]["bool"])
+        
+        # Verify KNN parameters
+        knn_query = body["query"]["bool"]["must"][-1]["knn"]["vector"]
+        self.assertEqual(knn_query["vector"], query_vector)
+        self.assertEqual(knn_query["k"], 5)
 
         # Verify results
         self.assertEqual(len(results), 1)
-        self.assertIsInstance(results[0], OutputData)
         self.assertEqual(results[0].id, "id1")
         self.assertEqual(results[0].score, 0.8)
         self.assertEqual(results[0].payload, {"key1": "value1"})