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