[ES\QL] Text embedding function constant folding

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/EsqlSpecTestCase.java

@@ -77,6 +77,7 @@
 import static org.elasticsearch.xpack.esql.action.EsqlCapabilities.Cap.RERANK;
 import static org.elasticsearch.xpack.esql.action.EsqlCapabilities.Cap.SEMANTIC_TEXT_FIELD_CAPS;
 import static org.elasticsearch.xpack.esql.action.EsqlCapabilities.Cap.SOURCE_FIELD_MAPPING;
+import static org.elasticsearch.xpack.esql.action.EsqlCapabilities.Cap.TEXT_EMBEDDING_FUNCTION;
 import static org.elasticsearch.xpack.esql.qa.rest.RestEsqlTestCase.assertNotPartial;
 import static org.elasticsearch.xpack.esql.qa.rest.RestEsqlTestCase.hasCapabilities;
 
@@ -205,7 +206,8 @@ protected boolean requiresInferenceEndpoint() {
             SEMANTIC_TEXT_FIELD_CAPS.capabilityName(),
             RERANK.capabilityName(),
             COMPLETION.capabilityName(),
-            KNN_FUNCTION_V5.capabilityName()
+            KNN_FUNCTION_V5.capabilityName(),
+            TEXT_EMBEDDING_FUNCTION.capabilityName()
         ).anyMatch(testCase.requiredCapabilities::contains);
     }
 

x-pack/plugin/esql/qa/testFixtures/src/main/resources/text-embedding.csv-spec

@@ -1,15 +1,68 @@
-placeholder 
+text_embedding using a row source operator
 required_capability: text_embedding_function
-required_capability: not_existing_capability
+required_capability: dense_vector_field_type
 
 // tag::embedding-eval[]
 ROW input="Who is Victor Hugo?"
 | EVAL embedding = TEXT_EMBEDDING("Who is Victor Hugo?", "test_dense_inference")
 ;
 // end::embedding-eval[]
 
+input:keyword       | embedding:dense_vector
+Who is Victor Hugo? | [56.0, 50.0, 48.0]
+;
+
+
+text_embedding using a row source operator with query build using CONCAT
+required_capability: text_embedding_function
+required_capability: dense_vector_field_type
+
+ROW input="Who is Victor Hugo?"
+| EVAL embedding = TEXT_EMBEDDING(CONCAT("Who is ", "Victor Hugo?"), "test_dense_inference")
+;
 
 input:keyword       | embedding:dense_vector
 Who is Victor Hugo? | [56.0, 50.0, 48.0]
 ;
 
+
+text_embedding with knn on semantic_text_dense_field
+required_capability: text_embedding_function
+required_capability: dense_vector_field_type
+required_capability: knn_function_v5
+required_capability: semantic_text_field_caps
+
+FROM semantic_text METADATA _score
+| EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference")
+| WHERE KNN(semantic_text_dense_field, query_embedding)
+| KEEP semantic_text_field, query_embedding, _score
+| EVAL _score = ROUND(_score, 4)
+| SORT _score DESC
+| LIMIT 10
+;
+
+semantic_text_field:text                                              | query_embedding:dense_vector | _score:double
+be excellent to each other                                            | [45.0, 55.0, 54.0]           | 1.0
+live long and prosper                                                 | [45.0, 55.0, 54.0]           | 0.0295
+all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0]           | 0.0214
+;
+
+text_embedding with knn (inline) on semantic_text_dense_field
+required_capability: text_embedding_function
+required_capability: dense_vector_field_type
+required_capability: knn_function_v5
+required_capability: semantic_text_field_caps
+
+FROM semantic_text METADATA _score
+| WHERE KNN(semantic_text_dense_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference"))
+| KEEP semantic_text_field, _score
+| EVAL _score = ROUND(_score, 4)
+| SORT _score DESC
+| LIMIT 10
+;
+
+semantic_text_field:text                                              | _score:double
+be excellent to each other                                            | 1.0
+live long and prosper                                                 | 0.0295
+all we have to decide is what to do with the time that is given to us | 0.0214
+;

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/execution/PlanExecutor.java

@@ -88,7 +88,7 @@ public void esql(
             indexResolver,
             enrichPolicyResolver,
             preAnalyzer,
-            new LogicalPlanPreOptimizer(new LogicalPreOptimizerContext(foldContext)),
+            new LogicalPlanPreOptimizer(new LogicalPreOptimizerContext(foldContext, services.inferenceService())),
             functionRegistry,
             new LogicalPlanOptimizer(new LogicalOptimizerContext(cfg, foldContext)),
             mapper,