Files
Artemis-Eapil/artemis/tools/diagnoser_submit_answer_tool.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

89 lines
3.1 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.
"""Universal tool for submitting final diagnosis and actionable recovery advice."""
from typing import Any
from langchain_core.tools import BaseTool
from pydantic import BaseModel, Field
from artemis.context import ArtemisContext
from artemis.data_engine.trace import trace_langchain_tool
from artemis.drivers.base import BaseDeviceDriver
from artemis.tools.base import ArtemisTool, ToolRegistry
from artemis.utils.logger import get_logger
logger = get_logger(__name__)
# pylint: disable=too-few-public-methods
class SubmitAnswerArgs(BaseModel):
"""Arguments schema for submitting final diagnoser answer."""
analysis: str = Field(
...,
description=(
"A detailed and concise explanation of the root cause of the"
" failure or query. Answer in one full paragraph."
),
)
actionable_steps: list[str] = Field(
default_factory=list,
description=(
"A list of concrete recovery instructions or next steps. Each"
" step is a short sentence. If no actionable steps are"
" available, return an empty list."
),
)
class SubmitAnswerTool(ArtemisTool):
"""Universal tool for submitting final diagnosis and actionable recovery advice."""
def __init__(self):
super().__init__(
name="submit_answer",
description=(
"[Finalization] Submits the final diagnosis and actionable recovery advice. "
"Call this tool when you have pinpointed the root cause."
),
args_schema=SubmitAnswerArgs,
category="custom",
)
# pylint: disable=too-many-arguments,too-many-positional-arguments
async def execute(
self,
driver: BaseDeviceDriver | None = None, # pylint: disable=unused-argument
ctx: ArtemisContext | None = None, # pylint: disable=unused-argument
analysis: str | None = None, # pylint: disable=unused-argument
actionable_steps: list[str] | None = None, # pylint: disable=unused-argument
**kwargs: Any,
) -> str:
logger.info("Diagnoser submitted answer via tool.")
return "Answer submitted successfully."
# Universal tool instance & aliases
submit_answer = SubmitAnswerTool()
SubmitAnswer = SubmitAnswerTool
DiagnoserSubmitAnswerTool = SubmitAnswerTool
ToolRegistry.register(submit_answer)
def get_submit_answer_tool(ctx: ArtemisContext) -> BaseTool:
"""Exports submit_answer as a LangChain BaseTool."""
return trace_langchain_tool(submit_answer.to_langchain_tool(ctx), ctx)