From 4b3e1bc6dd2ec3c9874e47db43307953d99b2d94 Mon Sep 17 00:00:00 2001 From: Siddhesh Agarwal Date: Fri, 1 Nov 2024 12:55:24 +0530 Subject: [PATCH] added methods to toggle logging --- simplemind/settings.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/simplemind/settings.py b/simplemind/settings.py index d561228..a0dad2f 100644 --- a/simplemind/settings.py +++ b/simplemind/settings.py @@ -51,5 +51,19 @@ class Settings(BaseSettings): key = getattr(self, f"{provider.upper()}_API_KEY", None) return key.get_secret_value() if key else None + def enable_logging(self) -> None: + """Enable logging for the application.""" + # adding imports here to avoid forced dependencies + import logfire + from logging import basicConfig + + self.logging.enabled = True + logfire.configure() + basicConfig(handlers=[logfire.LogfireLoggingHandler()]) + + def disable_logging(self) -> None: + """Disable logging for the application.""" + self.logging.enabled = False + settings = Settings()