mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-05-05 05:42:37 +02:00
feat: + DI args error
feat: + test case
This commit is contained in:
parent
d9f70730f7
commit
dd9b573eb4
6 changed files with 45 additions and 30 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue