Improve local agent chat and device health detection

2026-08-29 16:03:15 +08:00
parent 050d1329c4
commit a315c62f3a
11 changed files with 289 additions and 4 deletions
@@ -124,7 +124,7 @@ def default_tool_registry(
    from tools.tap import tap
    from tools.ui_tree import get_ui_tree

-    return {
+    registry = {
        "take_screenshot": _bind_manager(take_screenshot, manager),
        "screenshot": _bind_manager(take_screenshot, manager),
        "tap": _bind_manager(tap, manager),
@@ -143,6 +143,10 @@ def default_tool_registry(
        "find_icon": find_icon,
        "find_icon_on_screen": _bind_manager(find_icon_on_screen, manager),
    }
+    if manager is not None:
+        registry["list_devices"] = lambda: [device.to_dict() for device in manager.list_devices()]
+        registry["device_status"] = lambda device_id: {"device_id": device_id, "status": manager.status(device_id)}
+    return registry


 def _bind_manager(func: ToolCallable, manager: DeviceManager | None) -> ToolCallable: