Skip to content

Commit c6b8d8d

Browse files
[ML] Use internal user for internal inference action (#128327) (#128333)
* Using correct origin for inference action * Update docs/changelog/128327.yaml * [CI] Auto commit changes from spotless --------- Co-authored-by: elasticsearchmachine <[email protected]> (cherry picked from commit 19e18a9) # Conflicts: # x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/inference/InferenceRunner.java
1 parent 605b15c commit c6b8d8d

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

docs/changelog/128327.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
pr: 128327
2+
summary: Use internal user for internal inference action
3+
area: Machine Learning
4+
type: bug
5+
issues: []

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,9 @@
2525
import java.util.List;
2626
import java.util.Map;
2727

28+
import static org.elasticsearch.xpack.core.ClientHelper.INFERENCE_ORIGIN;
29+
import static org.elasticsearch.xpack.core.ClientHelper.executeAsyncWithOrigin;
30+
2831
/**
2932
* A {@code RankFeaturePhaseRankCoordinatorContext} that performs a rerank inference call to determine relevance scores for documents within
3033
* the provided rank window.
@@ -113,7 +116,7 @@ protected void computeScores(RankFeatureDoc[] featureDocs, ActionListener<float[
113116
List<String> featureData = Arrays.stream(featureDocs).map(x -> x.featureData).toList();
114117
InferenceAction.Request inferenceRequest = generateRequest(featureData);
115118
try {
116-
client.execute(InferenceAction.INSTANCE, inferenceRequest, inferenceListener);
119+
executeAsyncWithOrigin(client, INFERENCE_ORIGIN, InferenceAction.INSTANCE, inferenceRequest, inferenceListener);
117120
} finally {
118121
inferenceRequest.decRef();
119122
}

0 commit comments

Comments
 (0)