android-midscene-automation 0.2.4 → 1.1.0

package/artemis-main/artemis/tools/committee_prompts.json

@@ -0,0 +1,5 @@
+{
+  "planner_avatar": "You are the **Planner Avatar** (Meeting Host) in a committee resolving a task plan failure. Your role is to lead the meeting and advocate for a robust and correct high-level plan. You are aware of:\n- **History Analyzer**: The trajectory specialist who can analyze execution history and read notes. They have tools like `search_history`, `replay_steps`, `get_step_screenshot` and `read_note`.\n- **Diagnoser**: The forensics specialist who can analyze logs and videos to find why things failed. They have tools like `video_analyzer` and `analyze_logs`.\nYour master (Master Planner) has given you a specific directive for this meeting, which is at the top of the blackboard.\n**Your Task**:\n1. In the first round, you must open the meeting, state the problem based on the master's directive, and call on the other experts (Diagnoser or History Analyzer) to provide their analysis.\n2. In subsequent rounds, evaluate their suggestions and push for a concrete, revised plan of subgoals.\n3. In the final round, summarize the discussion and output the final recommended plan revisions and actions.",
+  "history_analyzer_expert": "You are the **History Analyzer** (Trajectory Specialist) in a committee resolving a task failure. Your role is to analyze the execution history of the current session, understand what has been tried, what succeeded, and what failed, and check persistent notes. You are aware of:\n- **Planner Avatar**: The meeting host who needs a concrete plan.\n- **Diagnoser**: The forensics specialist who analyzes logs and videos.\nYou have tools to search the execution history (`search_history`), replay steps in full (`replay_steps`), look at a step's stored screenshot (`get_step_screenshot`) and read notes. Wait for the Planner Avatar to call on you or respond to the current state of the blackboard. Be concise.",
+  "diagnoser_expert": "You are the **Diagnoser** (Forensics Specialist) in a committee resolving a UI automation failure. Your role is to analyze what went wrong in the past, look for error patterns in logs, and understand system state transitions. You are aware of:\n- **Planner Avatar**: The meeting host who needs a concrete plan.\n- **History Analyzer**: The trajectory specialist who analyzes history and notes.\nYou have tools to analyze video and logs. Wait for the Planner Avatar to call on you or respond to the current state of the blackboard. Be concise."
+}

package/artemis-main/artemis/tools/committee_tool.py

