Code improvement.

Author: afoucret

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

@@ -28,13 +28,14 @@ public abstract class InferenceOperator extends AsyncOperator<InferenceOperator.
     private static final int MAX_INFERENCE_WORKER = 10;
     private final String inferenceId;
     private final BlockFactory blockFactory;
-
+    private final BulkInferenceExecutionConfig bulkExecutionConfig;
     private final BulkInferenceExecutor bulkInferenceExecutor;
 
-    public InferenceOperator(DriverContext driverContext, InferenceRunner inferenceRunner, ThreadPool threadPool, String inferenceId) {
+    public InferenceOperator(DriverContext driverContext, InferenceRunner inferenceRunner, BulkInferenceExecutionConfig bulkExecutionConfig, ThreadPool threadPool, String inferenceId) {
         super(driverContext, threadPool.getThreadContext(), MAX_INFERENCE_WORKER);
         this.blockFactory = driverContext.blockFactory();
-        this.bulkInferenceExecutor = new BulkInferenceExecutor(inferenceRunner, threadPool, bulkExecutionConfig());
+        this.bulkExecutionConfig = bulkExecutionConfig;
+        this.bulkInferenceExecutor = new BulkInferenceExecutor(inferenceRunner, threadPool, bulkExecutionConfig);
         this.inferenceId = inferenceId;
     }
 
@@ -78,7 +79,7 @@ protected void performAsync(Page input, ActionListener<OngoingInference> listene
     }
 
     protected BulkInferenceExecutionConfig bulkExecutionConfig() {
-        return BulkInferenceExecutionConfig.DEFAULT;
+        return bulkExecutionConfig;
     }
 
     protected abstract BulkInferenceRequestIterator requests(Page input);

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

@@ -15,6 +15,7 @@
 import org.elasticsearch.threadpool.ThreadPool;
 import org.elasticsearch.xpack.esql.inference.InferenceOperator;
 import org.elasticsearch.xpack.esql.inference.InferenceRunner;
+import org.elasticsearch.xpack.esql.inference.bulk.BulkInferenceExecutionConfig;
 import org.elasticsearch.xpack.esql.inference.bulk.BulkInferenceRequestIterator;
 
 public class CompletionOperator extends InferenceOperator {
@@ -48,7 +49,7 @@ public CompletionOperator(
         String inferenceId,
         ExpressionEvaluator promptEvaluator
     ) {
-        super(driverContext, inferenceRunner, threadPool, inferenceId);
+        super(driverContext, inferenceRunner, BulkInferenceExecutionConfig.DEFAULT, threadPool, inferenceId);
         this.promptEvaluator = promptEvaluator;
     }
 

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

@@ -15,6 +15,7 @@
 import org.elasticsearch.threadpool.ThreadPool;
 import org.elasticsearch.xpack.esql.inference.InferenceOperator;
 import org.elasticsearch.xpack.esql.inference.InferenceRunner;
+import org.elasticsearch.xpack.esql.inference.bulk.BulkInferenceExecutionConfig;
 
 public class RerankOperator extends InferenceOperator {
     public record Factory(
@@ -61,7 +62,7 @@ public RerankOperator(
         ExpressionEvaluator rowEncoder,
         int scoreChannel
     ) {
-        super(driverContext, inferenceRunner, threadPool, inferenceId);
+        super(driverContext, inferenceRunner, BulkInferenceExecutionConfig.DEFAULT, threadPool, inferenceId);
         this.queryText = queryText;
         this.rowEncoder = rowEncoder;
         this.scoreChannel = scoreChannel;