afoucret
diff --git a/‎x-pack/plugin/esql/src/main/antlr/EsqlBaseParser.g4‎
Lines changed: 9 additions & 1 deletion b/‎x-pack/plugin/esql/src/main/antlr/EsqlBaseParser.g4‎
Lines changed: 9 additions & 1 deletion
diff --git a/‎x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/parser/EsqlBaseParser.interp‎
Lines changed: 3 additions & 1 deletion b/‎x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/parser/EsqlBaseParser.interp‎
Lines changed: 3 additions & 1 deletion
@@ -136,6 +136,14 @@ field
     : (qualifiedName ASSIGN)? booleanExpression
     ;
 
+rerankFields
+    : rerankField (COMMA rerankField)*
+    ;
+
+rerankField
+    : qualifiedName (ASSIGN booleanExpression)?
+    ;
+
 fromCommand
     : FROM indexPattern (COMMA indexPattern)* metadata?
     ;
@@ -370,7 +378,7 @@ joinPredicate
     ;
 
 rerankCommand
-    : DEV_RERANK queryText=constant ON fields WITH inferenceId=identifierOrParameter
+    : DEV_RERANK queryText=constant ON rerankFields WITH inferenceId=identifierOrParameter
     ;
 
 completionCommand
Original file line number	Diff line number	Diff line change
`@@ -136,6 +136,14 @@ field`
`136`	`136`	`: (qualifiedName ASSIGN)? booleanExpression`
`137`	`137`	`;`
`138`	`138`
	`139`	`+rerankFields`
	`140`	`+ : rerankField (COMMA rerankField)*`
	`141`	`+ ;`
	`142`	`+`
	`143`	`+rerankField`
	`144`	`+ : qualifiedName (ASSIGN booleanExpression)?`
	`145`	`+ ;`
	`146`	`+`
`139`	`147`	`fromCommand`
`140`	`148`	`: FROM indexPattern (COMMA indexPattern)* metadata?`
`141`	`149`	`;`
`@@ -370,7 +378,7 @@ joinPredicate`
`370`	`378`	`;`
`371`	`379`
`372`	`380`	`rerankCommand`
`373`		`- : DEV_RERANK queryText=constant ON fields WITH inferenceId=identifierOrParameter`
	`381`	`+ : DEV_RERANK queryText=constant ON rerankFields WITH inferenceId=identifierOrParameter`
`374`	`382`	`;`
`375`	`383`
`376`	`384`	`completionCommand`