minor test fix

Author: tteofili

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/action/filter/ShardBulkInferenceActionFilterTests.java

@@ -241,7 +241,7 @@ public void testItemFailures() throws Exception {
         );
         model.putResult(
             "I am a failure",
-            new ChunkedInferenceError(new InferenceException("inference exception", new IllegalArgumentException("boom")))
+            new ChunkedInferenceError(new IllegalArgumentException("boom"))
         );
         model.putResult("I am a success", randomChunkedInferenceEmbeddingSparse(List.of("I am a success")));
         CountDownLatch chainExecuted = new CountDownLatch(1);
@@ -255,8 +255,8 @@ public void testItemFailures() throws Exception {
                 assertNotNull(bulkShardRequest.items()[0].getPrimaryResponse());
                 assertTrue(bulkShardRequest.items()[0].getPrimaryResponse().isFailed());
                 BulkItemResponse.Failure failure = bulkShardRequest.items()[0].getPrimaryResponse().getFailure();
-                assertThat(failure.getCause().getCause().getMessage(), containsString("inference exception"));
-                assertThat(failure.getCause().getCause().getCause().getMessage(), containsString("boom"));
+                assertThat(failure.getCause().getMessage(), containsString("Exception when running inference"));
+                assertThat(failure.getCause().getCause().getMessage(), containsString("boom"));
                 assertThat(failure.getStatus(), is(RestStatus.BAD_REQUEST));
 
                 // item 1 is a success
@@ -276,8 +276,8 @@ public void testItemFailures() throws Exception {
                 assertNotNull(bulkShardRequest.items()[2].getPrimaryResponse());
                 assertTrue(bulkShardRequest.items()[2].getPrimaryResponse().isFailed());
                 failure = bulkShardRequest.items()[2].getPrimaryResponse().getFailure();
-                assertThat(failure.getCause().getCause().getMessage(), containsString("inference exception"));
-                assertThat(failure.getCause().getCause().getCause().getMessage(), containsString("boom"));
+                assertThat(failure.getCause().getMessage(), containsString("Exception when running inference"));
+                assertThat(failure.getCause().getCause().getMessage(), containsString("boom"));
                 assertThat(failure.getStatus(), is(RestStatus.BAD_REQUEST));
             } finally {
                 chainExecuted.countDown();