@@ -0,0 +1,422 @@
+# pylint: disable=too-many-statements
+# 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.
+
+"""Committee tool for multi-agent deliberation and failure recovery."""
+
+import asyncio
+import base64
+import json
+from pathlib import Path
+from typing import Any
+from uuid import uuid4
+
+from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
+
+from artemis.core.tool_failure import ToolFailure, is_tool_failure
+from artemis.context import ArtemisContext
+from artemis.data_engine.trace import (
+    CURRENT_TRACE_ID,
+    TraceSpan,
+    trace_langchain_tool,
+)
+from artemis.drivers.base import BaseDeviceDriver
+from artemis.graph.state import State
+from artemis.services.llm import get_llm, invoke_llm_with_timeout_message
+from artemis.tools.base import ArtemisTool, ToolCategory
+from artemis.tools.history import get_history_tools
+from artemis.tools.log_tool import get_analyze_logs_tool
+from artemis.tools.scratchpad import get_list_notes_tool, get_read_note_tool
+from artemis.tools.tool_wrapper import (
+    ToolWrapper,
+    get_tool_result_content,
+    invoke_tool_with_injection,
+    tool_result_messages,
+)
+from artemis.utils.cython_compat import CyFunctionDetector
+from artemis.tools.video_tool import get_video_analyzer_tool
+from artemis.utils.notes import get_note_file_path, get_notes_dir
+from artemis.memory.context_policy import build_history_for
+from artemis.utils.task_tree import (
+    get_active_subgoal_hashes,
+    get_recent_subgoal_hashes,
+)
+from third_party.mobile_use.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+class AskCommitteeArgs(BaseModel):
+    """Arguments schema for invoking the committee tool."""
+
+    model_config = {"ignored_types": (CyFunctionDetector,)}
+    avatar_directive: str = Field(
+        ...,
+        description="Directive to frame and steer the debate for your avatar.",
+    )
+
+
+ASK_COMMITTEE_DOCSTRING = (
+    "[COMMITTEE] Call this tool to summon a committee of sub-agents to debate"
+    " and resolve complex failures.\n\n"
+    "Your avatar will host the meeting and drive it based on your directive.\n"
+    "The result will be a synthesized recommendation."
+)
+
+
+class AskCommitteeTool(ArtemisTool):
+    """Universal tool for summoning a committee of sub-agents to debate and resolve failures."""
+
+    def __init__(self, category: ToolCategory = "custom"):
+        super().__init__(
+            name="ask_committee",
+            description=ASK_COMMITTEE_DOCSTRING,
+            args_schema=AskCommitteeArgs,
+            category=category,
+        )
+
+    # 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,
+        avatar_directive: str = "",
+        tool_call_id: str | None = None,  # pylint: disable=unused-argument
+        state: State | None = None,
+        **kwargs: Any,
+    ) -> str:
+        directive = avatar_directive or kwargs.get("AvatarDirective") or ""
+        return await _run_committee_logic(
+            ctx=ctx,
+            state=state,
+            avatar_directive=directive,
+        )
+
+
+# Universal tool instance & aliases
+ask_committee = AskCommitteeTool()
+AskCommittee = AskCommitteeTool
+AskCommitteeToolAlias = AskCommitteeTool
+
+
+def get_ask_committee_tool(ctx: ArtemisContext) -> BaseTool:
+    """Exports ask_committee as a LangChain BaseTool."""
+    return trace_langchain_tool(ask_committee.to_langchain_tool(ctx), ctx)
+
+
+# pylint: disable=too-many-branches,too-many-locals,too-many-statements
+async def _execute_committee(
+    ctx: ArtemisContext,
+    state: State | None,
+    avatar_directive: str,
+) -> str:
+    """Executes committee debate loop and returns the final conclusion."""
+    with TraceSpan(name="committee", trace_type="agent", ctx=ctx):
+        logger.info(f"Committee invoked with directive: {avatar_directive}")
+
+        # 1. Access or create blackboard file
+        if ctx.data_engine:
+            notes_dir = get_notes_dir(ctx.data_engine.base_dir)
+        else:
+            notes_dir = Path(".") / "notes"
+        notes_dir.mkdir(parents=True, exist_ok=True)
+
+        # Load latest screenshot as base64 to provide multimodal vision to the committee
+        screenshot_b64 = None
+        screenshot_path = getattr(state, "latest_screenshot", None) if state is not None else None
+        if screenshot_path and Path(screenshot_path).exists():
+            try:
+                with open(screenshot_path, "rb") as f:
+                    screenshot_b64 = base64.b64encode(f.read()).decode("utf-8")
+                logger.info(f"Committee successfully loaded screenshot from: {screenshot_path}")
+            except Exception as e:  # pylint: disable=broad-exception-caught
+                logger.error(f"Failed to read screenshot in committee: {e}")
+
+        trace_id = CURRENT_TRACE_ID.get() or str(uuid4())
+        blackboard_path = notes_dir / f"blackboard_{trace_id}.md"
+
+        # Initialize blackboard with goal and directive
+        initial_goal = (
+            getattr(state, "initial_goal", "Not specified.")
+            if state is not None
+            else "Not specified."
+        )
+        blackboard_content = (
+            "# Committee Discussion Blackboard\n\n"
+            f"## User's Initial Goal\n{initial_goal}\n\n"
+            f"## Master's Directive\n{avatar_directive}\n\n"
+        )
+        blackboard_path.write_text(blackboard_content, encoding="utf-8")
+
+        # 2. Extract history
+        history_summary = "## Previous Context\n"
+        if ctx.data_engine:
+            steps = ctx.data_engine.get_agent_friendly_steps()
+            if steps:
+                active_subgoal_hash = "default"
+                if ctx.data_engine:
+                    task_plan_path = get_note_file_path(ctx.data_engine.base_dir, "task_plan")
+                else:
+                    task_plan_path = notes_dir / "task_plan.md"
+                task_plan = "No task plan yet."
+                keep_hashes = None
+                if task_plan_path.exists():
+                    try:
+                        task_plan = task_plan_path.read_text(encoding="utf-8")
+
+                        active_subgoal_hash, _ = get_active_subgoal_hashes(task_plan)
+                        keep_hashes = get_recent_subgoal_hashes(
+                            steps,
+                            active_subgoal_hash,
+                            ctx.data_engine.base_dir,
+                        )
+                    except Exception as e:  # pylint: disable=broad-exception-caught
+                        logger.error(f"Failed to parse active subgoal in committee: {e}")
+
+                try:
+                    plan_and_history = build_history_for(
+                        "committee",
+                        task_plan,
+                        steps,
+                        active_subgoal_hash,
+                        keep_subgoal_hashes=keep_hashes,
+                        engine=ctx.data_engine,
+                    )
+                    history_summary += plan_and_history
+                except Exception as e:  # pylint: disable=broad-exception-caught
+                    logger.error(f"Failed to build plan and history in committee: {e}")
+                    history_summary += f"Error building plan and history: {e}"
+
+                # Read failed plans history to ensure the committee avoids
+                # repeating failed strategies
+                failed_plans_history = ""
+                failed_history_path = notes_dir / "failed_plans_history.md"
+                if failed_history_path.exists():
+                    try:
+                        failed_plans_history = failed_history_path.read_text(encoding="utf-8")
+                    except Exception as e:  # pylint: disable=broad-exception-caught
+                        logger.error(f"Failed to read failed plans history in committee: {e}")
+                if failed_plans_history:
+                    history_summary += (
+                        f"\n\n### Previous Failed Plan Attempts\n{failed_plans_history}\n"
+                    )
+
+        with open(blackboard_path, "a", encoding="utf-8") as f:
+            f.write(history_summary + "\n\n")
+
+        # 3. Load prompts from JSON
+        prompts = {}
+        prompts_path = Path(__file__).parent / "committee_prompts.json"
+        if prompts_path.exists():
+            try:
+                prompts = json.loads(prompts_path.read_text(encoding="utf-8"))
+            except Exception as e:  # pylint: disable=broad-exception-caught
+                logger.error(f"Failed to load committee prompts: {e}")
+
+        planner_prompt = prompts.get("planner_avatar", "You are the Planner Avatar.")
+        history_analyzer_prompt = prompts.get(
+            "history_analyzer_expert", "You are the History Analyzer."
+        )
+        diagnoser_prompt = prompts.get("diagnoser_expert", "You are the Diagnoser Expert.")
+
+        # 4. Prepare Specialized Tools for Members
+
+        # History Analyzer Tools: the shared history tools plus the note readers
+        read_note_tool_hist = get_read_note_tool(ctx)
+        list_notes_tool_hist = get_list_notes_tool(ctx)
+
+        history_analyzer_tools = [
+            *get_history_tools(ctx),
+            trace_langchain_tool(read_note_tool_hist, ctx),
+            trace_langchain_tool(list_notes_tool_hist, ctx),
+        ]
+
+        # Diagnoser Tools
+        video_tool = get_video_analyzer_tool(ctx, role="diagnoser")
+        log_tool = get_analyze_logs_tool(ctx)
+        read_note_tool = get_read_note_tool(ctx)
+        list_notes_tool = get_list_notes_tool(ctx)
+
+        diagnoser_tools = [
+            trace_langchain_tool(video_tool, ctx),
+            trace_langchain_tool(log_tool, ctx),
+            trace_langchain_tool(read_note_tool, ctx),
+            trace_langchain_tool(list_notes_tool, ctx),
+        ]
+
+        # 5. Instantiate LLMs
+        llm_pl = get_llm(ctx=ctx, name="planner_avatar")
+
+        llm_hist = get_llm(ctx=ctx, name="history_analyzer_expert")
+        llm_hist = llm_hist.bind_tools(tools=history_analyzer_tools)
+
+        llm_diag = get_llm(ctx=ctx, name="diagnoser_expert")
+        llm_diag = llm_diag.bind_tools(tools=diagnoser_tools)
+
+        # 6. Debate Loop (configured rounds)
+        exec_setup = getattr(ctx, "execution_setup", None)
+        rounds = getattr(exec_setup, "committee_debate_rounds", 2) if exec_setup else 2
+
+        async def run_agent_turn(llm, system_prompt, tools, agent_name):
+            current_blackboard = blackboard_path.read_text(encoding="utf-8")
+
+            human_content = [
+                {
+                    "type": "text",
+                    "text": f"Current Blackboard:\n{current_blackboard}",
+                }
+            ]
+            if screenshot_b64:
+                human_content.append(
+                    {
+                        "type": "image_url",
+                        "image_url": {"url": f"data:image/jpeg;base64,{screenshot_b64}"},
+                    }
+                )
+
+            messages = [
+                SystemMessage(content=system_prompt),
+                HumanMessage(content=human_content),
+            ]
+
+            max_iterations = 3
+            for _ in range(max_iterations):
+                response = await invoke_llm_with_timeout_message(llm.ainvoke(messages))
+
+                if not response.tool_calls:
+                    return response.content
+
+                messages.append(response)
+
+                async def run_tool(tc):
+                    """Keep the raw result so the caller can read ToolMessage.status."""
+                    tool_name = tc["name"]
+                    logger.info(f"{agent_name} requested tool: {tool_name}")
+                    tool_to_run = next((t for t in tools if t.name == tool_name), None)
+                    if tool_to_run:
+                        try:
+                            args = dict(tc["args"])
+                            return await invoke_tool_with_injection(
+                                tool=tool_to_run,
+                                args=args,
+                                tool_call_id=tc["id"],
+                                state=state,
+                            )
+                        except Exception as e:  # pylint: disable=broad-exception-caught
+                            return ToolFailure(f"Error: {e}")
+                    else:
+                        return ToolFailure(f"Error: Tool {tool_name} not found")
+
+                active_tool_calls = [
+                    tc for tc in response.tool_calls if tc["name"] != "google_search"
+                ]
+                tool_outputs = await asyncio.gather(*(run_tool(tc) for tc in active_tool_calls))
+
+                for tc, raw_result in zip(active_tool_calls, tool_outputs):
+                    # Read status before unwrapping ToolMessage.content.
+                    status = "error" if is_tool_failure(raw_result) else "success"
+                    messages.extend(
+                        tool_result_messages(
+                            tc["id"],
+                            get_tool_result_content(raw_result),
+                            name=tc["name"],
+                            status=status,
+                            llm=llm,
+                        )
+                    )
+            return ToolFailure("Error: Reached max iterations in turn.")
+
+        for r in range(1, rounds + 1):
+            logger.info(f"Committee Round {r}")
+
+            # 1. Planner Avatar speaks first (Host)
+            pl_response = await run_agent_turn(llm_pl, planner_prompt, [], "Planner Avatar")
+            with open(blackboard_path, "a", encoding="utf-8") as f:
+                f.write(f"### Planner Avatar (Round {r})\n{pl_response}\n\n")
+
+            # 2. Diagnoser speaks (Can use tools)
+            diag_response = await run_agent_turn(
+                llm_diag, diagnoser_prompt, diagnoser_tools, "Diagnoser"
+            )
+            with open(blackboard_path, "a", encoding="utf-8") as f:
+                f.write(f"### Diagnoser (Round {r})\n{diag_response}\n\n")
+
+            # 3. History Analyzer speaks (Can use tools)
+            hist_response = await run_agent_turn(
+                llm_hist,
+                history_analyzer_prompt,
+                history_analyzer_tools,
+                "History Analyzer",
+            )
+            with open(blackboard_path, "a", encoding="utf-8") as f:
+                f.write(f"### History Analyzer (Round {r})\n{hist_response}\n\n")
+
+        # Final Synthesis by Planner Avatar
+        current_blackboard = blackboard_path.read_text(encoding="utf-8")
+
+        final_human_content = [
+            {
+                "type": "text",
+                "text": f"Current Blackboard:\n{current_blackboard}",
+            }
+        ]
+        if screenshot_b64:
+            final_human_content.append(
+                {
+                    "type": "image_url",
+                    "image_url": {"url": f"data:image/jpeg;base64,{screenshot_b64}"},
+                }
+            )
+
+        msg_final = [
+            SystemMessage(
+                content=(
+                    "You are the Planner Avatar. Summarize the discussion"
+                    " and output the final recommended plan revisions and"
+                    " action plan based on your master's directive."
+                )
+            ),
+            HumanMessage(content=final_human_content),
+        ]
+        resp_final = await invoke_llm_with_timeout_message(llm_pl.ainvoke(msg_final))
+
+        with open(blackboard_path, "a", encoding="utf-8") as f:
+            f.write(f"## Final Conclusion\n{resp_final.content}\n")
+
+        return resp_final.content
+
+
+async def _run_committee_logic(
+    ctx: ArtemisContext | None,
+    state: State | None,
+    avatar_directive: str,
+) -> str:
+    """Executes multi-agent committee debate logic."""
+    if ctx is None:
+        return ToolFailure("Error: ArtemisContext is required for ask_committee.")
+
+    try:
+        return await asyncio.wait_for(_execute_committee(ctx, state, avatar_directive), timeout=300)
+    except TimeoutError:
+        logger.error("Committee timed out after 300 seconds.")
+        return ToolFailure("Error: Committee timed out after 300 seconds.")
+
+
+ask_committee_wrapper = ToolWrapper(
+    tool_fn_getter=get_ask_committee_tool,
+    on_success_fn=lambda output: f"Committee concluded:\n{output}",
+    on_failure_fn=lambda error: f"Committee failed: {error}",
+)

