Files
Artemis-Eapil/artemis/engine/pipeline.py
T
somew1nd 206a9ce921 feat: initial release of ARTEMIS Android Agent & Automation Platform
ARTEMIS is an autonomous multimodal Android agent and mobile UI automation
platform engineered for physical devices and emulators. It combines state-of-the-art
multimodal Vision-Language Models (VLMs) with hierarchical closed-loop planning,
resilient self-healing, and deep Android subsystem integration.

Key Features & Architectural Highlights:
- Dual-Engine Orchestration:
  * Pro Engine: Multi-agent closed-loop (Planner -> Operator -> Validator/Failure Analyzer -> Summarizer) powered by LangGraph.
  * Flash Engine: Ultra-fast reactive loop for deterministic, low-latency UI execution (<35 steps).
- Multimodal Perception & Grounding:
  * Hybrid visual screenshot grounding, real-time OCR extraction, and XML accessibility hierarchy parsing.
  * Dynamic-First, Coordinate-Fallback locating strategy for maximum resilience against UI drift.
- Autonomous Self-Healing & Diagnostics:
  * Failure analyzer and safety net for popup dismissal, crash recovery, and auto-retry.
  * Diagnostic sub-agents for log analysis, video recording inspection, and object detection.
- Unified CLI & Python SDK:
  * Rich CLI with commands: init, doctor, run, trace, batch, bench, server, and mcp.
  * Fluent Python SDK with Pydantic structured output models and package locking.
- Multi-Provider LLM Architecture:
  * Support for Google Gemini, OpenAI, Anthropic Claude, Vertex AI, OpenRouter, xAI Grok, and local endpoints (Ollama/vLLM).
  * Centralized configuration in config/artemis.jsonc.
- Native MCP & Observability:
  * Model Context Protocol (MCP) server endpoints for AI IDEs.
  * Full trace recording with before/after screenshots, action overlays, and execution replays.

Change-Id: I430b93ee2b76b54cc5a3c6eeccd15d2cf29830a1
2026-08-13 11:46:47 -07:00

92 lines
3.3 KiB
Python

# 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.
from typing import Literal
from artemis.config import OutputConfig, load_agent_config
from artemis.core.context import ExecutionContext
from artemis.core.state import ExecutionContextState
from artemis.drivers.base import BaseDeviceDriver
from artemis.drivers.mock.mock_driver import MockDeviceDriver
from artemis.engine.base_runner import BaseRunner
from artemis.engine.graph_runner import GraphRunner
from artemis.engine.reactive_runner import ReactiveRunner
from artemis.utils.logger import get_logger
logger = get_logger(__name__)
class Pipeline:
"""Orchestrates runner selection and execution flow."""
@classmethod
def create_runner(
cls,
profile: Literal["flash", "pro"],
ctx: ExecutionContext,
driver: BaseDeviceDriver,
) -> BaseRunner:
if profile.lower() == "flash":
return ReactiveRunner(ctx=ctx, driver=driver)
return GraphRunner(ctx=ctx, driver=driver)
@classmethod
async def execute(
cls,
goal: str,
profile: Literal["flash", "pro"] = "pro",
driver: BaseDeviceDriver | None = None,
device_id: str = "default-device",
output_config: OutputConfig | None = None,
enable_outputter: bool | None = None,
) -> ExecutionContextState:
ctx = ExecutionContext(task_goal=goal, device_id=device_id)
if not driver:
driver = MockDeviceDriver(device_id=device_id)
runner = cls.create_runner(profile=profile, ctx=ctx, driver=driver)
state = await runner.run()
# Check Outputter configuration
try:
agent_cfg = load_agent_config()
outputter_cfg = getattr(agent_cfg, "outputter", None)
is_enabled = (
enable_outputter
if enable_outputter is not None
else (outputter_cfg.enabled if outputter_cfg else True)
)
force_synth = outputter_cfg.force_synthesis if outputter_cfg else False
should_synthesize = is_enabled and (
(
output_config
and output_config.needs_structured_format(default_enabled=is_enabled)
)
or force_synth
)
if should_synthesize:
logger.info("Executing post-execution Outputter synthesis in Pipeline...")
summary = "\n".join(
f"- Step {s.step_number}: {s.action_name}({s.action_params}) -> {s.result or s.thought or 'completed'}"
for s in state.steps
)
state.metadata["structured_output"] = summary
state.final_output = summary
except Exception as e:
logger.warning(f"Pipeline outputter post-processing skipped: {e}")
return state