diff --git a/metagpt/llm.py b/metagpt/llm.py index c6755919d..b8aefec61 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -1,22 +1,17 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- """ -@Time : 2023/6/1 12:41 +@Time : 2023/5/11 14:45 @Author : alexanderwu -@File : logs.py +@File : llm.py """ -import sys +from metagpt.provider.anthropic_api import Claude2 as Claude +from metagpt.provider.openai_api import OpenAIGPTAPI as LLM -from loguru import logger as _logger +DEFAULT_LLM = LLM() +CLAUDE_LLM = Claude() -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() +async def ai_func(prompt): + """Use LLM for Q&A.""" + return await DEFAULT_LLM.aask(prompt)