diff --git a/examples/di/interacting_with_human.py b/examples/di/interacting_with_human.py index c0ce02a40..a02f7c3bc 100644 --- a/examples/di/interacting_with_human.py +++ b/examples/di/interacting_with_human.py @@ -1,4 +1,5 @@ import fire + from metagpt.environment.mgx.mgx_env import MGXEnv from metagpt.logs import logger from metagpt.roles.di.team_leader import TeamLeader diff --git a/examples/write_design.py b/examples/write_design.py index 7eaa1a87b..fbbe9e1f0 100644 --- a/examples/write_design.py +++ b/examples/write_design.py @@ -1,10 +1,10 @@ import asyncio +from metagpt.environment.mgx.mgx_env import MGXEnv from metagpt.logs import logger from metagpt.roles.architect import Architect -from metagpt.environment.mgx.mgx_env import MGXEnv -from metagpt.schema import Message from metagpt.roles.di.team_leader import TeamLeader +from metagpt.schema import Message async def main(): diff --git a/tests/metagpt/roles/test_architect.py b/tests/metagpt/roles/test_architect.py index 9a834ffbe..cb636b6a1 100644 --- a/tests/metagpt/roles/test_architect.py +++ b/tests/metagpt/roles/test_architect.py @@ -8,18 +8,19 @@ distribution feature for message handling. """ import uuid +from pathlib import Path import pytest from metagpt.actions import WritePRD +from metagpt.actions.di.run_command import RunCommand from metagpt.const import PRDS_FILE_REPO from metagpt.logs import logger from metagpt.roles import Architect from metagpt.schema import Message from metagpt.utils.common import any_to_str, awrite from tests.metagpt.roles.mock import MockMessages -from pathlib import Path -from metagpt.actions.di.run_command import RunCommand + @pytest.mark.asyncio async def test_architect(context): diff --git a/tests/metagpt/roles/test_engineer.py b/tests/metagpt/roles/test_engineer.py index c340e910f..18b297ae5 100644 --- a/tests/metagpt/roles/test_engineer.py +++ b/tests/metagpt/roles/test_engineer.py @@ -9,6 +9,7 @@ """ import json from pathlib import Path +from types import SimpleNamespace import pytest @@ -19,9 +20,8 @@ from metagpt.roles.engineer import Engineer from metagpt.schema import CodingContext, Message from metagpt.utils.common import CodeParser, any_to_name, any_to_str, aread, awrite from metagpt.utils.git_repository import ChangeType -from tests.metagpt.roles.mock import STRS_FOR_PARSING, TASKS, MockMessages from metagpt.utils.project_repo import ProjectRepo -from types import SimpleNamespace +from tests.metagpt.roles.mock import STRS_FOR_PARSING, TASKS, MockMessages @pytest.mark.asyncio