From fdf88913a09e48bf24668d32fadd89779f7062ac Mon Sep 17 00:00:00 2001 From: Jonathan Buttner <56361221+jonathan-buttner@users.noreply.github.com> Date: Thu, 22 May 2025 14:59:12 -0400 Subject: [PATCH] [ML] Use internal user for internal inference action (#128327) * Using correct origin for inference action * Update docs/changelog/128327.yaml * [CI] Auto commit changes from spotless --------- Co-authored-by: elasticsearchmachine (cherry picked from commit 19e18a94a655aa07a4c38edcd4bde1d688cbfafe) # Conflicts: # x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/inference/InferenceRunner.java --- docs/changelog/128327.yaml | 5 +++++ ...TextSimilarityRankFeaturePhaseRankCoordinatorContext.java | 5 ++++- 2 files changed, 9 insertions(+), 1 deletion(-) create mode 100644 docs/changelog/128327.yaml diff --git a/docs/changelog/128327.yaml b/docs/changelog/128327.yaml new file mode 100644 index 0000000000000..5f98ca6b9019c --- /dev/null +++ b/docs/changelog/128327.yaml @@ -0,0 +1,5 @@ +pr: 128327 +summary: Use internal user for internal inference action +area: Machine Learning +type: bug +issues: [] diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java index 7f245ae854eac..ee85c455a8ba9 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java @@ -25,6 +25,9 @@ import java.util.List; import java.util.Map; +import static org.elasticsearch.xpack.core.ClientHelper.INFERENCE_ORIGIN; +import static org.elasticsearch.xpack.core.ClientHelper.executeAsyncWithOrigin; + /** * A {@code RankFeaturePhaseRankCoordinatorContext} that performs a rerank inference call to determine relevance scores for documents within * the provided rank window. @@ -114,7 +117,7 @@ protected void computeScores(RankFeatureDoc[] featureDocs, ActionListener featureData = Arrays.stream(featureDocs).map(x -> x.featureData).toList(); InferenceAction.Request inferenceRequest = generateRequest(featureData); try { - client.execute(InferenceAction.INSTANCE, inferenceRequest, inferenceListener); + executeAsyncWithOrigin(client, INFERENCE_ORIGIN, InferenceAction.INSTANCE, inferenceRequest, inferenceListener); } finally { inferenceRequest.decRef(); }