diff --git a/tools/describe_screen_semantic.py b/runtime/describe_screen_semantic.py similarity index 100% rename from tools/describe_screen_semantic.py rename to runtime/describe_screen_semantic.py diff --git a/runtime/executor.py b/runtime/executor.py index 4a785c8..94d31a2 100644 --- a/runtime/executor.py +++ b/runtime/executor.py @@ -109,8 +109,8 @@ def default_tool_registry( *, manager: DeviceManager | None = None, ) -> dict[str, ToolCallable]: + from runtime.describe_screen_semantic import describe_screen_semantic 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 diff --git a/tests/test_describe_screen_semantic.py b/tests/test_describe_screen_semantic.py index 578a5d9..c1a4c3a 100644 --- a/tests/test_describe_screen_semantic.py +++ b/tests/test_describe_screen_semantic.py @@ -6,8 +6,8 @@ from core.models import Bounds, Scene, SceneElement from device.manager import DeviceManager from semantic.config import SemanticConfig from semantic.llm_client import EnrichmentUnavailable +from runtime.describe_screen_semantic import describe_screen_semantic from tools.describe_screen import describe_screen -from tools.describe_screen_semantic import describe_screen_semantic from tests.fakes import FakeDriver diff --git a/tests/test_semantic_task_loop.py b/tests/test_semantic_task_loop.py index afe93ae..c311d27 100644 --- a/tests/test_semantic_task_loop.py +++ b/tests/test_semantic_task_loop.py @@ -6,12 +6,12 @@ import pytest from core.models import Bounds, Scene, SceneElement, Task from device.manager import DeviceManager +from runtime.describe_screen_semantic import describe_screen_semantic from runtime.executor import Executor, ExecutorConfig from runtime.planner import PlannedStep, Planner from runtime.task import TaskRunner, TaskRunnerConfig from semantic.config import SemanticConfig from semantic.llm_client import EnrichmentUnavailable -from tools.describe_screen_semantic import describe_screen_semantic from tests.fakes import FakeDriver, PNG_10X20