From 0b2b074e07c8da32f1eb9f6d968eb80e98519ada Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 27 Sep 2023 08:51:17 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(logger.py):=20refactor=20c?=
 =?UTF-8?q?onfigure=20function=20to=20improve=20readability=20and=20add=20?=
 =?UTF-8?q?support=20for=20configurable=20log=20level=20=F0=9F=94=92=20cho?=
 =?UTF-8?q?re(logger.py):=20add=20VALID=5FLOG=5FLEVELS=20constant=20to=20d?=
 =?UTF-8?q?efine=20valid=20log=20levels=20for=20better=20maintainability?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/utils/logger.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/utils/logger.py b/src/backend/langflow/utils/logger.py
index 05b7adc0e..e49fa4bfd 100644
--- a/src/backend/langflow/utils/logger.py
+++ b/src/backend/langflow/utils/logger.py
@@ -7,6 +7,9 @@ import orjson
 import appdirs
 
 
+VALID_LOG_LEVELS = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
+
+
 def serialize(record):
     subset = {
         "timestamp": record["time"].timestamp(),
@@ -21,9 +24,11 @@ def patching(record):
     record["extra"]["serialized"] = serialize(record)
 
 
-def configure(log_level: str = "INFO", log_file: Optional[Path] = None):
-    if os.getenv("LANGFLOW_LOG_LEVEL"):
+def configure(log_level: Optional[str] = None, log_file: Optional[Path] = None):
+    if os.getenv("LANGFLOW_LOG_LEVEL") in VALID_LOG_LEVELS and not log_level:
         log_level = os.getenv("LANGFLOW_LOG_LEVEL")
+    elif not log_level:
+        log_level = "INFO"
     # Human-readable
     log_format = "<green>{time:YYYY-MM-DD HH:mm:ss}</green> - <level>{level: <8}</level> - {module} - <level>{message}</level>"