Spaces:
Sleeping
Sleeping
File size: 3,523 Bytes
ed1558c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
# coding: utf-8
# Copyright (c) 2025 inclusionAI.
import json
from examples.tools.tool_action import AndroidAction
from aworld.core.tool.action_factory import ActionFactory
from aworld.core.common import ActionModel, ActionResult
from examples.tools.android.action.adb_controller import ADBController
from examples.tools.android.config.android_action_space import AndroidActionParamEnum
from aworld.core.tool.action import ExecutableAction
@ActionFactory.register(name=AndroidAction.TAP.value.name,
desc=AndroidAction.TAP.value.desc,
tool_name="android")
class Tap(ExecutableAction):
def act(self, action: ActionModel, **kwargs) -> ActionResult:
controller: ADBController = kwargs.get('controller')
tap_index = action.params[AndroidActionParamEnum.TAP_INDEX.value]
if tap_index is None:
raise Exception(f'Invalid action: {action}')
controller.tap(tap_index)
return ActionResult(content="", keep=True)
@ActionFactory.register(name=AndroidAction.INPUT_TEXT.value.name,
desc=AndroidAction.INPUT_TEXT.value.desc,
tool_name="android")
class InputText(ExecutableAction):
def act(self, action: ActionModel, **kwargs) -> ActionResult:
controller: ADBController = kwargs.get('controller')
input_text = action.params[AndroidActionParamEnum.INPUT_TEXT.value]
if input_text is None:
raise Exception(f'Invalid action: {action}')
controller.text(input_text)
return ActionResult(content="", keep=True)
@ActionFactory.register(name=AndroidAction.LONG_PRESS.value.name,
desc=AndroidAction.LONG_PRESS.value.desc,
tool_name="android")
class LongPress(ExecutableAction):
def act(self, action: ActionModel, **kwargs) -> ActionResult:
controller: ADBController = kwargs.get('controller')
long_press_index = action.params[AndroidActionParamEnum.LONG_PRESS_INDEX.value]
if long_press_index is None:
raise Exception(f'Invalid action: {action}')
controller.long_press(long_press_index)
return ActionResult(content="", keep=True)
@ActionFactory.register(name=AndroidAction.SWIPE.value.name,
desc=AndroidAction.SWIPE.value.desc,
tool_name="android")
class Swipe(ExecutableAction):
def act(self, action: ActionModel, **kwargs) -> ActionResult:
controller: ADBController = kwargs.get('controller')
swipe_start_index = action.params[AndroidActionParamEnum.SWIPE_START_INDEX.value]
direction = action.params[AndroidActionParamEnum.DIRECTION.value]
dist = action.params.get(AndroidActionParamEnum.DIST.value, None)
if swipe_start_index is None or direction is None:
raise Exception(f'Invalid action: {action}')
if dist:
controller.swipe(swipe_start_index, direction, dist)
else:
controller.swipe(swipe_start_index, direction)
return ActionResult(content="", keep=True)
@ActionFactory.register(name=AndroidAction.DONE.value.name,
desc=AndroidAction.DONE.value.desc,
tool_name="android")
class Done(ExecutableAction):
def act(self, action: ActionModel, **kwargs) -> ActionResult:
output_dict = action.model_dump(exclude={'success'})
return ActionResult(is_done=True, success=True, content=json.dumps(output_dict))
|