isolate codes to be restructured in the future

metagpt/actions/__init__.py

@@ -22,9 +22,9 @@ from metagpt.actions.write_code_review import WriteCodeReview
 from metagpt.actions.write_prd import WritePRD
 from metagpt.actions.write_prd_review import WritePRDReview
 from metagpt.actions.write_test import WriteTest
-from metagpt.actions.execute_nb_code import ExecuteNbCode
-from metagpt.actions.write_analysis_code import WriteCodeByGenerate
-from metagpt.actions.write_plan import WritePlan
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.write_analysis_code import WriteCodeByGenerate
+from metagpt.actions.ci.write_plan import WritePlan
 
 
 class ActionType(Enum):

metagpt/actions/ci/debug_code.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from metagpt.actions.write_analysis_code import BaseWriteAnalysisCode
+from metagpt.actions.ci.write_analysis_code import BaseWriteAnalysisCode
 from metagpt.logs import logger
 from metagpt.schema import Message
 from metagpt.utils.common import create_func_call_config

metagpt/actions/ci/ml_action.py

@@ -1,14 +1,14 @@
 from typing import List, Tuple
 
 from metagpt.actions import Action
-from metagpt.actions.write_analysis_code import WriteCodeWithTools
-from metagpt.prompts.ml_action import (
+from metagpt.actions.ci.write_analysis_code import WriteCodeWithTools
+from metagpt.prompts.ci.ml_action import (
     GENERATE_CODE_PROMPT,
     ML_TOOL_USAGE_PROMPT,
     PRINT_DATA_COLUMNS,
     UPDATE_DATA_COLUMNS,
 )
-from metagpt.prompts.write_analysis_code import CODE_GENERATOR_WITH_TOOLS
+from metagpt.prompts.ci.write_analysis_code import CODE_GENERATOR_WITH_TOOLS
 from metagpt.schema import Message, Plan
 from metagpt.utils.common import create_func_call_config, remove_comments
 

metagpt/actions/ci/write_analysis_code.py

@@ -8,7 +8,7 @@ from typing import Tuple
 
 from metagpt.actions import Action
 from metagpt.logs import logger
-from metagpt.prompts.write_analysis_code import (
+from metagpt.prompts.ci.write_analysis_code import (
     CODE_GENERATOR_WITH_TOOLS,
     SELECT_FUNCTION_TOOLS,
     TOOL_RECOMMENDATION_PROMPT,

metagpt/actions/ci/write_plan.py

@@ -10,7 +10,7 @@ from typing import Dict, List, Tuple
 
 from metagpt.actions import Action
 from metagpt.logs import logger
-from metagpt.prompts.write_analysis_code import (
+from metagpt.prompts.ci.write_analysis_code import (
     ASSIGN_TASK_TYPE_CONFIG,
     ASSIGN_TASK_TYPE_PROMPT,
 )

metagpt/plan/planner.py

@@ -2,8 +2,8 @@ import json
 
 from pydantic import BaseModel, Field
 
-from metagpt.actions.ask_review import AskReview, ReviewConst
-from metagpt.actions.write_plan import (
+from metagpt.actions.ci.ask_review import AskReview, ReviewConst
+from metagpt.actions.ci.write_plan import (
     WritePlan,
     precheck_update_plan_from_rsp,
     update_plan_from_rsp,

metagpt/roles/code_interpreter.py

@@ -1,8 +1,11 @@
 from pydantic import Field
 
-from metagpt.actions.ask_review import ReviewConst
-from metagpt.actions.execute_nb_code import ExecuteNbCode
-from metagpt.actions.write_analysis_code import WriteCodeByGenerate, WriteCodeWithTools
+from metagpt.actions.ci.ask_review import ReviewConst
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.write_analysis_code import (
+    WriteCodeByGenerate,
+    WriteCodeWithTools,
+)
 from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message, Task, TaskResult

metagpt/roles/ml_engineer.py

@@ -1,6 +1,6 @@
-from metagpt.actions.debug_code import DebugCode
-from metagpt.actions.execute_nb_code import ExecuteNbCode
-from metagpt.actions.ml_action import UpdateDataColumns, WriteCodeWithToolsML
+from metagpt.actions.ci.debug_code import DebugCode
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.ml_action import UpdateDataColumns, WriteCodeWithToolsML
 from metagpt.logs import logger
 from metagpt.roles.code_interpreter import CodeInterpreter
 from metagpt.tools.tool_types import ToolTypes

tests/metagpt/actions/ci/test_ask_review.py

@@ -1,6 +1,6 @@
 import pytest
 
-from metagpt.actions.ask_review import AskReview
+from metagpt.actions.ci.ask_review import AskReview
 
 
 @pytest.mark.asyncio

tests/metagpt/actions/ci/test_debug_code.py

@@ -5,7 +5,7 @@
 
 import pytest
 
-from metagpt.actions.debug_code import DebugCode
+from metagpt.actions.ci.debug_code import DebugCode
 from metagpt.schema import Message
 
 ErrorStr = """Tested passed:

tests/metagpt/actions/ci/test_execute_nb_code.py

@@ -1,6 +1,6 @@
 import pytest
 
-from metagpt.actions.execute_nb_code import ExecuteNbCode, truncate
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode, truncate
 
 
 @pytest.mark.asyncio

tests/metagpt/actions/ci/test_ml_action.py

@@ -1,6 +1,6 @@
 import pytest
 
-from metagpt.actions.ml_action import WriteCodeWithToolsML
+from metagpt.actions.ci.ml_action import WriteCodeWithToolsML
 from metagpt.schema import Plan, Task
 
 

tests/metagpt/actions/ci/test_write_analysis_code.py

@@ -2,8 +2,11 @@ import asyncio
 
 import pytest
 
-from metagpt.actions.execute_nb_code import ExecuteNbCode
-from metagpt.actions.write_analysis_code import WriteCodeByGenerate, WriteCodeWithTools
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.write_analysis_code import (
+    WriteCodeByGenerate,
+    WriteCodeWithTools,
+)
 from metagpt.logs import logger
 from metagpt.plan.planner import STRUCTURAL_CONTEXT
 from metagpt.schema import Message, Plan, Task

tests/metagpt/actions/ci/test_write_plan.py

@@ -1,6 +1,6 @@
 import pytest
 
-from metagpt.actions.write_plan import (
+from metagpt.actions.ci.write_plan import (
     Plan,
     Task,
     WritePlan,

tests/metagpt/roles/run_code_interpreter.py

@@ -1,6 +1,6 @@
 import fire
 
-from metagpt.actions.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
 from metagpt.const import DATA_PATH
 from metagpt.logs import logger
 from metagpt.roles.code_interpreter import CodeInterpreter

tests/metagpt/roles/test_code_interpreter.py

@@ -7,7 +7,7 @@ from metagpt.roles.code_interpreter import CodeInterpreter
 @pytest.mark.asyncio
 @pytest.mark.parametrize("auto_run", [(True), (False)])
 async def test_code_interpreter(mocker, auto_run):
-    mocker.patch("metagpt.actions.execute_nb_code.ExecuteNbCode.run", return_value=("a successful run", True))
+    mocker.patch("metagpt.actions.ci.execute_nb_code.ExecuteNbCode.run", return_value=("a successful run", True))
     mocker.patch("builtins.input", return_value="confirm")
 
     requirement = "Run data analysis on sklearn Iris dataset, include a plot"

tests/metagpt/roles/test_ml_engineer.py

@@ -1,11 +1,11 @@
 import pytest
 
-from metagpt.actions.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
 from metagpt.logs import logger
 from metagpt.roles.ml_engineer import MLEngineer
 from metagpt.schema import Message, Plan, Task
 from metagpt.tools.tool_types import ToolTypes
-from tests.metagpt.actions.test_debug_code import CODE, DebugContext, ErrorStr
+from tests.metagpt.actions.ci.test_debug_code import CODE, DebugContext, ErrorStr
 
 
 def test_mle_init():

tests/metagpt/utils/test_save_code.py

@@ -6,7 +6,7 @@
 import nbformat
 import pytest
 
-from metagpt.actions.execute_nb_code import ExecuteNbCode
+from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
 from metagpt.utils.common import read_json_file
 from metagpt.utils.save_code import DATA_PATH, save_code_file