From 2440920ef1a111cb0247b8b1bf7b51b9a6ffbaeb Mon Sep 17 00:00:00 2001
From: Jan Kuipers <jan.kuipers@elastic.co>
Date: Wed, 16 Oct 2024 14:09:36 +0200
Subject: [PATCH] Fix setOnce in EmbeddingRequestChunker

---
 .../xpack/inference/chunking/EmbeddingRequestChunker.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java
index 3ae8dc0550391..c5897f32d6eb8 100644
--- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java
+++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java
@@ -324,7 +324,7 @@ private ElasticsearchStatusException unexpectedResultTypeException(String got, S
         public void onFailure(Exception e) {
             var errorResult = new ErrorChunkedInferenceResults(e);
             for (var pos : positions) {
-                errors.setOnce(pos.inputIndex(), errorResult);
+                errors.set(pos.inputIndex(), errorResult);
             }
 
             if (resultCount.incrementAndGet() == totalNumberOfRequests) {