diff --git a/lang_agent/config.py b/lang_agent/config.py index 1db3cb7..1419bd5 100644 --- a/lang_agent/config.py +++ b/lang_agent/config.py @@ -15,7 +15,7 @@ class PrintableConfig: lines = [self.__class__.__name__ + ":"] for key, val in vars(self).items(): - if key.endswith("_secret") or ("key" in key): + if self.is_secrete(key): val = str(val) val = val[:3] + "*"*(len(val) - 3)