package/artemis-main/artemis/tools/diagnoser_submit_answer_tool.py

@@ -0,0 +1,87 @@
+# 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
+from third_party.mobile_use.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
+
+
+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)

package/artemis-main/artemis/tools/diagnostic_tool.py

@@ -0,0 +1,120 @@
+# 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 diagnostic agent tool for deep system logs and video inspection."""
+
+from typing import Any
+
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
+
+from artemis.core.tool_failure import ToolFailure
+from artemis.agents.diagnoser.diagnoser import Diagnoser
+from artemis.context import ArtemisContext
+from artemis.data_engine.trace import trace_langchain_tool
+from artemis.drivers.base import BaseDeviceDriver
+from artemis.graph.state import State
+from artemis.tools.base import ArtemisTool
+from artemis.tools.tool_wrapper import ToolWrapper
+from artemis.utils.cython_compat import CyFunctionDetector
+from third_party.mobile_use.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+# pylint: disable=too-few-public-methods
+class AskDiagnoserArgs(BaseModel):
+    """Arguments schema for invoking the diagnostic subagent."""
+
+    model_config = {"ignored_types": (CyFunctionDetector,)}
+    query: str = Field(
+        ...,
+        description=(
+            "The diagnostic query or what you want to investigate. Please"
+            " provide a specific, narrow time range if applicable."
+        ),
+    )
+
+
+async def _run_diagnoser_logic(
+    ctx: ArtemisContext | None,
+    state: State | None,
+    query: str,
+) -> str:
+    logger.info(f"ask_diagnoser tool called with query: {query}")
+    if ctx is None:
+        return ToolFailure("Error: ArtemisContext is required for Diagnoser.")
+    agent = Diagnoser(ctx)
+    try:
+        result = await agent.run(query, state)
+        return f"Diagnoser replied:\n{result}"
+    except Exception as e:  # pylint: disable=broad-exception-caught
+        logger.error(f"Failed to run diagnoser: {e}")
+        return f"Diagnoser failed: {e}"
+
+
+class AskDiagnoserTool(ArtemisTool):
+    """Universal tool for invoking the Diagnoser subagent to inspect logs and videos."""
+
+    def __init__(self):
+        super().__init__(
+            name="ask_diagnoser",
+            description=(
+                "[DIAGNOSTIC] Analyzes system logs (ADB) and video recordings to "
+                "pinpoint the root cause of a step failure, screen freeze, or "
+                "unexpected UI drift. Use it when the live screen does not explain "
+                "why an action is not taking effect or why progress is stalled, "
+                "before committing to further physical actions. Note: When querying "
+                "video files, specify a narrow timeframe (e.g., '[10s, 20s]') for "
+                "faster analysis."
+            ),
+            args_schema=AskDiagnoserArgs,
+            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,
+        query: str | None = None,
+        state: State | None = None,
+        **kwargs: Any,
+    ) -> str:
+        q = query or kwargs.get("Query") or ""
+        st = state if state is not None else kwargs.get("state")
+        return await _run_diagnoser_logic(ctx, st, q)
+
+
+# Universal tool instance & aliases
+ask_diagnoser = AskDiagnoserTool()
+AskDiagnoser = AskDiagnoserTool
+DiagnosticTool = AskDiagnoserTool
+
+
+def get_ask_diagnostic_agent_tool(ctx: ArtemisContext) -> BaseTool:
+    """Exports ask_diagnoser as a LangChain BaseTool."""
+    return trace_langchain_tool(ask_diagnoser.to_langchain_tool(ctx), ctx)
+
+
+def get_ask_diagnoser_tool(ctx: ArtemisContext) -> BaseTool:
+    """Exports ask_diagnoser as a LangChain BaseTool (standard naming alias)."""
+    return trace_langchain_tool(ask_diagnoser.to_langchain_tool(ctx), ctx)
+
+
+ask_diagnoser_wrapper = ToolWrapper(
+    tool_fn_getter=get_ask_diagnostic_agent_tool,
+    on_success_fn=lambda output: f"Diagnoser replied:\n{output}",
+    on_failure_fn=lambda error: f"Diagnoser failed: {error}",
+)