Skip to content

Commit d0bf1de

Browse files
authored
Xcontent fix for chunk scorer (#137218)
1 parent 4245e18 commit d0bf1de

File tree

1 file changed

+14
-1
lines changed
  • x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity

1 file changed

+14
-1
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/ChunkScorerConfig.java

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,14 +11,16 @@
1111
import org.elasticsearch.common.io.stream.StreamOutput;
1212
import org.elasticsearch.common.io.stream.Writeable;
1313
import org.elasticsearch.inference.ChunkingSettings;
14+
import org.elasticsearch.xcontent.ToXContentObject;
15+
import org.elasticsearch.xcontent.XContentBuilder;
1416
import org.elasticsearch.xpack.core.inference.chunking.ChunkingSettingsBuilder;
1517
import org.elasticsearch.xpack.core.inference.chunking.SentenceBoundaryChunkingSettings;
1618

1719
import java.io.IOException;
1820
import java.util.Map;
1921
import java.util.Objects;
2022

21-
public class ChunkScorerConfig implements Writeable {
23+
public class ChunkScorerConfig implements Writeable, ToXContentObject {
2224

2325
public final Integer size;
2426
private final String inferenceText;
@@ -97,4 +99,15 @@ public boolean equals(Object o) {
9799
public int hashCode() {
98100
return Objects.hash(size, inferenceText, chunkingSettings);
99101
}
102+
103+
@Override
104+
public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
105+
builder.startObject();
106+
builder.field("size", size);
107+
builder.field("inference_text", inferenceText);
108+
builder.field("chunking_settings");
109+
chunkingSettings.toXContent(builder, params);
110+
builder.endObject();
111+
return builder;
112+
}
100113
}

0 commit comments

Comments
 (0)