mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
effac868c1
* action deserializing * add observation deserialization * add tests * refactor agents with serialization * fix some errors * fix lint * fix json parser
18 lines
1012 B
Python
18 lines
1012 B
Python
import pytest
|
|
from opendevin.observation import observation_from_dict, Observation, CmdOutputObservation
|
|
|
|
def test_observation_serialization_deserialization():
|
|
original_observation_dict = {
|
|
'observation': 'run',
|
|
'extras': {'exit_code': 0, 'command': 'ls -l', 'command_id': 3},
|
|
'message': 'Command `ls -l` executed with exit code 0.',
|
|
'content': 'foo.txt',
|
|
}
|
|
observation_instance = observation_from_dict(original_observation_dict)
|
|
assert isinstance(observation_instance, Observation), 'The observation instance should be an instance of Action.'
|
|
assert isinstance(observation_instance, CmdOutputObservation), 'The observation instance should be an instance of AgentThinkAction.'
|
|
serialized_observation_dict = observation_instance.to_dict()
|
|
assert serialized_observation_dict == original_observation_dict, 'The serialized observation should match the original observation dict.'
|
|
|
|
# Additional tests for various observation subclasses can be included here
|