Merge branch 'main' into replace_tuple_with_record

Author: idegtiarenko

build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/DependenciesInfoPlugin.java

@@ -15,9 +15,13 @@
 import org.gradle.api.Project;
 import org.gradle.api.artifacts.Configuration;
 import org.gradle.api.attributes.Category;
+import org.gradle.api.attributes.Usage;
 import org.gradle.api.plugins.JavaPlugin;
 
 public class DependenciesInfoPlugin implements Plugin<Project> {
+
+    public static String USAGE_ATTRIBUTE = "DependenciesInfo";
+
     @Override
     public void apply(final Project project) {
         project.getPlugins().apply(CompileOnlyResolvePlugin.class);
@@ -43,6 +47,9 @@ public void apply(final Project project) {
             )
         );
 
+        dependenciesInfoFilesConfiguration.attributes(
+            attributes -> attributes.attribute(Usage.USAGE_ATTRIBUTE, project.getObjects().named(Usage.class, USAGE_ATTRIBUTE))
+        );
         project.getArtifacts().add("dependenciesInfoFiles", depsInfo);
 
     }

distribution/build.gradle

@@ -31,6 +31,9 @@ configurations {
     attributes {
       attribute(Category.CATEGORY_ATTRIBUTE, project.getObjects().named(Category.class, Category.DOCUMENTATION))
     }
+    attributes {
+      attribute(Usage.USAGE_ATTRIBUTE, project.getObjects().named(Usage.class, DependenciesInfoPlugin.USAGE_ATTRIBUTE))
+    }
   }
   featuresMetadata {
     attributes {

docs/changelog/129181.yaml

@@ -0,0 +1,5 @@
+pr: 129181
+summary: Add Cluster Feature for L2 Norm
+area: "Search"
+type: bug
+issues: []

docs/reference/query-languages/esql/_snippets/functions/description/std_dev.md

@@ -5,8 +5,6 @@ tests:
 - class: org.elasticsearch.smoketest.WatcherYamlRestIT
   method: test {p0=watcher/usage/10_basic/Test watcher usage stats output}
   issue: https://github.com/elastic/elasticsearch/issues/112189
-- class: org.elasticsearch.ingest.geoip.IngestGeoIpClientYamlTestSuiteIT
-  issue: https://github.com/elastic/elasticsearch/issues/111497
 - class: org.elasticsearch.packaging.test.PackagesSecurityAutoConfigurationTests
   method: test20SecurityNotAutoConfiguredOnReInstallation
   issue: https://github.com/elastic/elasticsearch/issues/112635
@@ -519,6 +517,12 @@ tests:
 - class: org.elasticsearch.index.engine.ThreadPoolMergeExecutorServiceDiskSpaceTests
   method: testUnavailableBudgetBlocksNewMergeTasksFromStartingExecution
   issue: https://github.com/elastic/elasticsearch/issues/129148
+- class: org.elasticsearch.xpack.esql.qa.single_node.GenerativeForkIT
+  method: test {lookup-join.EnrichLookupStatsBug ASYNC}
+  issue: https://github.com/elastic/elasticsearch/issues/129228
+- class: org.elasticsearch.xpack.esql.qa.single_node.GenerativeForkIT
+  method: test {lookup-join.EnrichLookupStatsBug SYNC}
+  issue: https://github.com/elastic/elasticsearch/issues/129229
 
 # Examples:
 #

docs/reference/query-languages/esql/_snippets/functions/examples/std_dev.md

@@ -276,12 +276,11 @@ TopDocs doVectorQuery(byte[] vector, IndexSearcher searcher) throws IOException
     TopDocs doVectorQuery(float[] vector, IndexSearcher searcher) throws IOException {
         Query knnQuery;
         int topK = this.topK;
-        int efSearch = this.efSearch;
         if (overSamplingFactor > 1f) {
             // oversample the topK results to get more candidates for the final result
             topK = (int) Math.ceil(topK * overSamplingFactor);
-            efSearch = Math.max(topK, efSearch);
         }
+        int efSearch = Math.max(topK, this.efSearch);
         if (indexType == KnnIndexTester.IndexType.IVF) {
             knnQuery = new IVFKnnFloatVectorQuery(VECTOR_FIELD, vector, topK, efSearch, null, nProbe);
         } else {

docs/reference/query-languages/esql/kibana/definition/functions/std_dev.json

@@ -0,0 +1,46 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+package org.elasticsearch.index.codec.vectors;
+
+import org.apache.lucene.internal.hppc.IntArrayList;
+
+final class CentroidAssignments {
+
+    private final int numCentroids;
+    private final float[][] cachedCentroids;
+    private final IntArrayList[] assignmentsByCluster;
+
+    private CentroidAssignments(int numCentroids, float[][] cachedCentroids, IntArrayList[] assignmentsByCluster) {
+        this.numCentroids = numCentroids;
+        this.cachedCentroids = cachedCentroids;
+        this.assignmentsByCluster = assignmentsByCluster;
+    }
+
+    CentroidAssignments(float[][] centroids, IntArrayList[] assignmentsByCluster) {
+        this(centroids.length, centroids, assignmentsByCluster);
+    }
+
+    CentroidAssignments(int numCentroids, IntArrayList[] assignmentsByCluster) {
+        this(numCentroids, null, assignmentsByCluster);
+    }
+
+    // Getters and setters
+    public int numCentroids() {
+        return numCentroids;
+    }
+
+    public float[][] cachedCentroids() {
+        return cachedCentroids;
+    }
+
+    public IntArrayList[] assignmentsByCluster() {
+        return assignmentsByCluster;
+    }
+}