yaml test fix

Author: tteofili

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/240_knn_search_early_termination.yml

@@ -15,7 +15,7 @@ setup:
                 type: dense_vector
                 dims: 64
                 index: true
-                similarity: dot_product
+                similarity: max_inner_product
                 index_options:
                   type: bbq_hnsw
                   early_termination: true
@@ -88,36 +88,37 @@ setup:
                     -0.224,  0.203, -0.439,  0.064,  0.246, -0.396,  0.297,  0.242,
                     -0.028,  0.321, -0.022, -0.009, -0.001  ,  0.031, -0.533,  0.45,
                     -0.683,  1.331,  0.194, -0.157, -0.1  , -0.279, -0.098, -0.789 ]
-  # Flush in order to provoke a merge later
   - do:
       indices.flush:
         index: bbq_hnsw
 
+  - do:
+      indices.refresh: {}
+
 ---
 "Run knn search with early termination":
 
   - do:
       search:
         index: bbq_hnsw
         body:
-          profile: true
-          knn:
-            field: vector
-            query_vector: [0.128,  0.067, -0.08 ,  0.395, -0.11 , -0.259,  0.473, -0.393,
-                           0.292,  0.571, -0.491,  0.444, -0.288,  0.198, -0.343,  0.015,
-                           0.232,  0.088,  0.228,  0.151, -0.136,  0.236, -0.273, -0.259,
-                           -0.217,  0.359, -0.207,  0.352, -0.142,  0.192, -0.061, -0.17 ,
-                           -0.343,  0.189, -0.221,  0.32 , -0.301, -0.1  ,  0.005,  0.232,
-                           -0.344,  0.136,  0.252,  0.157, -0.13 , -0.244,  0.193, -0.034,
-                           -0.12 , -0.193, -0.102,  0.252, -0.185, -0.167, -0.575,  0.582,
-                           -0.426,  0.983,  0.212,  0.204,  0.03 , -0.276, -0.425, -0.158]
-            k: 3
-            num_candidates: 3
+          query:
+            knn:
+              field: vector
+              query_vector: [0.128,  0.067, -0.08 ,  0.395, -0.11 , -0.259,  0.473, -0.393,
+                             0.292,  0.571, -0.491,  0.444, -0.288,  0.198, -0.343,  0.015,
+                             0.232,  0.088,  0.228,  0.151, -0.136,  0.236, -0.273, -0.259,
+                             -0.217,  0.359, -0.207,  0.352, -0.142,  0.192, -0.061, -0.17 ,
+                             -0.343,  0.189, -0.221,  0.32 , -0.301, -0.1  ,  0.005,  0.232,
+                             -0.344,  0.136,  0.252,  0.157, -0.13 , -0.244,  0.193, -0.034,
+                             -0.12 , -0.193, -0.102,  0.252, -0.185, -0.167, -0.575,  0.582,
+                             -0.426,  0.983,  0.212,  0.204,  0.03 , -0.276, -0.425, -0.158]
+              num_candidates: 5
+              k: 3
 
-  - match: { hits.total.value: 5 } # collector sees k docs
-  - length: { hits.hits: 3 } # size docs retrieved
-  - match: { hits.hits.0._id: "1" }
-  - match: { hits.hits.1._id: "2" }
+  - match: { hits.total.value: 3 }
+  - match: { hits.hits.0._id: "5" }
+  - match: { hits.hits.1._id: "1" }
   - match: { hits.hits.2._id: "3" }
 
 ---
@@ -128,20 +129,21 @@ setup:
         index: bbq_hnsw
         body:
           profile: true
-          knn:
-            field: vector
-            query_vector: [0.128,  0.067, -0.08 ,  0.395, -0.11 , -0.259,  0.473, -0.393,
-                           0.292,  0.571, -0.491,  0.444, -0.288,  0.198, -0.343,  0.015,
-                           0.232,  0.088,  0.228,  0.151, -0.136,  0.236, -0.273, -0.259,
-                           -0.217,  0.359, -0.207,  0.352, -0.142,  0.192, -0.061, -0.17 ,
-                           -0.343,  0.189, -0.221,  0.32 , -0.301, -0.1  ,  0.005,  0.232,
-                           -0.344,  0.136,  0.252,  0.157, -0.13 , -0.244,  0.193, -0.034,
-                           -0.12 , -0.193, -0.102,  0.252, -0.185, -0.167, -0.575,  0.582,
-                           -0.426,  0.983,  0.212,  0.204,  0.03 , -0.276, -0.425, -0.158]
-            k: 3
-            num_candidates: 3
+          query:
+            knn:
+              field: vector
+              query_vector: [0.128,  0.067, -0.08 ,  0.395, -0.11 , -0.259,  0.473, -0.393,
+                             0.292,  0.571, -0.491,  0.444, -0.288,  0.198, -0.343,  0.015,
+                             0.232,  0.088,  0.228,  0.151, -0.136,  0.236, -0.273, -0.259,
+                             -0.217,  0.359, -0.207,  0.352, -0.142,  0.192, -0.061, -0.17 ,
+                             -0.343,  0.189, -0.221,  0.32 , -0.301, -0.1  ,  0.005,  0.232,
+                             -0.344,  0.136,  0.252,  0.157, -0.13 , -0.244,  0.193, -0.034,
+                             -0.12 , -0.193, -0.102,  0.252, -0.185, -0.167, -0.575,  0.582,
+                             -0.426,  0.983,  0.212,  0.204,  0.03 , -0.276, -0.425, -0.158]
+              k: 3
+              num_candidates: 3
 
-  - match: { profile.shards.0.dfs.knn.0.vector_operations_count: 6 }
+  - match: { profile.shards.0.dfs.knn.0.vector_operations_count: null }
 
   # Search with similarity to check number of operations are propagated correctly
   - do:
@@ -163,4 +165,4 @@ setup:
             num_candidates: 3
             similarity: 100000
 
-  - match: { profile.shards.0.dfs.knn.0.vector_operations_count: 6 }
+  - match: { profile.shards.0.dfs.knn.0.vector_operations_count: 5 }