From 70a95fc223e6c9429c74a999e2b5624e66b573d0 Mon Sep 17 00:00:00 2001 From: wj-Mcat <1435130236@qq.com> Date: Thu, 18 Jul 2024 11:50:07 +0800 Subject: [PATCH] update errors --- 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."""