[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 <infra-root+elasticsearchmachine@elastic.co>
This commit is contained in:
parent
a3bff18ec1
commit
19e18a94a6
|
@ -0,0 +1,5 @@
|
||||||
|
pr: 128327
|
||||||
|
summary: Use internal user for internal inference action
|
||||||
|
area: Machine Learning
|
||||||
|
type: bug
|
||||||
|
issues: []
|
|
@ -21,6 +21,9 @@ import java.util.List;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import static org.elasticsearch.xpack.core.ClientHelper.INFERENCE_ORIGIN;
|
||||||
|
import static org.elasticsearch.xpack.core.ClientHelper.executeAsyncWithOrigin;
|
||||||
|
|
||||||
public class InferenceRunner {
|
public class InferenceRunner {
|
||||||
|
|
||||||
private final Client client;
|
private final Client client;
|
||||||
|
@ -73,6 +76,6 @@ public class InferenceRunner {
|
||||||
}
|
}
|
||||||
|
|
||||||
public void doInference(InferenceAction.Request request, ActionListener<InferenceAction.Response> listener) {
|
public void doInference(InferenceAction.Request request, ActionListener<InferenceAction.Response> listener) {
|
||||||
client.execute(InferenceAction.INSTANCE, request, listener);
|
executeAsyncWithOrigin(client, INFERENCE_ORIGIN, InferenceAction.INSTANCE, request, listener);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -25,6 +25,9 @@ import java.util.Arrays;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
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
|
* A {@code RankFeaturePhaseRankCoordinatorContext} that performs a rerank inference call to determine relevance scores for documents within
|
||||||
* the provided rank window.
|
* the provided rank window.
|
||||||
|
@ -114,7 +117,7 @@ public class TextSimilarityRankFeaturePhaseRankCoordinatorContext extends RankFe
|
||||||
List<String> featureData = Arrays.stream(featureDocs).map(x -> x.featureData).toList();
|
List<String> featureData = Arrays.stream(featureDocs).map(x -> x.featureData).toList();
|
||||||
InferenceAction.Request inferenceRequest = generateRequest(featureData);
|
InferenceAction.Request inferenceRequest = generateRequest(featureData);
|
||||||
try {
|
try {
|
||||||
client.execute(InferenceAction.INSTANCE, inferenceRequest, inferenceListener);
|
executeAsyncWithOrigin(client, INFERENCE_ORIGIN, InferenceAction.INSTANCE, inferenceRequest, inferenceListener);
|
||||||
} finally {
|
} finally {
|
||||||
inferenceRequest.decRef();
|
inferenceRequest.decRef();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue