Throwing a parsing exception when the RERANK (or COMPLETION) command is disabled.

Author: afoucret

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/parser/LogicalPlanBuilder.java

@@ -55,6 +55,7 @@
 import org.elasticsearch.xpack.esql.expression.predicate.logical.Not;
 import org.elasticsearch.xpack.esql.expression.predicate.operator.comparison.Equals;
 import org.elasticsearch.xpack.esql.expression.predicate.operator.comparison.EsqlBinaryComparison;
+import org.elasticsearch.xpack.esql.inference.InferenceCommandConfig;
 import org.elasticsearch.xpack.esql.parser.promql.PromqlParserUtils;
 import org.elasticsearch.xpack.esql.plan.EsqlStatement;
 import org.elasticsearch.xpack.esql.plan.IndexPattern;
@@ -1119,9 +1120,14 @@ public PlanFactory visitRerankCommand(EsqlBaseParser.RerankCommandContext ctx) {
             throw qualifiersUnsupportedInFieldDefinitions(scoreAttribute.source(), ctx.targetField.getText());
         }
 
-        Literal rowLimit = Literal.integer(Source.EMPTY, context.inferenceCommandConfigProvider().get("rerank").rowLimit());
+        InferenceCommandConfig commandConfig = context.inferenceCommandConfigProvider().get("rerank");
+        Literal rowLimit = Literal.integer(Source.EMPTY, commandConfig.rowLimit());
 
         return p -> {
+            if (commandConfig.enabled() == false) {
+                throw new ParsingException(source, "RERANK command is disabled");
+            }
+
             checkForRemoteClusters(p, source, "RERANK");
             return applyRerankOptions(
                 new Rerank(source, p, queryText, rerankFields, scoreAttribute, rowLimit),
@@ -1164,11 +1170,17 @@ public PlanFactory visitCompletionCommand(EsqlBaseParser.CompletionCommandContex
         if (targetField.qualifier() != null) {
             throw qualifiersUnsupportedInFieldDefinitions(targetField.source(), ctx.targetField.getText());
         }
-
-        Literal rowLimit = Literal.integer(Source.EMPTY, context.inferenceCommandConfigProvider().get("completion").rowLimit());
+        
+        InferenceCommandConfig commandConfig = context.inferenceCommandConfigProvider().get("completion");
+        Literal rowLimit = Literal.integer(Source.EMPTY, commandConfig.rowLimit());
 
         return p -> {
+            if (commandConfig.enabled() == false) {
+                throw new ParsingException(source, "COMPLETION command is disabled");
+            }
+
             checkForRemoteClusters(p, source, "COMPLETION");
+
             return applyCompletionOptions(new Completion(source, p, prompt, targetField, rowLimit), ctx.commandNamedParameters());
         };
     }

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/parser/StatementParserTests.java

@@ -4224,6 +4224,17 @@ public void testRerankRowLimitOverride() {
         assertThat(plan.rowLimit(), equalTo(Literal.integer(EMPTY, customRowLimit)));
     }
 
+    public void testRerankCommandDisabled() {
+        EsqlParser parserWithDisabledRerank = new EsqlParser(
+            Settings.builder().put(InferenceCommandConfig.RERANK_ENABLED_SETTING.getKey(), false).build()
+        );
+        ParsingException pe = expectThrows(
+            ParsingException.class,
+            () -> parserWithDisabledRerank.createStatement("FROM foo* | RERANK \"query text\" ON title")
+        );
+        assertThat(pe.getMessage(), containsString("RERANK command is disabled"));
+    }
+
     public void testInvalidRerank() {
         expectError(
             "FROM foo* | RERANK \"query text\" ON title WITH { \"inference_id\": 3 }",
@@ -4303,6 +4314,19 @@ public void testCompletionRowLimitOverride() {
         assertThat(plan.rowLimit(), equalTo(Literal.integer(EMPTY, customRowLimit)));
     }
 
+    public void testCompletionCommandDisabled() {
+        EsqlParser parserWithDisabledCompletion = new EsqlParser(
+            Settings.builder().put(InferenceCommandConfig.COMPLETION_ENABLED_SETTING.getKey(), false).build()
+        );
+        ParsingException pe = expectThrows(
+            ParsingException.class,
+            () -> parserWithDisabledCompletion.createStatement(
+                "FROM foo* | COMPLETION targetField = prompt WITH { \"inference_id\": \"test\" }"
+            )
+        );
+        assertThat(pe.getMessage(), containsString("COMPLETION command is disabled"));
+    }
+
     public void testCompletionWithPositionalParameters() {
         var queryParams = new QueryParams(List.of(paramAsConstant(null, "inferenceId")));
         var plan = as(