android-midscene-automation 0.2.4 → 1.1.1

package/artemis-main/packages/artemis-client/tests/test_transport.py

@@ -0,0 +1,85 @@
+# 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
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+
+from __future__ import annotations
+
+import io
+import json
+import unittest
+import urllib.error
+from email.message import Message
+from unittest.mock import patch
+
+from artemis_client import AuthenticationError, JsonTransport, NetworkError, ProtocolError
+
+
+class FakeResponse:
+    def __init__(self, payload: bytes) -> None:
+        self.payload = payload
+
+    def __enter__(self) -> FakeResponse:
+        return self
+
+    def __exit__(self, *args: object) -> None:
+        return None
+
+    def read(self) -> bytes:
+        return self.payload
+
+
+class JsonTransportTests(unittest.TestCase):
+    def test_rejects_non_http_base_url(self) -> None:
+        with self.assertRaises(ValueError):
+            JsonTransport("file:///tmp/artemis")
+
+    @patch("urllib.request.urlopen")
+    def test_sends_json_and_bearer_token(self, urlopen) -> None:
+        urlopen.return_value = FakeResponse(json.dumps({"ok": True}).encode())
+        transport = JsonTransport("https://host.example/", token="secret")
+
+        result = transport.request("POST", "/api/run", json_body={"goal": "test"})
+
+        self.assertEqual(result, {"ok": True})
+        request = urlopen.call_args.args[0]
+        self.assertEqual(request.full_url, "https://host.example/api/run")
+        self.assertEqual(request.headers["Authorization"], "Bearer secret")
+        self.assertEqual(json.loads(request.data), {"goal": "test"})
+
+    @patch("urllib.request.urlopen")
+    def test_maps_unauthorized_response(self, urlopen) -> None:
+        urlopen.side_effect = urllib.error.HTTPError(
+            "https://host.example/api/run",
+            401,
+            "Unauthorized",
+            Message(),
+            io.BytesIO(b'{"detail":"bad token"}'),
+        )
+        transport = JsonTransport("https://host.example")
+
+        with self.assertRaisesRegex(AuthenticationError, "bad token"):
+            transport.request("GET", "/api/run")
+
+    @patch("urllib.request.urlopen")
+    def test_invalid_json_raises_protocol_error(self, urlopen) -> None:
+        urlopen.return_value = FakeResponse(b"not-json")
+        transport = JsonTransport("https://host.example")
+
+        with self.assertRaises(ProtocolError):
+            transport.request("GET", "/api/status")
+
+    @patch("urllib.request.urlopen")
+    def test_network_failure_is_wrapped(self, urlopen) -> None:
+        urlopen.side_effect = urllib.error.URLError("connection refused")
+        transport = JsonTransport("https://host.example")
+
+        with self.assertRaisesRegex(NetworkError, "connection refused"):
+            transport.request("GET", "/api/status")
+
+
+if __name__ == "__main__":
+    unittest.main()

package/artemis-main/pyproject.toml

