Skip to content

Commit 84bbf39

Browse files
[CI] Auto commit changes from spotless
1 parent 51e9baa commit 84bbf39

File tree

2 files changed

+8
-12
lines changed

2 files changed

+8
-12
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,6 @@ protected BulkInferenceRequestIterator requests(Page inputPage) {
7373
return new CompletionOperatorRequestIterator(inputPage.getBlock(inputBlockChannel), inferenceId());
7474
}
7575

76-
7776
@Override
7877
protected DenseEmbeddingOperatorOutputBuilder outputBuilder(Page input) {
7978
// TODO: add multiply expectd size by the dimension
@@ -89,7 +88,7 @@ protected DenseEmbeddingOperatorOutputBuilder outputBuilder(Page input) {
8988
*/
9089
public record Factory(InferenceRunner inferenceRunner, String inferenceId, ExpressionEvaluator.Factory inputTextEvaluatorFactory)
9190
implements
92-
OperatorFactory {
91+
OperatorFactory {
9392
@Override
9493
public String describe() {
9594
return "CompletionOperator[inference_id=[" + inferenceId + "]]";

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

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ public void close() {
4040
releasePageOnAnyThread(inputPage);
4141
}
4242

43-
4443
@Override
4544
public void addInferenceResponse(InferenceAction.Response inferenceResponse) {
4645
EmbeddingResults<?> embeddingResults = InferenceOperator.OutputBuilder.inferenceResults(inferenceResponse, EmbeddingResults.class);
@@ -59,22 +58,20 @@ private List<float[]> readEmbeddings(InferenceAction.Response inferenceResponse)
5958
if (embeddingResults instanceof TextEmbeddingFloatResults(List<TextEmbeddingFloatResults.Embedding> embeddings)) {
6059
return embeddings.stream().map(TextEmbeddingFloatResults.Embedding::values).toList();
6160
} else if (embeddingResults instanceof TextEmbeddingByteResults || embeddingResults instanceof TextEmbeddingBitResults) {
62-
return embeddingResults.embeddings().stream().map(TextEmbeddingByteResults.Embedding.class::cast)
63-
.map(embedding -> {
64-
float[] values = new float[embedding.values().length];
65-
for (int i = 0; i < embedding.values().length; i++) {
66-
values[i] = embedding.values()[i];
67-
}
68-
return values;
69-
}).toList();
61+
return embeddingResults.embeddings().stream().map(TextEmbeddingByteResults.Embedding.class::cast).map(embedding -> {
62+
float[] values = new float[embedding.values().length];
63+
for (int i = 0; i < embedding.values().length; i++) {
64+
values[i] = embedding.values()[i];
65+
}
66+
return values;
67+
}).toList();
7068
}
7169

7270
throw new IllegalStateException(
7371
"Unknown embedding result type [" + embeddingResults.getClass() + "]; expecting a [{" + TextEmbeddingResults.class + "}] "
7472
);
7573
}
7674

77-
7875
/**
7976
* Constructs a new output {@link Page} which contains all original blocks from the input page, with the embeddings inserted at the end.
8077
*/

0 commit comments

Comments
 (0)