File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change 77
77
import static org .elasticsearch .xpack .esql .action .EsqlCapabilities .Cap .RERANK ;
78
78
import static org .elasticsearch .xpack .esql .action .EsqlCapabilities .Cap .SEMANTIC_TEXT_FIELD_CAPS ;
79
79
import static org .elasticsearch .xpack .esql .action .EsqlCapabilities .Cap .SOURCE_FIELD_MAPPING ;
80
+ import static org .elasticsearch .xpack .esql .action .EsqlCapabilities .Cap .TEXT_EMBEDDING_FUNCTION ;
80
81
import static org .hamcrest .Matchers .anyOf ;
81
82
import static org .hamcrest .Matchers .is ;
82
83
import static org .hamcrest .Matchers .nullValue ;
@@ -251,8 +252,12 @@ protected boolean supportsInferenceTestService() {
251
252
}
252
253
253
254
protected boolean requiresInferenceEndpoint () {
254
- return Stream .of (SEMANTIC_TEXT_FIELD_CAPS .capabilityName (), RERANK .capabilityName (), COMPLETION .capabilityName ())
255
- .anyMatch (testCase .requiredCapabilities ::contains );
255
+ return Stream .of (
256
+ SEMANTIC_TEXT_FIELD_CAPS .capabilityName (),
257
+ RERANK .capabilityName (),
258
+ COMPLETION .capabilityName (),
259
+ TEXT_EMBEDDING_FUNCTION .capabilityName ()
260
+ ).anyMatch (testCase .requiredCapabilities ::contains );
256
261
}
257
262
258
263
protected boolean supportsIndexModeLookup () throws IOException {
You can’t perform that action at this time.
0 commit comments