@@ -0,0 +1,259 @@
+# 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.
+
+[project]
+name = "artemis"
+version = "1.0"
+description = "Autonomous AI Assistant for Mobile Automation"
+
+authors = [
+    { name = "Farley Wang", email="farleyw@google.com" },
+]
+
+requires-python = ">=3.12"
+
+dependencies = [
+    # Remote Python SDK (kept as a zero-dependency workspace package)
+    "artemis-client>=0.1.0",
+    # Orchestration & Framework
+    "langgraph>=1.0.2,<2.0.0",
+    "langchain>=1.0.0",
+    "langchain-core>=1.0.0",
+    "langchain-community>=0.4.1",
+    "langchain-mcp-adapters>=0.2.0",
+    "mcp>=1.26.0",
+    # LLM Providers & GenAI SDKs
+    "google-genai>=2.22.0",
+    "langchain-google-genai>=4.0.0",
+    "langchain-google-vertexai>=3.0.0",
+    "langchain-openai>=1.0.0",
+    # Configuration, Models & Validation
+    "pydantic>=2.7.0",
+    "pydantic-settings>=2.10.1",
+    "python-dotenv>=1.0.1",
+    "jinja2>=3.1.5",
+    "typing-extensions>=4.12.0",
+    # Android Automation & Device Subsystem
+    "adbutils>=2.12.0",
+    "uiautomator2>=3.5.0",
+    # CLI & Rich Terminal UI
+    "typer>=0.12.0",
+    "rich>=13.7.0",
+    "inquirer>=3.4.0",
+    "colorama>=0.4.6",
+    # Web, API & Networking
+    "fastapi>=0.111.0",
+    "uvicorn[standard]>=0.30.1",
+    "httpx>=0.28.1",
+    "requests>=2.31.0",
+    "websockets>=12.0",
+    "sseclient-py>=1.8.0",
+    # Image, Video & Multimodal Perception
+    "pillow>=10.0.0",
+    "opencv-python>=4.9.0.80",
+    "imageio-ffmpeg>=0.5.1",
+    "matplotlib>=3.8.0",
+    "numpy>=1.26.0",
+    "scipy>=1.13.0",
+    # System, Utilities & Telemetry
+    "psutil>=5.9.0",
+    "retry>=0.9.2",
+    "posthog>=3.5.0",
+    "uuid-utils>=0.9.0",
+    "jupyter-client>=8.6.0",
+    "ipykernel>=6.29.0",
+    "langchain-anthropic>=1.5.4",
+]
+
+[project.optional-dependencies]
+dev = [
+    "ruff>=0.15.2",
+    "pytest>=8.4.1",
+    "pytest-asyncio>=0.23.0",
+    "pytest-cov>=5.0.0",
+    "syrupy>=4.0.0",
+    "pyright>=1.1.389",
+    "pre-commit>=3.5.0",
+]
+
+[project.scripts]
+artemis = "artemis.interfaces.cli.main:cli"
+artemis-admin = "apps.admin_console.server:main"
+
+
+[build-system]
+# requires = ["uv_build>=0.7.20,<0.8.0"]
+# build-backend = "uv_build"
+requires = ["setuptools>=61.0", "Cython"]
+build-backend = "setuptools.build_meta"
+
+[tool.uv.workspace]
+members = ["packages/artemis-client"]
+
+[tool.uv.sources]
+artemis-client = { workspace = true }
+
+[tool.ruff]
+# Exclude a variety of commonly ignored directories.
+exclude = [
+    ".bzr",
+    ".direnv",
+    ".eggs",
+    ".git",
+    ".git-rewrite",
+    ".hg",
+    ".ipynb_checkpoints",
+    ".mypy_cache",
+    ".nox",
+    ".pants.d",
+    ".pyenv",
+    ".pytest_cache",
+    ".pytype",
+    ".ruff_cache",
+    ".svn",
+    ".tox",
+    ".venv",
+    ".vscode",
+    "__pypackages__",
+    "_build",
+    "buck-out",
+    "build",
+    "dist",
+    "node_modules",
+    "site-packages",
+    "venv",
+]
+
+# Same as Black.
+line-length = 100
+indent-width = 4
+
+target-version = "py312"
+
+[tool.ruff.lint]
+# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`) codes by default.
+# Also enable TID (flake8-tidy-imports) to enforce absolute imports
+select = ["E", "F", "TID", "UP"]
+ignore = [
+    "E501", # Line too long
+    "E402", # Module level import not at top of file
+    "E712", # Avoid equality comparisons to True/False
+    "F401", # Unused imports
+    "F841", # Local variable is assigned to but never used
+    "UP035", # Deprecated import
+    "UP036", # Outdated version block
+    "UP042", # Class inherits from both str and enum.Enum
+    "UP046", # Generic class uses Generic subclass instead of type parameters
+    "UP047", # Generic function should use type parameters
+]
+
+# Allow fix for all enabled rules (when `--fix`) is provided.
+fixable = ["ALL"]
+unfixable = []
+
+# Allow unused variables when underscore-prefixed.
+dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
+
+[tool.ruff.lint.flake8-tidy-imports]
+# Ban relative imports
+ban-relative-imports = "all"
+
+[tool.ruff.format]
+# Like Black, use double quotes for strings.
+quote-style = "double"
+
+# Like Black, indent with spaces, rather than tabs.
+indent-style = "space"
+
+# Like Black, respect magic trailing commas.
+skip-magic-trailing-comma = false
+
+# Like Black, automatically detect the appropriate line ending.
+line-ending = "auto"
+
+[tool.pytest.ini_options]
+# Add project root and apps to Python path for clean absolute imports
+pythonpath = [
+    ".",
+    "apps",
+    "apps/admin_console",
+]
+
+
+# Specify test discovery patterns
+testpaths = [
+    "tests/unit",
+    "tests/tools",
+    "packages/artemis-client/tests",
+]
+python_files = [
+    "*.test.py",
+    "test_*.py",
+    "*_test.py"
+]
+python_functions = [
+    "test_*"
+]
+python_classes = [
+    "Test*"
+]
+# Add useful options
+addopts = [
+    "--tb=short",           # shorter traceback format
+    "--strict-config",      # reject misspelled or stale pytest configuration
+    "--strict-markers",     # strict marker checking
+    "-m",                   # keep the default suite hermetic
+    "not integration and not e2e and not cloud and not manual and not android",
+]
+
+markers = [
+    "android: tests requiring Android device or emulator",
+    "integration: tests spanning multiple Artemis components",
+    "e2e: end-to-end tests requiring a running Artemis environment",
+    "cloud: tests requiring the optional cloud_service package",
+    "manual: tests that must only be launched explicitly by an operator",
+]
+
+[tool.pylint.master]
+ignore = [".venv", "build", "dist"]
+recursive = "yes"
+
+[tool.pylint.messages_control]
+disable = [
+    "C0114", # missing-module-docstring
+    "R0801", # duplicate-code
+]
+
+[tool.pyright]
+exclude = [".venv/**"]
+
+[tool.coverage.run]
+branch = true
+relative_files = true
+
+[tool.coverage.report]
+fail_under = 60
+show_missing = true
+skip_covered = true
+
+[dependency-groups]
+dev = [
+    "ruff>=0.15.2",
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.2.0",
+    "pytest-cov>=5.0.0",
+    "pyright>=1.1.389",
+    "pre-commit>=3.5.0",
+]
+

