apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: rfc243

Browse source
This commit is contained in:
莘权 马 2024-06-18 10:33:15 +08:00
parent b71af7b46a
commit 4dfe6ef26f
5 changed files with 77 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

22
metagpt/tools/libs/software_development.py
View file

@ -264,7 +264,7 @@ async def write_trd(
user_requirements: str,
investment: float = 10,
context: Optional[Context] = None,
) -> (str, str):
) -> str:
"""
Handles the writing of a Technical Requirements Document (TRD) based on user requirements.
@ -397,3 +397,23 @@ async def write_framework(
file_list = await save_framework(dir_data=framework, output_dir=output_dir)
logger.info(f"Output:\n{file_list}")
return "## Software Framework" + "".join([f"\n- {i}" for i in file_list])
@register_tool(tags=["system design", "write trd and framework", "Write a TRD and the framework"])
async def write_trd_and_framework(
use_case_actors: str,
user_requirements: str,
additional_technical_requirements: str,
investment: float = 15.0,
output_dir: Optional[str] = "",
context: Optional[Context] = None,
) -> str:
context = context or Context(cost_manager=CostManager(max_budget=investment))
trd = await write_trd(use_case_actors=use_case_actors, user_requirements=user_requirements, context=context)
return await write_framework(
use_case_actors=use_case_actors,
trd=trd,
additional_technical_requirements=additional_technical_requirements,
output_dir=output_dir,
context=context,
)