Revert "refactor: clean up legacy code, rename hopper to entity_extractor, and add third-party notices"
This reverts commit 54fcef9887.
Please do not submit unrelated docs and codes in the same commit. I am reverting this commit so that the related files can be submitted in better-organized future commits.
This commit is contained in:
yaoyao-open-source committed
1 parent
54fcef9887
commit
1d82d59ec8
83 files changed
+3677
-2013
No files matched your search
@@ -1,63 +0,0 @@
|
||||
# Copyright 2026 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
"""Unit tests validating Artemis SDK exception hierarchy and diagnostics."""
|
||||
|
||||
from artemis.sdk.types.exceptions import (
|
||||
AgentError,
|
||||
AgentNotInitializedError,
|
||||
AgentProfileNotFoundError,
|
||||
AgentTaskRequestError,
|
||||
ArtemisError,
|
||||
DeviceError,
|
||||
DeviceNotFoundError,
|
||||
ExecutableNotFoundError,
|
||||
ServerError,
|
||||
ServerStartupError,
|
||||
)
|
||||
|
||||
|
||||
def test_exception_hierarchy():
|
||||
"""Verify all custom exceptions inherit properly from ArtemisError."""
|
||||
assert issubclass(DeviceError, ArtemisError)
|
||||
assert issubclass(DeviceNotFoundError, DeviceError)
|
||||
assert issubclass(ServerError, ArtemisError)
|
||||
assert issubclass(ServerStartupError, ServerError)
|
||||
assert issubclass(AgentError, ArtemisError)
|
||||
assert issubclass(AgentNotInitializedError, AgentError)
|
||||
assert issubclass(AgentTaskRequestError, AgentError)
|
||||
assert issubclass(AgentProfileNotFoundError, AgentTaskRequestError)
|
||||
assert issubclass(ExecutableNotFoundError, ArtemisError)
|
||||
|
||||
|
||||
def test_executable_not_found_error_adb_guidance():
|
||||
"""Verify executable error formats download instructions for known tools."""
|
||||
err = ExecutableNotFoundError("adb")
|
||||
assert "adb" in str(err)
|
||||
assert "developer.android.com" in str(err)
|
||||
assert err.executable_name == "adb"
|
||||
|
||||
|
||||
def test_server_startup_error_formatting():
|
||||
"""Verify server startup error encapsulates service identifier."""
|
||||
err = ServerStartupError(server_name="mcp_daemon")
|
||||
assert "mcp_daemon" in str(err)
|
||||
assert err.server_name == "mcp_daemon"
|
||||
|
||||
|
||||
def test_agent_profile_not_found_error():
|
||||
"""Verify profile error retains profile name."""
|
||||
err = AgentProfileNotFoundError("nonexistent_profile")
|
||||
assert "nonexistent_profile" in str(err)
|
||||
assert err.profile_name == "nonexistent_profile"
|
||||
@@ -1,111 +0,0 @@
|
||||
# Copyright 2026 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
"""Unit tests for Artemis SDK TaskRequestBuilder."""
|
||||
|
||||
from pathlib import Path
|
||||
from pydantic import BaseModel
|
||||
import pytest
|
||||
|
||||
from artemis.sdk.builders.task_request_builder import TaskRequestBuilder, TaskRequestCommonBuilder
|
||||
from artemis.sdk.types.task import TaskRequest
|
||||
|
||||
|
||||
class DummyOutputSchema(BaseModel):
|
||||
summary: str
|
||||
item_count: int
|
||||
|
||||
|
||||
def test_task_request_builder_fluent_chaining():
|
||||
"""Verify basic fluent configuration and build."""
|
||||
req = (
|
||||
TaskRequestBuilder(goal="Open Settings and verify battery level")
|
||||
.with_max_steps(25)
|
||||
.with_trace_recording(True, path="custom_traces")
|
||||
.with_locked_app_package("com.android.settings")
|
||||
.with_app_path("build/app.apk")
|
||||
.using_profile("flash")
|
||||
.with_name("Battery Check")
|
||||
.build()
|
||||
)
|
||||
|
||||
assert isinstance(req, TaskRequest)
|
||||
assert req.goal == "Open Settings and verify battery level"
|
||||
assert req.max_steps == 25
|
||||
assert req.record_trace is True
|
||||
assert req.trace_path == Path("custom_traces")
|
||||
assert req.locked_app_package == "com.android.settings"
|
||||
assert req.app_path == Path("build/app.apk")
|
||||
assert req.profile == "flash"
|
||||
assert req.task_name == "Battery Check"
|
||||
|
||||
|
||||
def test_task_request_builder_with_output_format():
|
||||
"""Verify builder binds structured schema properly."""
|
||||
req = (
|
||||
TaskRequestBuilder(goal="Extract items")
|
||||
.with_output_format(DummyOutputSchema)
|
||||
.build()
|
||||
)
|
||||
|
||||
assert req.output_format is DummyOutputSchema
|
||||
assert req.output_description is None
|
||||
|
||||
|
||||
def test_task_request_builder_with_output_description():
|
||||
"""Verify builder accepts unformatted string deliverables."""
|
||||
req = (
|
||||
TaskRequestBuilder(goal="Extract items")
|
||||
.with_output_description("Return a numbered list")
|
||||
.build()
|
||||
)
|
||||
|
||||
assert req.output_description == "Return a numbered list"
|
||||
assert req.output_format is None
|
||||
|
||||
|
||||
def test_task_request_builder_conflicting_outputs_raises():
|
||||
"""Verify that specifying both output_format and output_description raises ValueError."""
|
||||
builder = (
|
||||
TaskRequestBuilder(goal="Extract items")
|
||||
.with_output_format(DummyOutputSchema)
|
||||
.with_output_description("Description text")
|
||||
)
|
||||
with pytest.raises(ValueError, match="Cannot specify both output_format and output_description"):
|
||||
builder.build()
|
||||
|
||||
|
||||
def test_task_request_builder_empty_goal_raises():
|
||||
"""Verify empty or whitespace-only goal raises ValueError."""
|
||||
with pytest.raises(ValueError, match="Task goal must be a non-empty"):
|
||||
TaskRequestBuilder(goal=" ").build()
|
||||
|
||||
|
||||
def test_task_request_builder_from_common():
|
||||
"""Verify task request initialization derived from common base settings."""
|
||||
common_settings = (
|
||||
TaskRequestCommonBuilder()
|
||||
.with_max_steps(42)
|
||||
.with_trace_recording(False)
|
||||
.with_locked_app_package("org.artemis.target")
|
||||
.with_app_path(Path("/tmp/target.apk"))
|
||||
.build()
|
||||
)
|
||||
|
||||
derived = TaskRequestBuilder.from_common("Derived goal", common_settings).build()
|
||||
assert derived.goal == "Derived goal"
|
||||
assert derived.max_steps == 42
|
||||
assert derived.record_trace is False
|
||||
assert derived.locked_app_package == "org.artemis.target"
|
||||
assert derived.app_path == Path("/tmp/target.apk")
|
||||
Reference in new issue
Block a user