Prevent model server crash if data is missing in metadata file during…

… callHome
deepjavalibrary · Oct 3, 2023 · 36d5af0 · 36d5af0
1 parent 27c6a57
commit 36d5af0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/java/ai/djl/util/Ec2Utils.java b/api/src/main/java/ai/djl/util/Ec2Utils.java
@@ -178,7 +178,7 @@ private static String readEndpointMetadata(String key) {
         Path path = Paths.get(ENDPOINT_METADATA_FILE);
         try (Reader reader = Files.newBufferedReader(path)) {
             JsonObject json = JsonUtils.GSON.fromJson(reader, JsonObject.class);
-            return json.get(key).getAsString();
+            return (json.get(key) != null) ? json.get(key).getAsString() : null;
         } catch (IOException e) {
             // ignore
         }