Merge pull request #1124 from azurewtl/feature_mute_stream_log_for_info_level

Feature mute stream log for info level
This commit is contained in:
Alexander Wu 2024-04-05 21:35:42 +08:00 committed by GitHub
commit c6766e8cbf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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
@ -39,4 +43,6 @@ def set_llm_stream_logfunc(func):
_llm_stream_log = func
_llm_stream_log = partial(print, end="")
def _llm_stream_log(msg):
if _print_level in ["DEBUG"]:
print(msg, end="")