Skip to content

Commit 255019d

Browse files
committed
use internal ml user
1 parent f87ac85 commit 255019d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportStopTrainedModelDeploymentAction.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import org.elasticsearch.action.support.ActionFilters;
1919
import org.elasticsearch.action.support.tasks.TransportTasksAction;
2020
import org.elasticsearch.client.internal.Client;
21+
import org.elasticsearch.client.internal.OriginSettingClient;
2122
import org.elasticsearch.cluster.ClusterState;
2223
import org.elasticsearch.cluster.node.DiscoveryNode;
2324
import org.elasticsearch.cluster.node.DiscoveryNodes;
@@ -51,6 +52,7 @@
5152
import java.util.Set;
5253

5354
import static org.elasticsearch.core.Strings.format;
55+
import static org.elasticsearch.xpack.core.ClientHelper.ML_ORIGIN;
5456
import static org.elasticsearch.xpack.ml.action.TransportDeleteTrainedModelAction.getModelAliases;
5557

5658
/**
@@ -67,7 +69,7 @@ public class TransportStopTrainedModelDeploymentAction extends TransportTasksAct
6769

6870
private static final Logger logger = LogManager.getLogger(TransportStopTrainedModelDeploymentAction.class);
6971

70-
private final Client client;
72+
private final OriginSettingClient client;
7173
private final TrainedModelAssignmentClusterService trainedModelAssignmentClusterService;
7274
private final InferenceAuditor auditor;
7375

@@ -89,7 +91,7 @@ public TransportStopTrainedModelDeploymentAction(
8991
StopTrainedModelDeploymentAction.Response::new,
9092
EsExecutors.DIRECT_EXECUTOR_SERVICE
9193
);
92-
this.client = client;
94+
this.client = new OriginSettingClient(client, ML_ORIGIN);
9395
this.trainedModelAssignmentClusterService = trainedModelAssignmentClusterService;
9496
this.auditor = Objects.requireNonNull(auditor);
9597
}

0 commit comments

Comments
 (0)