mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-10 23:38:08 -05:00
415843476c
* add image feature * fix-linting * check model support for images * add comment * Add image support to other models * Add images to chat * fix linting * fix test issues * refactor variable names and import * fix tests * fix chat message tests * fix linting * add pydantic class message * use message * remove redundant comments * remove redundant comments * change Message class * remove unintended change * fix integration tests using regenerate.sh * rename image_bas64 to images_url, fix tests * rename Message.py to message, change reminder append logic, add unit tests * remove comment, fix error to merge * codeact_swe_agent * fix f string * update eventstream integration tests * add missing if check in codeact_swe_agent * update integration tests * Update frontend/src/components/chat/ChatInput.tsx * Update frontend/src/components/chat/ChatInput.tsx * Update frontend/src/components/chat/ChatInput.tsx * Update frontend/src/components/chat/ChatInput.tsx * Update frontend/src/components/chat/ChatMessage.tsx --------- Co-authored-by: tobitege <tobitege@gmx.de> Co-authored-by: Xingyao Wang <xingyao6@illinois.edu> Co-authored-by: sp.wack <83104063+amanape@users.noreply.github.com>
102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
import json
|
|
import os
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
import yaml
|
|
from pytest import TempPathFactory
|
|
|
|
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.stream import EventStream
|
|
from opendevin.memory.history import ShortTermHistory
|
|
from opendevin.storage import get_file_store
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_dir(tmp_path_factory: TempPathFactory) -> str:
|
|
return str(tmp_path_factory.mktemp('test_micro_agents'))
|
|
|
|
|
|
@pytest.fixture
|
|
def event_stream(temp_dir):
|
|
file_store = get_file_store('local', temp_dir)
|
|
event_stream = EventStream('asdf', file_store)
|
|
yield event_stream
|
|
|
|
# clear after each test
|
|
event_stream.clear()
|
|
|
|
|
|
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(event_stream: EventStream):
|
|
"""Coder agent should render code summary as part of prompt"""
|
|
mock_llm = MagicMock()
|
|
content = json.dumps({'action': 'finish', 'args': {}})
|
|
mock_llm.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 = ShortTermHistory()
|
|
history.set_event_stream(event_stream)
|
|
event_stream.add_event(MessageAction(content=task), EventSource.USER)
|
|
|
|
summary = 'This is a dummy summary about this repo'
|
|
state = State(history=history, inputs={'summary': summary})
|
|
coder_agent.step(state)
|
|
|
|
mock_llm.completion.assert_called_once()
|
|
_, kwargs = mock_llm.completion.call_args
|
|
prompt = kwargs['messages'][0]['content'][0]['text']
|
|
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(event_stream: EventStream):
|
|
"""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.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 = ShortTermHistory()
|
|
history.set_event_stream(event_stream)
|
|
event_stream.add_event(MessageAction(content=task), EventSource.USER)
|
|
|
|
# set state without codebase summary
|
|
state = State(history=history)
|
|
coder_agent.step(state)
|
|
|
|
mock_llm.completion.assert_called_once()
|
|
_, kwargs = mock_llm.completion.call_args
|
|
prompt = kwargs['messages'][0]['content'][0]['text']
|
|
assert task in prompt
|
|
assert "Here's a summary of the codebase, as it relates to this task" not in prompt
|