Add capability check to avoid errors on mixed clusters

Author: elasticsearchmachine

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-cosine-similarity.csv-spec

@@ -2,6 +2,7 @@
 
  similarityWithVectorField
  required_capability: cosine_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-cosine-similarity[]
  from colors
@@ -30,6 +31,7 @@ gray           | 0.9082483053207397
 
  similarityAsPartOfExpression
  required_capability: cosine_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
  from colors
  | where color != "black" 
@@ -54,6 +56,7 @@ gray         | 1.454
 
 similarityWithLiteralVectors
 required_capability: cosine_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval similarity = round(v_cosine([1, 2, 3], [0, 1, 2]), 3) 
@@ -66,6 +69,7 @@ similarity:double
 
  similarityWithStats
  required_capability: cosine_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
  from colors
  | where color != "black" 
@@ -80,6 +84,7 @@ avg:double | min:double | max:double
 similarityWithNull
 required_capability: cosine_vector_similarity_function
 required_capability: vector_similarity_functions_support_null
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = v_cosine(rgb_vector, null) 
@@ -93,6 +98,7 @@ total_null:long
 similarityWithRow
 required_capability: cosine_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3]) 
 | eval similarity = round(v_cosine(vector, [0, 1, 2]), 3) 
@@ -105,6 +111,7 @@ vector: dense_vector | similarity:double
 similarityWithVectorField
 required_capability: cosine_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | where color != "black" 

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-dot-product.csv-spec

@@ -2,6 +2,7 @@
 
  similarityWithVectorField
  required_capability: dot_product_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-dot-product[]
  from colors
@@ -29,6 +30,7 @@ old lace   | 60563.0
 
 similarityAsPartOfExpression
 required_capability: dot_product_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval score = round((1 + v_dot_product(rgb_vector, [0, 255, 255]) / 2), 3) 
@@ -52,6 +54,7 @@ old lace   | 30282.5
 
 similarityWithLiteralVectors
 required_capability: dot_product_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval similarity = round(v_dot_product([1, 2, 3], [0, 1, 2]), 3) 
@@ -64,6 +67,7 @@ similarity:double
 
 similarityWithStats
 required_capability: dot_product_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = round(v_dot_product(rgb_vector, [0, 255, 255]), 3) 
@@ -77,6 +81,7 @@ avg:double | min:double | max:double
 similarityWithNull
 required_capability: dot_product_vector_similarity_function
 required_capability: vector_similarity_functions_support_null
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = v_dot_product(rgb_vector, null) 
@@ -91,6 +96,7 @@ total_null:long
 similarityWithRow
 required_capability: dot_product_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3])
 | eval similarity = round(v_dot_product(vector, [0, 1, 2]), 3) 
@@ -103,6 +109,7 @@ vector: dense_vector | similarity:double
 similarityWithVectorField
 required_capability: dot_product_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval query = to_dense_vector([0, 255, 255])

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-hamming.csv-spec

@@ -2,6 +2,7 @@
 
  similarityWithVectorField
  required_capability: hamming_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-hamming[]
  from colors
@@ -29,6 +30,7 @@ firebrick  | 16.0
 
 similarityAsPartOfExpression
 required_capability: hamming_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval score = round((1 + v_hamming(rgb_vector, [0, 255, 255]) / 2), 3) 
@@ -52,6 +54,7 @@ firebrick  | 9.0
 
 similarityWithLiteralVectors
 required_capability: hamming_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval similarity = round(v_hamming([1, 2, 3], [0, 1, 2]), 3) 
@@ -64,6 +67,7 @@ similarity:double
 
 similarityWithStats
 required_capability: hamming_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = round(v_hamming(rgb_vector, [0, 255, 255]), 3) 
@@ -77,6 +81,7 @@ avg:double | min:double | max:double
 similarityWithNull
 required_capability: hamming_vector_similarity_function
 required_capability: vector_similarity_functions_support_null
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = v_hamming(rgb_vector, null) 
@@ -90,6 +95,7 @@ total_null:long
 similarityWithRow
 required_capability: hamming_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3])
 | eval similarity = round(v_hamming(vector, [0, 1, 2]), 3) 
@@ -102,6 +108,7 @@ vector: dense_vector | similarity:double
 similarityWithVectorField
 required_capability: hamming_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval query = to_dense_vector([0, 255, 255])

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-l1-norm.csv-spec

@@ -2,6 +2,7 @@
 
  similarityWithVectorField
  required_capability: l1_norm_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-l1-norm[]
  from colors
@@ -29,6 +30,7 @@ gold       | 550.0
 
 similarityAsPartOfExpression
 required_capability: l1_norm_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval score = round((1 + v_l1_norm(rgb_vector, [0, 255, 255]) / 2), 3) 
@@ -52,6 +54,7 @@ gold       | 276.0
 
 similarityWithLiteralVectors
 required_capability: l1_norm_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval similarity = round(v_l1_norm([1, 2, 3], [0, 1, 2]), 3) 
@@ -64,6 +67,7 @@ similarity:double
 
 similarityWithStats
 required_capability: l1_norm_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = round(v_l1_norm(rgb_vector, [0, 255, 255]), 3) 
