MetaGPT

1
#!/usr/bin/env python
2
# -*- coding: utf-8 -*-
3
"""
4
@Time    : 2023/6/1 12:41
5
@Author  : alexanderwu
6
@File    : logs.py
7
"""
8

9
import sys
10
from datetime import datetime
11

12
from loguru import logger as _logger
13

14
from metagpt.const import METAGPT_ROOT
15

16
_print_level = "INFO"
17

18

19
def define_log_level(print_level="INFO", logfile_level="DEBUG", name: str = None):
20
    """Adjust the log level to above level"""
21
    global _print_level
22
    _print_level = print_level
23

24
    current_date = datetime.now()
25
    formatted_date = current_date.strftime("%Y%m%d")
26
    log_name = f"{name}_{formatted_date}" if name else formatted_date  # name a log with prefix name
27

28
    _logger.remove()
29
    _logger.add(sys.stderr, level=print_level)
30
    _logger.add(METAGPT_ROOT / f"logs/{log_name}.txt", level=logfile_level)
31
    return _logger
32

33

34
logger = define_log_level()
35

36

37
def log_llm_stream(msg):
38
    _llm_stream_log(msg)
39

40

41
def set_llm_stream_logfunc(func):
42
    global _llm_stream_log
43
    _llm_stream_log = func
44

45

46
def _llm_stream_log(msg):
47
    if _print_level in ["INFO"]:
48
        print(msg, end="")
49