catch exception other than APIException

SeldonIO · Oct 28, 2019 · 214c684 · 214c684
1 parent 0f6a531
commit 214c684
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/engine/src/main/java/io/seldon/engine/service/InternalPredictionService.java b/engine/src/main/java/io/seldon/engine/service/InternalPredictionService.java
@@ -523,6 +523,14 @@ private SeldonMessage queryREST(
           logger.error("Invalid protocol buffer during Json Format merge - ", e);
           throw new APIException(
                   APIException.ApiExceptionType.ENGINE_MICROSERVICE_ERROR, e.toString());
+      } catch (APIException e)
+      {
+          throw e;
+      } catch (Exception e)
+      {
+          logger.error("Couldn't retrieve prediction from external prediction server - ", e);
+          throw new APIException(
+                  APIException.ApiExceptionType.ENGINE_MICROSERVICE_ERROR, e.toString());
       }
     }
     logger.error("Failed to retrueve predictions after {} attempts", restRetries);