Skip to content

Commit 906b1e9

Browse files
sending null so infernece queries will pick up cluster timeout settings
1 parent 9bccc5d commit 906b1e9

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/search/SparseVectorQueryBuilder.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ protected QueryBuilder doRewrite(QueryRewriteContext queryRewriteContext) {
258258
List.of(query),
259259
TextExpansionConfigUpdate.EMPTY_UPDATE,
260260
false,
261-
InferModelAction.Request.DEFAULT_TIMEOUT_FOR_API
261+
null
262262
);
263263
inferRequest.setHighPriority(true);
264264
inferRequest.setPrefixType(TrainedModelPrefixStrings.PrefixType.SEARCH);

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/search/TextExpansionQueryBuilder.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ protected QueryBuilder doRewrite(QueryRewriteContext queryRewriteContext) {
162162
List.of(modelText),
163163
TextExpansionConfigUpdate.EMPTY_UPDATE,
164164
false,
165-
InferModelAction.Request.DEFAULT_TIMEOUT_FOR_API
165+
null
166166
);
167167
inferRequest.setHighPriority(true);
168168
inferRequest.setPrefixType(TrainedModelPrefixStrings.PrefixType.SEARCH);

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/vectors/TextEmbeddingQueryVectorBuilder.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ public void buildVector(Client client, ActionListener<float[]> listener) {
116116
List.of(modelText),
117117
TextEmbeddingConfigUpdate.EMPTY_INSTANCE,
118118
false,
119-
InferModelAction.Request.DEFAULT_TIMEOUT_FOR_API
119+
null
120120
);
121121

122122
inferRequest.setHighPriority(true);

0 commit comments

Comments
 (0)