Skip to content

Commit 276b6f1

Browse files
committed
Removed matched text from chunk
1 parent a935afd commit 276b6f1

File tree

4 files changed

+5
-7
lines changed

4 files changed

+5
-7
lines changed

server/src/main/java/org/elasticsearch/inference/ChunkedInference.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,11 @@ public interface ChunkedInference {
2626
Iterator<Chunk> chunksAsMatchedTextAndByteReference(XContent xcontent) throws IOException;
2727

2828
/**
29-
* A chunk of inference results containing matched text, the substring location
30-
* in the original text and the bytes reference.
31-
* @param matchedText
29+
* A chunk of inference results containing the substring location in the original text and the bytes reference.
3230
* @param textOffset
3331
* @param bytesReference
3432
*/
35-
record Chunk(String matchedText, TextOffset textOffset, BytesReference bytesReference) {}
33+
record Chunk(TextOffset textOffset, BytesReference bytesReference) {}
3634

3735
record TextOffset(int start, int end) {}
3836
}

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/SparseEmbeddingResults.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ public record Chunk(List<WeightedToken> weightedTokens, String matchedText, Chun
185185
EmbeddingResults.Chunk {
186186

187187
public ChunkedInference.Chunk toChunk(XContent xcontent) throws IOException {
188-
return new ChunkedInference.Chunk(matchedText, offset, toBytesReference(xcontent, weightedTokens));
188+
return new ChunkedInference.Chunk(offset, toBytesReference(xcontent, weightedTokens));
189189
}
190190

191191
private static BytesReference toBytesReference(XContent xContent, List<WeightedToken> tokens) throws IOException {

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/TextEmbeddingByteResults.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ public Chunk toChunk(String text, ChunkedInference.TextOffset offset) {
198198
public record Chunk(byte[] embedding, String matchedText, ChunkedInference.TextOffset offset) implements EmbeddingResults.Chunk {
199199

200200
public ChunkedInference.Chunk toChunk(XContent xcontent) throws IOException {
201-
return new ChunkedInference.Chunk(matchedText, offset, toBytesReference(xcontent, embedding));
201+
return new ChunkedInference.Chunk(offset, toBytesReference(xcontent, embedding));
202202
}
203203

204204
private static BytesReference toBytesReference(XContent xContent, byte[] value) throws IOException {

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/TextEmbeddingFloatResults.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ public Chunk toChunk(String text, ChunkedInference.TextOffset offset) {
229229
public record Chunk(float[] embedding, String matchedText, ChunkedInference.TextOffset offset) implements EmbeddingResults.Chunk {
230230

231231
public ChunkedInference.Chunk toChunk(XContent xcontent) throws IOException {
232-
return new ChunkedInference.Chunk(matchedText, offset, toBytesReference(xcontent, embedding));
232+
return new ChunkedInference.Chunk(offset, toBytesReference(xcontent, embedding));
233233
}
234234

235235
/**

0 commit comments

Comments
 (0)