Update plugin/src/main/java/org/opensearch/ml/action/handler/MLSearch…

…Handler.java Co-authored-by: Hailong Cui <[email protected]> Signed-off-by: samuel-oci <[email protected]>
opensearch-project · Jan 12, 2024 · 091c47c · 091c47c
1 parent 8db913e
commit 091c47c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/handler/MLSearchHandler.java b/plugin/src/main/java/org/opensearch/ml/action/handler/MLSearchHandler.java
@@ -202,7 +202,7 @@ private QueryBuilder rewriteQueryBuilder(QueryBuilder queryBuilder, List<String>
     public static <T> ActionListener<T> wrapRestActionListener(ActionListener<T> actionListener, String generalErrorMessage) {
         return ActionListener.<T>wrap(r -> { actionListener.onResponse(r); }, e -> {
             log.error("Wrap exception before sending back to user", e);
-            Throwable cause = ExceptionUtils.getRootCause(e);
+            Throwable cause = org.opensearch.ExceptionsHelper.unwrap(e);
             if (isProperExceptionToReturn(e)) {
                 actionListener.onFailure(e);
             } else if (isProperExceptionToReturn(cause)) {