42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from typing import List
|
||
from agent.device import controller
|
||
from agent.tools.base_tool import BaseTool, ToolResult
|
||
|
||
|
||
|
||
class Tool(BaseTool):
|
||
name = "uav_control"
|
||
description = (
|
||
"该工具用于对无人系统进行控制并在飞行控制中完成一系列的活动（Action），例如起飞、移动、悬停、照相等"
|
||
)
|
||
parameters = {
|
||
"type": "object",
|
||
"properties": {
|
||
"command": {
|
||
"type": "string",
|
||
"description": "无人系统飞行控制命令，例如arm、disarm、takeoff、rtl、goto等",
|
||
},
|
||
"params": {
|
||
"type": "object",
|
||
"description": "控制命令参数"
|
||
},
|
||
"actions": {
|
||
"type": "array",
|
||
"description": "过程中执行的一些列的动作，例如pause、capture_phone、standby等"
|
||
}
|
||
},
|
||
"required": ["command"],
|
||
}
|
||
|
||
|
||
def execute(self, command: str, params: dict={}, actions: List[object]=()) -> ToolResult:
|
||
if not controller.connect():
|
||
return ToolResult(success=False, output="连接无人机失败")
|
||
try:
|
||
command_handler = getattr(controller, command)
|
||
try:
|
||
return ToolResult(success=True, output=str(command_handler(**params)))
|
||
except Exception as e:
|
||
return ToolResult(success=False, output=str(e))
|
||
finally:
|
||
controller.disconnect() |