mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
823298e0d0
* avoid repeat logging of unneeded messages * refactored append/edit_file (tests next) * agentskills and unit test fixes * testing * more changes and test prompts * smaller changes * final test fixes * remove dead code from test_agent.py * reverting unneeded changes * updated tests, more tweaks to skills * refactor (#2442) * chores: fix DelegatorAgent description (#2446) * change * change comments * fix * stopped container to prevent port issues. (#2447) * chore: remove useless browsing code in CodeActSWEAgent (#2438) * remove useless * fix integration test * Regenerate test_ipython_module artifacts for CodeActSWEAgent --------- Co-authored-by: Boxuan Li <liboxuan@connect.hku.hk> * Merge remote-tracking branch 'upstream/main' into agent-fileops * unneeded tweak * * fix edit_file to not introduce extra newline * updated docstrings with more details for LLM * fix legacy typo in prompts causing ]] instead of ] * several mock files regenerated * Regen'ed CodeActSWEAgent integration tests * fix _print_window signature; explicit exception type in _is_valid_path * splitlines with named param --------- Co-authored-by: Yufan Song <33971064+yufansong@users.noreply.github.com> Co-authored-by: மனோஜ்குமார் பழனிச்சாமி <smartmanoj42857@gmail.com> Co-authored-by: Boxuan Li <liboxuan@connect.hku.hk>
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import json
|
|
import os
|
|
from unittest.mock import MagicMock
|
|
|
|
import yaml
|
|
|
|
from agenthub.micro.registry import all_microagents
|
|
from opendevin.controller.agent import Agent
|
|
from opendevin.controller.state.state import State
|
|
from opendevin.events import EventSource
|
|
from opendevin.events.action import MessageAction
|
|
from opendevin.events.observation import NullObservation
|
|
|
|
|
|
def test_all_agents_are_loaded():
|
|
assert all_microagents is not None
|
|
assert len(all_microagents) > 1
|
|
|
|
base = os.path.join('agenthub', 'micro')
|
|
full_path = os.path.dirname(__file__) + '/../../' + base
|
|
agent_names = set()
|
|
for root, _, files in os.walk(full_path):
|
|
for file in files:
|
|
if file == 'agent.yaml':
|
|
file_path = os.path.join(root, file)
|
|
with open(file_path, 'r') as yaml_file:
|
|
data = yaml.safe_load(yaml_file)
|
|
agent_names.add(data['name'])
|
|
assert agent_names == set(all_microagents.keys())
|
|
|
|
|
|
def test_coder_agent_with_summary():
|
|
"""
|
|
Coder agent should render code summary as part of prompt
|
|
"""
|
|
mock_llm = MagicMock()
|
|
content = json.dumps({'action': 'finish', 'args': {}})
|
|
mock_llm.do_completion.return_value = {
|
|
'choices': [{'message': {'content': content}}]
|
|
}
|
|
|
|
coder_agent = Agent.get_cls('CoderAgent')(llm=mock_llm)
|
|
assert coder_agent is not None
|
|
|
|
task = 'This is a dummy task'
|
|
history = [(MessageAction(content=task), NullObservation(''))]
|
|
history[0][0]._source = EventSource.USER
|
|
summary = 'This is a dummy summary about this repo'
|
|
state = State(history=history, inputs={'summary': summary})
|
|
coder_agent.step(state)
|
|
|
|
mock_llm.do_completion.assert_called_once()
|
|
_, kwargs = mock_llm.do_completion.call_args
|
|
prompt = kwargs['messages'][0]['content']
|
|
assert task in prompt
|
|
assert "Here's a summary of the codebase, as it relates to this task" in prompt
|
|
assert summary in prompt
|
|
|
|
|
|
def test_coder_agent_without_summary():
|
|
"""
|
|
When there's no codebase_summary available, there shouldn't be any prompt
|
|
about 'code summary'
|
|
"""
|
|
mock_llm = MagicMock()
|
|
content = json.dumps({'action': 'finish', 'args': {}})
|
|
mock_llm.do_completion.return_value = {
|
|
'choices': [{'message': {'content': content}}]
|
|
}
|
|
|
|
coder_agent = Agent.get_cls('CoderAgent')(llm=mock_llm)
|
|
assert coder_agent is not None
|
|
|
|
task = 'This is a dummy task'
|
|
history = [(MessageAction(content=task), NullObservation(''))]
|
|
history[0][0]._source = EventSource.USER
|
|
state = State(history=history)
|
|
coder_agent.step(state)
|
|
|
|
mock_llm.do_completion.assert_called_once()
|
|
_, kwargs = mock_llm.do_completion.call_args
|
|
prompt = kwargs['messages'][0]['content']
|
|
assert task in prompt
|
|
assert "Here's a summary of the codebase, as it relates to this task" not in prompt
|