mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
from opendevin.action import (
|
|
Action,
|
|
AddTaskAction,
|
|
AgentFinishAction,
|
|
AgentRecallAction,
|
|
AgentThinkAction,
|
|
BrowseURLAction,
|
|
CmdKillAction,
|
|
CmdRunAction,
|
|
FileReadAction,
|
|
FileWriteAction,
|
|
GitHubPushAction,
|
|
ModifyTaskAction,
|
|
action_from_dict,
|
|
)
|
|
|
|
|
|
def serialization_deserialization(original_action_dict, cls):
|
|
action_instance = action_from_dict(original_action_dict)
|
|
assert isinstance(
|
|
action_instance, Action), 'The action instance should be an instance of Action.'
|
|
assert isinstance(
|
|
action_instance, cls), f'The action instance should be an instance of {cls.__name__}.'
|
|
serialized_action_dict = action_instance.to_dict()
|
|
serialized_action_memory = action_instance.to_memory()
|
|
serialized_action_dict.pop('message')
|
|
assert serialized_action_dict == original_action_dict, 'The serialized action should match the original action dict.'
|
|
assert serialized_action_memory == original_action_dict, 'The serialized action in memory should match the original action dict.'
|
|
|
|
|
|
def test_agent_think_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'think',
|
|
'args': {'thought': 'This is a test.'}
|
|
}
|
|
serialization_deserialization(original_action_dict, AgentThinkAction)
|
|
|
|
|
|
def test_agent_recall_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'recall',
|
|
'args': {'query': 'Test query.'}
|
|
}
|
|
serialization_deserialization(original_action_dict, AgentRecallAction)
|
|
|
|
|
|
def test_agent_finish_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'finish',
|
|
'args': {'outputs': {}},
|
|
}
|
|
serialization_deserialization(original_action_dict, AgentFinishAction)
|
|
|
|
|
|
def test_cmd_kill_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'kill',
|
|
'args': {'id': '1337'}
|
|
}
|
|
serialization_deserialization(original_action_dict, CmdKillAction)
|
|
|
|
|
|
def test_cmd_run_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'run',
|
|
'args': {'command': 'echo "Hello world"', 'background': True}
|
|
}
|
|
serialization_deserialization(original_action_dict, CmdRunAction)
|
|
|
|
|
|
def test_browse_url_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'browse',
|
|
'args': {'url': 'https://www.example.com'}
|
|
}
|
|
serialization_deserialization(original_action_dict, BrowseURLAction)
|
|
|
|
|
|
def test_github_push_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'push',
|
|
'args': {'owner': 'myname', 'repo': 'myrepo', 'branch': 'main'}
|
|
}
|
|
serialization_deserialization(original_action_dict, GitHubPushAction)
|
|
|
|
|
|
def test_file_read_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'read',
|
|
'args': {'path': '/path/to/file.txt', 'start': 0, 'end': -1, 'thoughts': 'None'}
|
|
}
|
|
serialization_deserialization(original_action_dict, FileReadAction)
|
|
|
|
|
|
def test_file_write_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'write',
|
|
'args': {'path': '/path/to/file.txt', 'content': 'Hello world', 'start': 0, 'end': 1, 'thoughts': 'None'}
|
|
}
|
|
serialization_deserialization(original_action_dict, FileWriteAction)
|
|
|
|
|
|
def test_add_task_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'add_task',
|
|
'args': {'parent': 'Test parent', 'goal': 'Test goal', 'subtasks': []}
|
|
}
|
|
serialization_deserialization(original_action_dict, AddTaskAction)
|
|
|
|
|
|
def test_modify_task_action_serialization_deserialization():
|
|
original_action_dict = {
|
|
'action': 'modify_task',
|
|
'args': {'id': 1, 'state': 'Test state.'}
|
|
}
|
|
serialization_deserialization(original_action_dict, ModifyTaskAction)
|