apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-05 14:55:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add sweagent to env

Browse source
This commit is contained in:
garylin2099 2024-07-04 17:33:55 +08:00
parent caaf42987d
commit 715e25f9b1
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/roles/di/swe_agent.py
View file

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

2
tests/metagpt/environment/mgx_env/run_mgx_env.py
View file

@ -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=["<all>"]),
SWEAgent(),
]
)