Skip to content

Commit c07cb29

Browse files
committed
Only add product use case header if not present
1 parent d5c0778 commit c07cb29

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/action/BaseTransportInferenceAction.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,13 @@ protected void doExecute(Task task, Request request, ActionListener<InferenceAct
150150
// We want to pass InferenceContext through the various infer methods in InferenceService in the long term
151151
var context = request.getContext();
152152
if (Objects.nonNull(context)) {
153-
threadPool.getThreadContext().putHeader(InferencePlugin.X_ELASTIC_PRODUCT_USE_CASE_HTTP_HEADER, context.productUseCase());
153+
var headerNotPresentInThreadContext = Objects.isNull(
154+
threadPool.getThreadContext().getHeader(InferencePlugin.X_ELASTIC_PRODUCT_USE_CASE_HTTP_HEADER)
155+
);
156+
if (headerNotPresentInThreadContext) {
157+
threadPool.getThreadContext()
158+
.putHeader(InferencePlugin.X_ELASTIC_PRODUCT_USE_CASE_HTTP_HEADER, context.productUseCase());
159+
}
154160
}
155161

156162
var service = serviceRegistry.getService(serviceName).get();

0 commit comments

Comments
 (0)