Skip to content

Commit aa3851f

Browse files
afoucretjoshua-adams-1
authored andcommitted
[ES|QL] Enable telemetry for the rerank command (elastic#128679)
1 parent 632478d commit aa3851f

File tree

1 file changed

+2
-1
lines changed
  • x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/logical/inference

1 file changed

+2
-1
lines changed

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/logical/inference/Rerank.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import org.elasticsearch.common.io.stream.StreamInput;
1212
import org.elasticsearch.common.io.stream.StreamOutput;
1313
import org.elasticsearch.inference.TaskType;
14+
import org.elasticsearch.xpack.esql.capabilities.TelemetryAware;
1415
import org.elasticsearch.xpack.esql.core.capabilities.Resolvables;
1516
import org.elasticsearch.xpack.esql.core.expression.Alias;
1617
import org.elasticsearch.xpack.esql.core.expression.Attribute;
@@ -37,7 +38,7 @@
3738
import static org.elasticsearch.xpack.esql.core.expression.Expressions.asAttributes;
3839
import static org.elasticsearch.xpack.esql.expression.NamedExpressions.mergeOutputAttributes;
3940

40-
public class Rerank extends InferencePlan<Rerank> implements SortAgnostic, SurrogateLogicalPlan {
41+
public class Rerank extends InferencePlan<Rerank> implements SortAgnostic, SurrogateLogicalPlan, TelemetryAware {
4142

4243
public static final NamedWriteableRegistry.Entry ENTRY = new NamedWriteableRegistry.Entry(LogicalPlan.class, "Rerank", Rerank::new);
4344
private final Attribute scoreAttribute;

0 commit comments

Comments
 (0)