mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
0eccf31604
* Refactor monologue to use the messages in state history * add messages, clean up * fix monologue * update integration tests * move private method * update SWE agent to use the history from State * integration tests for SWE agent * rename monologue to initial_thoughts, since that is what it is
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import json
|
|
|
|
import pytest
|
|
|
|
from opendevin.events import EventSource, EventStream
|
|
from opendevin.events.action import NullAction
|
|
from opendevin.events.observation import NullObservation
|
|
|
|
|
|
def collect_events(stream):
|
|
return [event for event in stream.get_events()]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_basic_flow():
|
|
stream = EventStream('abc')
|
|
await stream.add_event(NullAction(), EventSource.AGENT)
|
|
assert len(collect_events(stream)) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stream_storage():
|
|
stream = EventStream('def')
|
|
await stream.add_event(NullObservation(''), EventSource.AGENT)
|
|
assert len(collect_events(stream)) == 1
|
|
content = stream._file_store.read('sessions/def/events/0.json')
|
|
assert content is not None
|
|
data = json.loads(content)
|
|
assert 'timestamp' in data
|
|
del data['timestamp']
|
|
assert data == {
|
|
'id': 0,
|
|
'source': 'agent',
|
|
'observation': 'null',
|
|
'content': '',
|
|
'extras': {},
|
|
'message': 'No observation',
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_rehydration():
|
|
stream1 = EventStream('es1')
|
|
await stream1.add_event(NullObservation('obs1'), EventSource.AGENT)
|
|
await stream1.add_event(NullObservation('obs2'), EventSource.AGENT)
|
|
assert len(collect_events(stream1)) == 2
|
|
|
|
stream2 = EventStream('es2')
|
|
assert len(collect_events(stream2)) == 0
|
|
|
|
stream1rehydrated = EventStream('es1')
|
|
events = collect_events(stream1rehydrated)
|
|
assert len(events) == 2
|
|
assert events[0].content == 'obs1'
|
|
assert events[1].content == 'obs2'
|