feat: checkpoint device agent runtime milestones
This commit is contained in:
+3
-1
@@ -5,8 +5,8 @@ from dataclasses import dataclass, field
|
||||
from time import sleep
|
||||
from typing import Any
|
||||
|
||||
from core.device_manager import DeviceManager
|
||||
from core.errors import ElementNotFoundError
|
||||
from device.manager import DeviceManager
|
||||
from runtime.context import TaskContext
|
||||
from runtime.planner import PlannedStep
|
||||
|
||||
@@ -110,6 +110,7 @@ def default_tool_registry(
|
||||
manager: DeviceManager | None = None,
|
||||
) -> dict[str, ToolCallable]:
|
||||
from tools.describe_screen import describe_screen
|
||||
from tools.describe_screen_semantic import describe_screen_semantic
|
||||
from tools.find_icon import find_icon, find_icon_on_screen
|
||||
from tools.find_text import find_text, find_text_on_screen
|
||||
from tools.input_text import input_text
|
||||
@@ -130,6 +131,7 @@ def default_tool_registry(
|
||||
"get_ui_tree": _bind_manager(get_ui_tree, manager),
|
||||
"ui_tree": _bind_manager(get_ui_tree, manager),
|
||||
"describe_screen": _bind_manager(describe_screen, manager),
|
||||
"describe_screen_semantic": _bind_manager(describe_screen_semantic, manager),
|
||||
"find_text": find_text,
|
||||
"find_text_on_screen": _bind_manager(find_text_on_screen, manager),
|
||||
"find_icon": find_icon,
|
||||
|
||||
Reference in New Issue
Block a user