Browse Source

Fix setOnce in EmbeddingRequestChunker (#114900) (#114908)

Jan Kuipers 1 year ago
parent
commit
553dee9b7c

+ 1 - 1
x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java

@@ -324,7 +324,7 @@ public class EmbeddingRequestChunker {
         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) {