mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-10 07:18:10 -05:00
39a851bd95
* Add to_memory for observations * Don't send 'message' property to the llm from observation; fix other agents
21 lines
1.2 KiB
Python
21 lines
1.2 KiB
Python
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()
|
|
serialized_observation_memory = observation_instance.to_memory()
|
|
assert serialized_observation_dict == original_observation_dict, 'The serialized observation should match the original observation dict.'
|
|
original_observation_dict.pop('message')
|
|
assert serialized_observation_memory == original_observation_dict, 'The serialized observation in memory should match the original observation dict.'
|
|
|
|
# Additional tests for various observation subclasses can be included here
|