mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
da548d308c
Co-authored-by: Tim O'Farrell <tofarr@gmail.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com> Co-authored-by: Robert Brennan <accounts@rbren.io> Co-authored-by: Graham Neubig <neubig@gmail.com>
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
from openhands.core.exceptions import LLMMalformedActionError
|
|
from openhands.events.action.action import Action
|
|
from openhands.events.action.agent import (
|
|
AgentDelegateAction,
|
|
AgentFinishAction,
|
|
AgentRejectAction,
|
|
ChangeAgentStateAction,
|
|
)
|
|
from openhands.events.action.browse import BrowseInteractiveAction, BrowseURLAction
|
|
from openhands.events.action.commands import (
|
|
CmdRunAction,
|
|
IPythonRunCellAction,
|
|
)
|
|
from openhands.events.action.empty import NullAction
|
|
from openhands.events.action.files import (
|
|
FileEditAction,
|
|
FileReadAction,
|
|
FileWriteAction,
|
|
)
|
|
from openhands.events.action.message import MessageAction
|
|
from openhands.events.action.tasks import AddTaskAction, ModifyTaskAction
|
|
|
|
actions = (
|
|
NullAction,
|
|
CmdRunAction,
|
|
IPythonRunCellAction,
|
|
BrowseURLAction,
|
|
BrowseInteractiveAction,
|
|
FileReadAction,
|
|
FileWriteAction,
|
|
FileEditAction,
|
|
AgentFinishAction,
|
|
AgentRejectAction,
|
|
AgentDelegateAction,
|
|
AddTaskAction,
|
|
ModifyTaskAction,
|
|
ChangeAgentStateAction,
|
|
MessageAction,
|
|
)
|
|
|
|
ACTION_TYPE_TO_CLASS = {action_class.action: action_class for action_class in actions} # type: ignore[attr-defined]
|
|
|
|
|
|
def action_from_dict(action: dict) -> Action:
|
|
if not isinstance(action, dict):
|
|
raise LLMMalformedActionError('action must be a dictionary')
|
|
action = action.copy()
|
|
if 'action' not in action:
|
|
raise LLMMalformedActionError(f"'action' key is not found in {action=}")
|
|
if not isinstance(action['action'], str):
|
|
raise LLMMalformedActionError(
|
|
f"'{action['action']=}' is not defined. Available actions: {ACTION_TYPE_TO_CLASS.keys()}"
|
|
)
|
|
action_class = ACTION_TYPE_TO_CLASS.get(action['action'])
|
|
if action_class is None:
|
|
raise LLMMalformedActionError(
|
|
f"'{action['action']=}' is not defined. Available actions: {ACTION_TYPE_TO_CLASS.keys()}"
|
|
)
|
|
args = action.get('args', {})
|
|
# Remove timestamp from args if present
|
|
timestamp = args.pop('timestamp', None)
|
|
|
|
try:
|
|
decoded_action = action_class(**args)
|
|
if 'timeout' in action:
|
|
decoded_action.timeout = action['timeout']
|
|
|
|
# Set timestamp if it was provided
|
|
if timestamp:
|
|
decoded_action._timestamp = timestamp
|
|
|
|
except TypeError as e:
|
|
raise LLMMalformedActionError(
|
|
f'action={action} has the wrong arguments: {str(e)}'
|
|
)
|
|
return decoded_action
|