- define_log_level(print_level="info") will mute stream log

This commit is contained in:
Azure Wang 2024-03-27 18:01:51 +08:00
parent 643450388a
commit de901ce3ce

View file

@ -14,9 +14,13 @@ from loguru import logger as _logger
from metagpt.const import METAGPT_ROOT
_print_level = "INFO"
def define_log_level(print_level="INFO", logfile_level="DEBUG", name: str = None):
"""Adjust the log level to above level"""
global _print_level
_print_level = print_level
current_date = datetime.now()
formatted_date = current_date.strftime("%Y%m%d")
log_name = f"{name}_{formatted_date}" if name else formatted_date # name a log with prefix name
@ -31,7 +35,8 @@ logger = define_log_level()
def log_llm_stream(msg):
_llm_stream_log(msg)
if _print_level in ["DEBUG"]:
_llm_stream_log(msg)
def set_llm_stream_logfunc(func):