feat: adding intermediate steps processing

This was required because if the user uses log-level debug the logs show up in the chat
This commit is contained in:
Gabriel Almeida 2023-04-03 10:11:53 -03:00
commit 3e66c5a5ce
3 changed files with 42 additions and 11 deletions

View file

@ -1,14 +1,20 @@
import logging
from pathlib import Path
from rich.logging import RichHandler
logger = logging.getLogger("langflow")
def configure(log_level: str = "INFO", log_file: Path = None): # type: ignore
log_format = "%(asctime)s - %(levelname)s - %(message)s"
log_level = getattr(logging, log_level.upper(), logging.INFO)
log_level_value = getattr(logging, log_level.upper(), logging.INFO)
logging.basicConfig(level=log_level, format=log_format)
logging.basicConfig(
level=log_level_value,
format=log_format,
datefmt="[%X]",
handlers=[RichHandler(rich_tracebacks=True)],
)
if log_file:
log_file = Path(log_file)
@ -18,6 +24,6 @@ def configure(log_level: str = "INFO", log_file: Path = None): # type: ignore
file_handler.setFormatter(logging.Formatter(log_format))
logger.addHandler(file_handler)
logger.info(f"Logger set up with log level: {log_level}")
logger.info(f"Logger set up with log level: {log_level_value}({log_level})")
if log_file:
logger.info(f"Log file: {log_file}")