From 4a999095a3bb202d0a3de4e3f3774fc64ebf3f05 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AA=91=E9=A9=AC=E5=B0=8F=E7=8C=AB?= <1435130236@qq.com> Date: Thu, 18 Jul 2024 12:02:09 +0800 Subject: [PATCH] update errors (#357) --- erniebot/src/erniebot/errors.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/erniebot/src/erniebot/errors.py b/erniebot/src/erniebot/errors.py index c664c8c96..71a3b853c 100644 --- a/erniebot/src/erniebot/errors.py +++ b/erniebot/src/erniebot/errors.py @@ -35,6 +35,12 @@ class EBError(Exception): """Base exception class for the erniebot library.""" + def __init__(self, message: Optional[Any] = None, *args, **kwargs): + if message is None: + message = str(self.__class__.__name__) + message = message + super().__init__(message, *args, **kwargs) + class ArgumentNotFoundError(EBError): """An argument was not found."""