Browse Source

ESQL - Enable telemetry for COMPLETION command (#127731)

Svilen Mihaylov 4 months ago
parent
commit
7eed09644c

+ 5 - 0
docs/changelog/127731.yaml

@@ -0,0 +1,5 @@
+pr: 127731
+summary: ESQL - Enable telemetry for COMPLETION command
+area: Search
+type: feature
+issues: []

+ 2 - 0
x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/logical/inference/Completion.java

@@ -12,6 +12,7 @@ import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.common.io.stream.StreamOutput;
 import org.elasticsearch.inference.TaskType;
 import org.elasticsearch.xpack.esql.capabilities.PostAnalysisVerificationAware;
+import org.elasticsearch.xpack.esql.capabilities.TelemetryAware;
 import org.elasticsearch.xpack.esql.common.Failures;
 import org.elasticsearch.xpack.esql.core.expression.Attribute;
 import org.elasticsearch.xpack.esql.core.expression.AttributeSet;
@@ -37,6 +38,7 @@ public class Completion extends InferencePlan<Completion>
     implements
         GeneratingPlan<Completion>,
         SortAgnostic,
+        TelemetryAware,
         PostAnalysisVerificationAware {
 
     public static final String DEFAULT_OUTPUT_FIELD_NAME = "completion";