Skip to content

Commit 3809f63

Browse files
committed
better names
1 parent ccc4023 commit 3809f63

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/elasticsearch/ElasticsearchInternalService.java

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -675,7 +675,7 @@ public void chunkedInfer(
675675
// Avoid filling the inference queue by executing the batches in series
676676
// Each batch contains up to EMBEDDING_MAX_BATCH_SIZE inference request
677677
var sequentialRunner = new BatchIterator(esModel, inputType, timeout, batchedRequests);
678-
sequentialRunner.doNextRequest();
678+
sequentialRunner.run();
679679
}
680680
} else {
681681
listener.onFailure(notElasticsearchModelException(model));
@@ -1015,11 +1015,11 @@ class BatchIterator {
10151015
this.timeout = timeout;
10161016
}
10171017

1018-
void doNextRequest() {
1019-
inferenceExecutor.execute(() -> inferOnBatch(requestAndListeners.get(index.get())));
1018+
void run() {
1019+
inferenceExecutor.execute(() -> inferBatchAndRunAfter(requestAndListeners.get(index.get())));
10201020
}
10211021

1022-
private void inferOnBatch(EmbeddingRequestChunker.BatchRequestAndListener batch) {
1022+
private void inferBatchAndRunAfter(EmbeddingRequestChunker.BatchRequestAndListener batch) {
10231023
var inferenceRequest = buildInferenceRequest(
10241024
esModel.mlNodeDeploymentId(),
10251025
EmptyConfigUpdate.INSTANCE,
@@ -1034,14 +1034,14 @@ private void inferOnBatch(EmbeddingRequestChunker.BatchRequestAndListener batch)
10341034
);
10351035

10361036
// schedule the next request once the results have been processed
1037-
var scheduleNextListener = ActionListener.runAfter(mlResultsListener, () -> {
1037+
var runNextListener = ActionListener.runAfter(mlResultsListener, () -> {
10381038
if (index.incrementAndGet() < requestAndListeners.size()) {
1039-
doNextRequest();
1039+
run();
10401040
}
10411041
});
10421042

1043-
var maybeDeployListener = scheduleNextListener.delegateResponse(
1044-
(l, exception) -> maybeStartDeployment(esModel, exception, inferenceRequest, scheduleNextListener)
1043+
var maybeDeployListener = runNextListener.delegateResponse(
1044+
(l, exception) -> maybeStartDeployment(esModel, exception, inferenceRequest, runNextListener)
10451045
);
10461046

10471047
client.execute(InferModelAction.INSTANCE, inferenceRequest, maybeDeployListener);

0 commit comments

Comments
 (0)