From 10cae23501bf1ff5fbc8b515e77c4a15350b78ee Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Thu, 28 Dec 2023 16:15:51 +0800
Subject: [PATCH] refine code

---
 metagpt/actions/__init__.py               |  3 +--
 metagpt/actions/add_requirement.py        |  3 ---
 metagpt/actions/design_api_an.py          | 10 ----------
 metagpt/actions/project_management.py     |  6 ------
 tests/metagpt/actions/test_invoice_ocr.py |  2 +-
 5 files changed, 2 insertions(+), 22 deletions(-)

diff --git a/metagpt/actions/__init__.py b/metagpt/actions/__init__.py
index c34c72ed2..5b995bab6 100644
--- a/metagpt/actions/__init__.py
+++ b/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
diff --git a/metagpt/actions/add_requirement.py b/metagpt/actions/add_requirement.py
index d77d423ba..5d2a489b2 100644
--- a/metagpt/actions/add_requirement.py
+++ b/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
diff --git a/metagpt/actions/design_api_an.py b/metagpt/actions/design_api_an.py
index 7d6802381..3737203cf 100644
--- a/metagpt/actions/design_api_an.py
+++ b/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()
diff --git a/metagpt/actions/project_management.py b/metagpt/actions/project_management.py
index 7eda89130..3fde6e171 100644
--- a/metagpt/actions/project_management.py
+++ b/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
diff --git a/tests/metagpt/actions/test_invoice_ocr.py b/tests/metagpt/actions/test_invoice_ocr.py
index 12b1b4b30..d569fda21 100644
--- a/tests/metagpt/actions/test_invoice_ocr.py
+++ b/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):