From b1b8b84d85968d886c69603a76f896283743d903 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Mon, 12 Aug 2024 11:41:18 +0800
Subject: [PATCH] feat: disable write trd

---
 metagpt/roles/architect.py                 | 5 +----
 metagpt/tools/libs/software_development.py | 4 ----
 2 files changed, 1 insertion(+), 8 deletions(-)

diff --git a/metagpt/roles/architect.py b/metagpt/roles/architect.py
index e37f00913..1af169ca1 100644
--- a/metagpt/roles/architect.py
+++ b/metagpt/roles/architect.py
@@ -8,7 +8,6 @@
 from metagpt.actions import WritePRD
 from metagpt.actions.design_api import WriteDesign
 from metagpt.roles.di.role_zero import RoleZero
-from metagpt.tools.libs.software_development import write_trd_and_framework
 from metagpt.utils.common import tool2name
 
 ARCHITECT_INSTRUCTION = """
@@ -33,7 +32,7 @@ class Architect(RoleZero):
 
     name: str = "Bob"
     profile: str = "Architect"
-    goal: str = "design a concise, usable, complete software system. ouput the system design or software framework."
+    goal: str = "design a concise, usable, complete software system. output the system design."
     constraints: str = (
         "make sure the architecture is simple enough and use  appropriate open source "
         "libraries. Use same language as user requirement"
@@ -45,7 +44,6 @@ class Architect(RoleZero):
         "Editor:write,read,write_content",
         "RoleZero",
         "WriteDesign",
-        write_trd_and_framework.__name__,
     ]
 
     def __init__(self, **kwargs) -> None:
@@ -64,7 +62,6 @@ class Architect(RoleZero):
         self.tool_execution_map.update(tool2name(WriteDesign, ["run"], write_design.run))
         self.tool_execution_map.update(
             {
-                write_trd_and_framework.__name__: write_trd_and_framework,
                 "run": write_design.run,  # alias
             }
         )
diff --git a/metagpt/tools/libs/software_development.py b/metagpt/tools/libs/software_development.py
index 1f8538dfc..1a20bf087 100644
--- a/metagpt/tools/libs/software_development.py
+++ b/metagpt/tools/libs/software_development.py
@@ -21,7 +21,6 @@ from metagpt.actions.requirement_analysis.trd import (
 from metagpt.const import ASSISTANT_ALIAS, DEFAULT_WORKSPACE_ROOT, TEST_DATA_PATH
 from metagpt.context import Context
 from metagpt.logs import ToolLogItem, log_tool_output, logger
-from metagpt.tools.tool_registry import register_tool
 from metagpt.utils.common import aread
 from metagpt.utils.cost_manager import CostManager
 
@@ -86,7 +85,6 @@ async def mock_asearch_acknowledgement(use_case_actors: str):
     return await aread(filename=TEST_DATA_PATH / "requirements/1.acknowledge.md")
 
 
-@register_tool(tags=["system design", "write trd", "Write a TRD"])
 async def write_trd(
     use_case_actors: str,
     user_requirements: str,
@@ -155,7 +153,6 @@ async def write_trd(
     return trd
 
 
-@register_tool(tags=["system design", "write software framework", "Write a software framework based on a TRD"])
 async def write_framework(
     use_case_actors: str,
     trd: str,
@@ -240,7 +237,6 @@ async def write_framework(
     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,