refine code

metagpt/actions/__init__.py

@@ -13,7 +13,7 @@ from metagpt.actions.add_requirement import UserRequirement
 from metagpt.actions.debug_error import DebugError
 from metagpt.actions.design_api import WriteDesign
 from metagpt.actions.design_api_review import DesignReview
-from metagpt.actions.project_management import AssignTasks, WriteTasks
+from metagpt.actions.project_management import WriteTasks
 from metagpt.actions.research import CollectLinks, WebBrowseAndSummarize, ConductResearch
 from metagpt.actions.run_code import RunCode
 from metagpt.actions.search_and_summarize import SearchAndSummarize
@@ -38,7 +38,6 @@ class ActionType(Enum):
     RUN_CODE = RunCode
     DEBUG_ERROR = DebugError
     WRITE_TASKS = WriteTasks
-    ASSIGN_TASKS = AssignTasks
     SEARCH_AND_SUMMARIZE = SearchAndSummarize
     COLLECT_LINKS = CollectLinks
     WEB_BROWSE_AND_SUMMARIZE = WebBrowseAndSummarize

metagpt/actions/add_requirement.py

@@ -10,6 +10,3 @@ from metagpt.actions import Action
 
 class UserRequirement(Action):
     """User Requirement without any implementation details"""
-
-    async def run(self, *args, **kwargs):
-        raise NotImplementedError

metagpt/actions/design_api_an.py

@@ -8,7 +8,6 @@
 from typing import List
 
 from metagpt.actions.action_node import ActionNode
-from metagpt.logs import logger
 from metagpt.utils.mermaid import MMC1, MMC2
 
 IMPLEMENTATION_APPROACH = ActionNode(
@@ -63,12 +62,3 @@ NODES = [
 ]
 
 DESIGN_API_NODE = ActionNode.from_children("DesignAPI", NODES)
-
-
-def main():
-    prompt = DESIGN_API_NODE.compile(context="")
-    logger.info(prompt)
-
-
-if __name__ == "__main__":
-    main()

metagpt/actions/project_management.py

@@ -123,9 +123,3 @@ class WriteTasks(Action):
     @staticmethod
     async def _save_pdf(task_doc):
         await FileRepository.save_as(doc=task_doc, with_suffix=".md", relative_path=TASK_PDF_FILE_REPO)
-
-
-class AssignTasks(Action):
-    async def run(self, *args, **kwargs):
-        # Here you should implement the actual action
-        pass

tests/metagpt/actions/test_invoice_ocr.py

@@ -20,7 +20,7 @@ from metagpt.actions.invoice_ocr import GenerateTable, InvoiceOCR, ReplyQuestion
     "invoice_path",
     [
         "../../data/invoices/invoice-3.jpg",
-        "../../data/invoices/invoice-4.zip",
+        # "../../data/invoices/invoice-4.zip",
     ],
 )
 async def test_invoice_ocr(invoice_path: str):