package/artemis-main/pyright-core.json

@@ -0,0 +1,17 @@
+{
+  "venvPath": ".",
+  "venv": ".venv",
+  "include": [
+    "artemis/graph/state.py",
+    "artemis/graph/visibility.py",
+    "artemis/runtime/adb_endpoint.py",
+    "artemis/runtime/process_probe.py",
+    "artemis/mcp/action_types.py",
+    "artemis/data_engine/models.py",
+    "artemis/agents/explorer/native_runner.py",
+    "artemis/agents/explorer/perception_tools.py",
+    "mcp_server/utils/device_utils.py",
+    "packages/artemis-client/src/artemis_client"
+  ],
+  "exclude": [".venv/**"]
+}

package/artemis-main/scripts/check_dependency_sources.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+# 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.
+
+"""Reject private dependency sources before installing project dependencies.
+
+Uses only the Python 3.12+ standard library so CI can run it before uv sync.
+"""
+
+import json
+import sys
+import tomllib
+from pathlib import Path
+from urllib.parse import urlsplit
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+def main() -> int:
+    errors: list[str] = []
+
+    def check_url(label: str, url: str, host: str) -> None:
+        parsed = urlsplit(url)
+        if parsed.scheme != "https" or parsed.netloc != host:
+            errors.append(f"{label}: expected an HTTPS URL on {host}")
+
+    lock = tomllib.loads((ROOT / "uv.lock").read_text(encoding="utf-8"))
+    workspace_sources = {
+        "artemis": {"editable": "."},
+        "artemis-client": {"editable": "packages/artemis-client"},
+    }
+    for package in lock["package"]:
+        label = f"uv.lock: {package['name']}=={package['version']}"
+        source = package["source"]
+        if source != workspace_sources.get(package["name"]):
+            if source != {"registry": "https://pypi.org/simple"}:
+                errors.append(f"{label}: expected the public PyPI registry")
+        artifacts = list(package.get("wheels", []))
+        if "sdist" in package:
+            artifacts.append(package["sdist"])
+        for artifact in artifacts:
+            check_url(f"{label}: artifact", artifact.get("url", ""), "files.pythonhosted.org")
+
+    npm_lock = json.loads((ROOT / "apps/showcase_ui/package-lock.json").read_text(encoding="utf-8"))
+    for name, package in npm_lock["packages"].items():
+        if "resolved" in package:
+            check_url(f"package-lock.json: {name}", package["resolved"], "registry.npmjs.org")
+
+    if errors:
+        print("Dependency source check failed:\n" + "\n".join(errors), file=sys.stderr)
+        return 1
+    print("Dependency sources use public PyPI and npm registries.")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())