From 715e25f9b11673a0df0cf4efe6f2e5b0c48bc06c Mon Sep 17 00:00:00 2001 From: garylin2099 Date: Thu, 4 Jul 2024 17:33:55 +0800 Subject: [PATCH] add sweagent to env --- metagpt/roles/di/swe_agent.py | 6 ++++-- tests/metagpt/environment/mgx_env/run_mgx_env.py | 2 ++ 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/di/swe_agent.py b/metagpt/roles/di/swe_agent.py index 166de3918..2e1fb6412 100644 --- a/metagpt/roles/di/swe_agent.py +++ b/metagpt/roles/di/swe_agent.py @@ -4,14 +4,13 @@ import os from pydantic import Field from metagpt.logs import logger -from metagpt.prompts.di.swe import ( +from metagpt.prompts.di.swe_agent import ( MINIMAL_EXAMPLE, NEXT_STEP_TEMPLATE, SWE_AGENT_SYSTEM_TEMPLATE, ) from metagpt.roles.di.role_zero import RoleZero from metagpt.tools.libs.terminal import Bash -from metagpt.tools.swe_agent_commands.swe_agent_utils import extract_patch class SWEAgent(RoleZero): @@ -72,6 +71,9 @@ class SWEAgent(RoleZero): This function is specifically added for SWE bench evaluation. """ + # only import when evaluation is needed + from metagpt.tools.swe_agent_commands.swe_agent_utils import extract_patch + commands, ok = await self._parse_commands() if not ok: return diff --git a/tests/metagpt/environment/mgx_env/run_mgx_env.py b/tests/metagpt/environment/mgx_env/run_mgx_env.py index 0d5287412..2cd299f17 100644 --- a/tests/metagpt/environment/mgx_env/run_mgx_env.py +++ b/tests/metagpt/environment/mgx_env/run_mgx_env.py @@ -6,6 +6,7 @@ from metagpt.environment.mgx.mgx_env import MGXEnv from metagpt.roles import Architect, Engineer, ProductManager, ProjectManager from metagpt.roles.di.data_analyst import DataAnalyst from metagpt.roles.di.engineer2 import Engineer2 +from metagpt.roles.di.swe_agent import SWEAgent from metagpt.roles.di.team_leader import TeamLeader from metagpt.schema import Message @@ -26,6 +27,7 @@ async def main(requirement="", enable_human_input=False, use_fixed_sop=False): engineer, # QaEngineer(), DataAnalyst(tools=[""]), + SWEAgent(), ] )