[ML] Pass timeout to chat completion

prwhelan · prwhelan · commit 44af9140e355 · 2025-05-22T16:17:44.000-04:00
diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/action/TransportUnifiedCompletionInferenceAction.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/action/TransportUnifiedCompletionInferenceAction.java
@@ -89,7 +89,7 @@ protected void doInference(
         InferenceService service,
         ActionListener<InferenceServiceResults> listener
     ) {
-        service.unifiedCompletionInfer(model, request.getUnifiedCompletionRequest(), null, listener);
+        service.unifiedCompletionInfer(model, request.getUnifiedCompletionRequest(), request.getTimeout(), listener);
     }
 
     @Override

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ protected void doInference(`
`89`	`89`	`InferenceService service,`
`90`	`90`	`ActionListener<InferenceServiceResults> listener`
`91`	`91`	`) {`
`92`		`- service.unifiedCompletionInfer(model, request.getUnifiedCompletionRequest(), null, listener);`
	`92`	`+ service.unifiedCompletionInfer(model, request.getUnifiedCompletionRequest(), request.getTimeout(), listener);`
`93`	`93`	`}`
`94`	`94`
`95`	`95`	`@Override`