feat: + DI args error

feat: + test case
This commit is contained in:
莘权 马 2024-03-31 11:49:16 +08:00
parent d9f70730f7
commit dd9b573eb4
6 changed files with 45 additions and 30 deletions

View file

@ -8,6 +8,7 @@
from __future__ import annotations
import json
import sys
from datetime import datetime
from functools import partial
@ -17,7 +18,6 @@ from loguru import logger as _logger
from pydantic import BaseModel, Field
from metagpt.const import METAGPT_ROOT
from metagpt.schema import BaseEnum
class ToolOutputItem(BaseModel):
@ -26,21 +26,6 @@ class ToolOutputItem(BaseModel):
value: str
class ToolName(str, BaseEnum):
Terminal = "Terminal"
Plan = "Plan"
Browser = "Browser"
Files = "Files"
WritePRD = "WritePRD"
WriteDesign = "WriteDesign"
WriteProjectPlan = "WriteProjectPlan"
WriteCode = "WriteCode"
WriteUntTest = "WriteUntTest"
FixBug = "FixBug"
GitArchive = "GitArchive"
ImportRepo = "ImportRepo"
def define_log_level(print_level="INFO", logfile_level="DEBUG", name: str = None):
"""Adjust the log level to above level"""
current_date = datetime.now()
@ -66,7 +51,7 @@ def log_tool_output(output: ToolOutputItem | List[ToolOutputItem], tool_name: st
return
outputs = output if isinstance(output, list) else [output]
_tool_output_log(output=[i.model_dump() for i in outputs], tool_name=tool_name)
_tool_output_log(output=json.dumps([i.model_dump() for i in outputs]), tool_name=tool_name)
def set_llm_stream_logfunc(func):
@ -82,4 +67,8 @@ def set_tool_output_logfunc(func):
_llm_stream_log = partial(print, end="")
_tool_output_log = partial(print, end="")
def _default_tool_output_log(*args, **kwargs):
print(*args, str(kwargs), end="")
_tool_output_log = _default_tool_output_log