Skip to content

Commit 06f29f5

Browse files
committed
Added SemanticQueryBuilder that takes pre-computed inference results
1 parent 48aa247 commit 06f29f5

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/queries/SemanticQueryBuilder.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,10 @@ public SemanticQueryBuilder(String fieldName, String query) {
8383
}
8484

8585
public SemanticQueryBuilder(String fieldName, String query, Boolean lenient) {
86+
this(fieldName, query, lenient, null);
87+
}
88+
89+
protected SemanticQueryBuilder(String fieldName, String query, Boolean lenient, InferenceResultsProvider inferenceResultsProvider) {
8690
if (fieldName == null) {
8791
throw new IllegalArgumentException("[" + NAME + "] requires a " + FIELD_FIELD.getPreferredName() + " value");
8892
}
@@ -91,7 +95,7 @@ public SemanticQueryBuilder(String fieldName, String query, Boolean lenient) {
9195
}
9296
this.fieldName = fieldName;
9397
this.query = query;
94-
this.inferenceResultsProvider = null;
98+
this.inferenceResultsProvider = inferenceResultsProvider;
9599
this.lenient = lenient;
96100
}
97101

0 commit comments

Comments
 (0)