@@ -77,6 +81,7 @@ avg:double | min:double | max:double
 similarityWithNull
 required_capability: l1_norm_vector_similarity_function
 required_capability: vector_similarity_functions_support_null
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = v_l1_norm(rgb_vector, null) 
@@ -90,6 +95,7 @@ total_null:long
 similarityWithRow
 required_capability: l1_norm_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3]) 
 | eval similarity = round(v_l1_norm(vector, [0, 1, 2]), 3) 
@@ -102,6 +108,7 @@ vector: dense_vector | similarity:double
 similarityWithVectorField
 required_capability: l1_norm_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval query = to_dense_vector([0, 255, 255])

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-l2-norm.csv-spec

@@ -2,6 +2,7 @@
 
  similarityWithVectorField
  required_capability: l2_norm_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-l2-norm[]
  from colors
@@ -29,6 +30,7 @@ tomato     | 351.0227966308594
 
  similarityAsPartOfExpression
  required_capability: l2_norm_vector_similarity_function
+ required_capability: dense_vector_field_type_released
 
 from colors
 | eval score = round((1 + v_l2_norm(rgb_vector, [0, 255, 255]) / 2), 3) 
@@ -52,6 +54,7 @@ tomato     | 176.511
 
 similarityWithLiteralVectors
 required_capability: l2_norm_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval similarity = round(v_l2_norm([1, 2, 3], [0, 1, 2]), 3) 
@@ -64,6 +67,7 @@ similarity:double
 
 similarityWithStats
 required_capability: l2_norm_vector_similarity_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = round(v_l2_norm(rgb_vector, [0, 255, 255]), 3) 
@@ -77,6 +81,7 @@ avg:double | min:double | max:double
 similarityWithNull
 required_capability: l2_norm_vector_similarity_function
 required_capability: vector_similarity_functions_support_null
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval similarity = v_l2_norm(rgb_vector, null) 
@@ -90,6 +95,7 @@ total_null:long
 similarityWithRow
 required_capability: l2_norm_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3]) 
 | eval similarity = round(v_l2_norm(vector, [0, 1, 2]), 3) 
@@ -102,6 +108,7 @@ vector: dense_vector | similarity:double
 similarityWithVectorField
 required_capability: l2_norm_vector_similarity_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 from colors
 | eval query = to_dense_vector([0, 255, 255])

x-pack/plugin/esql/qa/testFixtures/src/main/resources/vector-magnitude.csv-spec

@@ -2,6 +2,7 @@
 
  magnitudeWithVectorField
  required_capability: magnitude_scalar_vector_function
+ required_capability: dense_vector_field_type_released
 
 // tag::vector-magnitude[]
  from colors
@@ -29,6 +30,7 @@ linen      | 415.93267822265625
 
  magnitudeAsPartOfExpression
  required_capability: magnitude_scalar_vector_function
+ required_capability: dense_vector_field_type_released
 
  from colors
  | eval score = round((1 + v_magnitude(rgb_vector) / 2), 3) 
@@ -52,6 +54,7 @@ linen      | 208.966
 
 magnitudeWithLiteralVectors
 required_capability: magnitude_scalar_vector_function
+required_capability: dense_vector_field_type_released
 
 row a = 1
 | eval magnitude = round(v_magnitude([1, 2, 3]), 3) 
@@ -64,6 +67,7 @@ magnitude:double
 
  magnitudeWithStats
  required_capability: magnitude_scalar_vector_function
+ required_capability: dense_vector_field_type_released
 
  from colors
  | eval magnitude = round(v_magnitude(rgb_vector), 3) 
@@ -76,7 +80,8 @@ avg:double | min:double | max:double
 
 magnitudeWithNull
 required_capability: magnitude_scalar_vector_function
-
+required_capability: dense_vector_field_type_released
+ 
 row a = 1
 | eval magnitude = v_magnitude(null) 
 | keep magnitude
@@ -89,6 +94,7 @@ null
 magnitudeWithRow
 required_capability: magnitude_scalar_vector_function
 required_capability: to_dense_vector_function
+required_capability: dense_vector_field_type_released
 
 row vector = to_dense_vector([1, 2, 3]) 
 | eval magnitude = round(v_magnitude(vector), 3) 

x-pack/plugin/src/yamlRestTest/resources/rest-api-spec/test/esql/230_folding.yml

@@ -5,7 +5,7 @@ setup:
         - method: POST
           path: /_query
           parameters: [ ]
-          capabilities: [ dense_vector_field_type ]
+          capabilities: [ dense_vector_field_type_released ]
       test_runner_features: [ capabilities, contains ]
       reason: "make sure new functions run where supported only, dense vector needed for index creation"
   - do:

x-pack/plugin/src/yamlRestTest/resources/rest-api-spec/test/esql/40_unsupported_types.yml

@@ -145,7 +145,7 @@ unsupported:
         - method: POST
           path: /_query
           parameters: []
-          capabilities: [dense_vector_field_type, dense_vector_agg_metric_double_if_fns]
+          capabilities: [dense_vector_field_type_released, dense_vector_agg_metric_double_if_fns]
       reason: "uses original_type"
 
   - do:
@@ -338,7 +338,7 @@ unsupported with sort:
         - method: POST
           path: /_query
           parameters: [ ]
-          capabilities: [ dense_vector_field_type, dense_vector_agg_metric_double_if_fns ]
+          capabilities: [ dense_vector_field_type_released, dense_vector_agg_metric_double_if_fns ]
       reason: "support for sorting when dense_vector_field_type present"
 
   - do: