Skip to content

Commit 842c659

Browse files
committed
For now, all inference is happening on coordinator, so make it clear
1 parent baaeec6 commit 842c659

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
import org.elasticsearch.xpack.esql.core.tree.Source;
2424
import org.elasticsearch.xpack.esql.core.type.DataType;
2525
import org.elasticsearch.xpack.esql.io.stream.PlanStreamInput;
26-
import org.elasticsearch.xpack.esql.plan.logical.ExecutesOn;
2726
import org.elasticsearch.xpack.esql.plan.logical.LogicalPlan;
2827

2928
import java.io.IOException;
@@ -34,7 +33,7 @@
3433
import static org.elasticsearch.xpack.esql.core.type.DataType.TEXT;
3534
import static org.elasticsearch.xpack.esql.expression.NamedExpressions.mergeOutputAttributes;
3635

37-
public class Completion extends InferencePlan<Completion> implements TelemetryAware, PostAnalysisVerificationAware, ExecutesOn.Coordinator {
36+
public class Completion extends InferencePlan<Completion> implements TelemetryAware, PostAnalysisVerificationAware {
3837

3938
public static final String DEFAULT_OUTPUT_FIELD_NAME = "completion";
4039

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import org.elasticsearch.xpack.esql.core.expression.UnresolvedAttribute;
1414
import org.elasticsearch.xpack.esql.core.tree.Source;
1515
import org.elasticsearch.xpack.esql.plan.GeneratingPlan;
16+
import org.elasticsearch.xpack.esql.plan.logical.ExecutesOn;
1617
import org.elasticsearch.xpack.esql.plan.logical.LogicalPlan;
1718
import org.elasticsearch.xpack.esql.plan.logical.SortAgnostic;
1819
import org.elasticsearch.xpack.esql.plan.logical.UnaryPlan;
@@ -24,7 +25,8 @@
2425
public abstract class InferencePlan<PlanType extends InferencePlan<PlanType>> extends UnaryPlan
2526
implements
2627
SortAgnostic,
27-
GeneratingPlan<InferencePlan<PlanType>> {
28+
GeneratingPlan<InferencePlan<PlanType>>,
29+
ExecutesOn.Coordinator {
2830

2931
public static final String INFERENCE_ID_OPTION_NAME = "inference_id";
3032
public static final List<String> VALID_INFERENCE_OPTION_NAMES = List.of(INFERENCE_ID_OPTION_NAME);

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
import org.elasticsearch.xpack.esql.core.type.DataType;
2727
import org.elasticsearch.xpack.esql.io.stream.PlanStreamInput;
2828
import org.elasticsearch.xpack.esql.plan.logical.Eval;
29-
import org.elasticsearch.xpack.esql.plan.logical.ExecutesOn;
3029
import org.elasticsearch.xpack.esql.plan.logical.LogicalPlan;
3130
import org.elasticsearch.xpack.esql.plan.logical.UnaryPlan;
3231

@@ -38,7 +37,7 @@
3837
import static org.elasticsearch.xpack.esql.common.Failure.fail;
3938
import static org.elasticsearch.xpack.esql.expression.NamedExpressions.mergeOutputAttributes;
4039

41-
public class Rerank extends InferencePlan<Rerank> implements PostAnalysisVerificationAware, TelemetryAware, ExecutesOn.Coordinator {
40+
public class Rerank extends InferencePlan<Rerank> implements PostAnalysisVerificationAware, TelemetryAware {
4241

4342
public static final NamedWriteableRegistry.Entry ENTRY = new NamedWriteableRegistry.Entry(LogicalPlan.class, "Rerank", Rerank::new);
4443
public static final String DEFAULT_INFERENCE_ID = ".rerank-v1-elasticsearch";

0 commit comments

Comments
 (0)