isolate codes to be restructured in the future

This commit is contained in:
yzlin 2024-02-04 10:45:02 +08:00
parent 3648669bd2
commit 24d2c5c8e6
22 changed files with 35 additions and 29 deletions

View file

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

View file

@ -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:

View file

@ -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

View file

@ -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

View file

@ -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

View file

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

View file

@ -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

View file

@ -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"

View file

@ -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():

View file

@ -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