From 36d5af08639417570438894150e36c62a7ad6323 Mon Sep 17 00:00:00 2001
From: Aaqib Ansari <maaquib@gmail.com>
Date: Tue, 3 Oct 2023 14:00:43 -0700
Subject: [PATCH] Prevent model server crash if data is missing in metadata
 file during callHome

---
 api/src/main/java/ai/djl/util/Ec2Utils.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/src/main/java/ai/djl/util/Ec2Utils.java b/api/src/main/java/ai/djl/util/Ec2Utils.java
index 178c3d7efe7..ca430fb3eb8 100644
--- 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
         }