feat: show full tracebacks in debug mode

bentoml · Sep 13, 2022 · 0535e19 · 0535e19
1 parent 7b5efb2
commit 0535e19
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/bentoml_cli/utils.py b/bentoml_cli/utils.py
@@ -15,6 +15,7 @@
 
 from bentoml.exceptions import BentoMLException
 from bentoml._internal.log import configure_logging
+from bentoml._internal.configuration import get_debug_mode
 from bentoml._internal.configuration import set_debug_mode
 from bentoml._internal.configuration import set_quiet_mode
 from bentoml._internal.utils.analytics import track
@@ -163,8 +164,12 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> t.Any:
             try:
                 return func(*args, **kwargs)
             except BentoMLException as err:
-                msg = f"[{cmd_group.name}] '{command_name}' failed: {str(err)}"
-                raise ClickException(click.style(msg, fg="red")) from err
+                msg = f"[{cmd_group.name}] `{command_name}` failed: {str(err)}"
+                if get_debug_mode():
+                    ClickException(click.style(msg, fg="red")).show()
+                    raise err from None
+                else:
+                    raise ClickException(click.style(msg, fg="red")) from err
 
         return t.cast("ClickFunctionWrapper[t.Any]", wrapper)