From 6de204116194e39a9ade73fbd2cec79b33d60049 Mon Sep 17 00:00:00 2001
From: Frost Ming <me@frostming.com>
Date: Mon, 6 May 2024 18:04:19 +0800
Subject: [PATCH] fix: show warning instead of hard failure when model is
 missing (#4693)

Signed-off-by: Frost Ming <me@frostming.com>
---
 src/bentoml/_internal/bento/bento.py | 10 +++++++++-
 src/bentoml/_internal/log.py         |  7 +++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/bentoml/_internal/bento/bento.py b/src/bentoml/_internal/bento/bento.py
index 623b344dbdb..41e15702b9e 100644
--- a/src/bentoml/_internal/bento/bento.py
+++ b/src/bentoml/_internal/bento/bento.py
@@ -420,7 +420,15 @@ def total_size(
                     continue
                 except NotFound:
                     pass
-            global_model = model_store.get(model.tag)
+            try:
+                global_model = model_store.get(model.tag)
+            except NotFound:
+                logger.warning(
+                    "Bento: %s: Missing model %s",
+                    self.tag,
+                    model.tag,
+                )
+                continue
             total_size += global_model.file_size
         return total_size
 
diff --git a/src/bentoml/_internal/log.py b/src/bentoml/_internal/log.py
index 57b1545b4d0..68dbb262213 100644
--- a/src/bentoml/_internal/log.py
+++ b/src/bentoml/_internal/log.py
@@ -23,15 +23,22 @@ def filter(self, record: logging.LogRecord) -> bool:
     "version": 1,
     "disable_existing_loggers": True,
     "filters": {"infofilter": {"()": InfoFilter}},
+    "formatters": {
+        "simple": {
+            "format": "%(levelname)s: %(message)s",
+        }
+    },
     "handlers": {
         "bentomlhandler": {
             "class": "logging.StreamHandler",
             "filters": ["infofilter"],
             "stream": "ext://sys.stdout",
+            "formatter": "simple",
         },
         "defaulthandler": {
             "class": "logging.StreamHandler",
             "level": logging.WARNING,
+            "formatter": "simple",
         },
     },
     "loggers": {