diff --git a/metagpt/llm.py b/metagpt/llm.py index b8aefec61..c6755919d 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -1,17 +1,22 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- """ -@Time : 2023/5/11 14:45 +@Time : 2023/6/1 12:41 @Author : alexanderwu -@File : llm.py +@File : logs.py """ -from metagpt.provider.anthropic_api import Claude2 as Claude -from metagpt.provider.openai_api import OpenAIGPTAPI as LLM +import sys -DEFAULT_LLM = LLM() -CLAUDE_LLM = Claude() +from loguru import logger as _logger -async def ai_func(prompt): - """Use LLM for Q&A.""" - return await DEFAULT_LLM.aask(prompt) +from metagpt.const import PROJECT_ROOT + +def define_log_level(print_level="INFO", logfile_level="DEBUG"): + """Adjust log level to above the specified level.""" + _logger.remove() + _logger.add(sys.stderr, level=print_level) + _logger.add(PROJECT_ROOT / 'logs/log.txt', level=logfile_level) + return _logger + +logger = define_log_level()