Lint

Author: afoucret

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/inference/InferenceOperator.java

@@ -56,7 +56,10 @@ public Page getOutput() {
 
     @Override
     protected void performAsync(Page input, ActionListener<Page> listener) {
-        try (BulkInferenceRequestIterator requests = requests(input); BulkInferenceOutputBuilder<IR, Page> outputBuilder = outputBuilder(input)) {
+        try (
+            BulkInferenceRequestIterator requests = requests(input);
+            BulkInferenceOutputBuilder<IR, Page> outputBuilder = outputBuilder(input)
+        ) {
             bulkInferenceExecutor.execute(requests, outputBuilder, listener);
         } catch (Exception e) {
             listener.onFailure(e);

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/inference/bulk/BulkInferenceExecutor.java

@@ -22,23 +22,28 @@
 
 import static org.elasticsearch.common.logging.LoggerMessageFormat.format;
 
-public class BulkInferenceExecutor <IR extends InferenceServiceResults, OutputType>{
+public class BulkInferenceExecutor<IR extends InferenceServiceResults, OutputType> {
     private static final String TASK_RUNNER_NAME = "bulk_inference_operation";
     private final ThrottledInferenceRunner throttledInferenceRunner;
 
     public BulkInferenceExecutor(InferenceRunner inferenceRunner, ThreadPool threadPool, BulkInferenceExecutionConfig bulkExecutionConfig) {
         throttledInferenceRunner = ThrottledInferenceRunner.create(inferenceRunner, threadPool, bulkExecutionConfig);
     }
 
-    public void execute(BulkInferenceRequestIterator requests, BulkInferenceOutputBuilder<IR, OutputType> outputBuilder, ActionListener<OutputType> listener) {
+    public void execute(
+        BulkInferenceRequestIterator requests,
+        BulkInferenceOutputBuilder<IR, OutputType> outputBuilder,
+        ActionListener<OutputType> listener
+    ) {
         try {
             listener.onResponse(doExecute(requests, outputBuilder));
         } catch (Exception e) {
             listener.onFailure(e);
         }
     }
 
-    public OutputType doExecute(BulkInferenceRequestIterator requests, BulkInferenceOutputBuilder<IR, OutputType> outputBuilder) throws Exception {
+    public OutputType doExecute(BulkInferenceRequestIterator requests, BulkInferenceOutputBuilder<IR, OutputType> outputBuilder)
+        throws Exception {
         final BulkInferenceExecutionState bulkExecutionState = new BulkInferenceExecutionState();
 
         if (requests.hasNext()) {
@@ -57,8 +62,8 @@ private Consumer<InferenceAction.Response> inferenceResultPersister(BulkInferenc
         return inferenceResponse -> {
             InferenceServiceResults results = inferenceResponse.getResults();
             if (outputBuilder.inferenceResultsClass().isInstance(results)) {
-                 outputBuilder.addInferenceResults(outputBuilder.inferenceResultsClass().cast(results));
-                 return;
+                outputBuilder.addInferenceResults(outputBuilder.inferenceResultsClass().cast(results));
+                return;
             }
 
             throw new IllegalStateException(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/inference/completion/CompletionOperatorOutputBuilder.java

@@ -13,7 +13,6 @@
 import org.elasticsearch.compute.data.Page;
 import org.elasticsearch.core.Releasables;
 import org.elasticsearch.xpack.core.inference.results.ChatCompletionResults;
-import org.elasticsearch.xpack.esql.inference.InferenceOperator;
 import org.elasticsearch.xpack.esql.inference.bulk.BulkInferenceOutputBuilder;
 
 import java.util.concurrent.atomic.AtomicBoolean;

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/inference/bulk/BulkInferenceExecutorTests.java

@@ -89,11 +89,14 @@ public void testSuccessfulExecution() throws Exception {
         assertBusy(() -> {
             verify(listener).onResponse(any());
             verify(listener, never()).onFailure(any());
-            assertThat(output.get(), allOf(
-                notNullValue(),
-                hasSize(requests.size()),
-                contains(responses.stream().map(InferenceAction.Response::getResults).toArray())
-            ));
+            assertThat(
+                output.get(),
+                allOf(
+                    notNullValue(),
+                    hasSize(requests.size()),
+                    contains(responses.stream().map(InferenceAction.Response::getResults).toArray())
+                )
+            );
         });
     }