somew1nd
67248c887e
Add execution incidents, action bursts and shared Flash history. Expose verification and Explorer settings with streamed check results. Support graceful cancellation, recording recovery and bundled UI resources. Add CI quality checks and clean up comments and documentation.
279 lines
9.9 KiB
Python
279 lines
9.9 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.
|
|
|
|
"""Batch tasks execution command (artemis batch)."""
|
|
|
|
import asyncio
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Annotated
|
|
|
|
from artemis.config import initialize_llm_config
|
|
from artemis.sdk import Agent
|
|
from artemis.sdk.builders import Builders
|
|
from artemis.sdk.types.task import AgentProfile
|
|
from artemis.utils.logger import get_logger
|
|
from rich.console import Console
|
|
from rich.table import Table
|
|
import typer
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
async def run_batch_tasks(
|
|
tasks: list[str],
|
|
profile_name: str = "pro",
|
|
delay_seconds: float = 5.0,
|
|
verification_level: str | None = None,
|
|
explorer_pro_mode: str | None = None,
|
|
) -> None:
|
|
"""Executes a list of automation tasks sequentially.
|
|
|
|
Args:
|
|
tasks: Goals to run one after another on a single agent.
|
|
profile_name: Execution profile ('flash' or 'pro') applied to every goal.
|
|
delay_seconds: Pause between successive goals.
|
|
verification_level: Coarse Checker preset ('off', 'final', 'checkpoints',
|
|
'strict') for the Pro profile; ignored by Flash.
|
|
explorer_pro_mode: Explorer tier ('flash', 'pro', 'ultra') behind
|
|
``ask_explorer`` under the Pro profile; ignored by Flash.
|
|
"""
|
|
if not os.environ.get("ARTEMIS_TASK_INGRESS"):
|
|
os.environ["ARTEMIS_TASK_INGRESS"] = "cli"
|
|
llm_config = initialize_llm_config()
|
|
profile = AgentProfile(name="default", llm_config=llm_config)
|
|
config_builder = Builders.AgentConfig.with_default_profile(profile)
|
|
if verification_level is not None:
|
|
config_builder.with_verification_level(verification_level)
|
|
if explorer_pro_mode is not None:
|
|
config_builder.with_explorer(pro_mode=explorer_pro_mode)
|
|
config = config_builder.build()
|
|
|
|
agent = Agent(config=config)
|
|
await agent.init()
|
|
|
|
console = Console()
|
|
results = []
|
|
|
|
try:
|
|
for idx, goal in enumerate(tasks, start=1):
|
|
console.rule(f"[bold cyan]Task {idx}/{len(tasks)}: {goal}[/bold cyan]")
|
|
status = "SUCCESS"
|
|
error_msg = ""
|
|
try:
|
|
result = await agent.run_task(goal=goal, profile=profile_name)
|
|
logger.info(f"Task {idx} completed: {result}")
|
|
except Exception as e:
|
|
status = "FAILED"
|
|
error_msg = str(e)
|
|
logger.error(f"Task {idx} failed: {e}")
|
|
|
|
results.append({"task": goal, "status": status, "error": error_msg})
|
|
|
|
if idx < len(tasks) and delay_seconds > 0:
|
|
await asyncio.sleep(delay_seconds)
|
|
finally:
|
|
await agent.clean()
|
|
|
|
# Print summary table
|
|
table = Table(title="Batch Execution Summary")
|
|
table.add_column("#", justify="right", style="cyan")
|
|
table.add_column("Task Goal", style="white")
|
|
table.add_column("Status", justify="center")
|
|
table.add_column("Notes", style="red")
|
|
|
|
for i, res in enumerate(results, 1):
|
|
status_styled = (
|
|
"[bold green]PASS[/bold green]"
|
|
if res["status"] == "SUCCESS"
|
|
else "[bold red]FAIL[/bold red]"
|
|
)
|
|
table.add_row(str(i), res["task"], status_styled, res["error"])
|
|
|
|
console.print(table)
|
|
|
|
|
|
def batch_command(
|
|
tasks_file: Annotated[
|
|
Path | None,
|
|
typer.Option(
|
|
"--file",
|
|
"-f",
|
|
help="Path to a text or JSON file containing task goals (one per line or JSON array).",
|
|
),
|
|
] = None,
|
|
goals: Annotated[
|
|
list[str] | None,
|
|
typer.Argument(
|
|
help="Task goals specified directly on the command line.",
|
|
),
|
|
] = None,
|
|
profile: Annotated[
|
|
str,
|
|
typer.Option(
|
|
"--profile",
|
|
"-p",
|
|
help="Execution profile to use for all tasks ('flash' or 'pro').",
|
|
),
|
|
] = "pro",
|
|
delay: Annotated[
|
|
float,
|
|
typer.Option(
|
|
"--delay",
|
|
"-d",
|
|
help="Delay in seconds between successive tasks.",
|
|
),
|
|
] = 5.0,
|
|
standalone: Annotated[
|
|
bool,
|
|
typer.Option(
|
|
"--standalone",
|
|
help="Run in standalone embedded mode without routing through Artemis Daemon.",
|
|
),
|
|
] = False,
|
|
verification_level: Annotated[
|
|
str | None,
|
|
typer.Option(
|
|
"--verification-level",
|
|
help=(
|
|
"Checker preset for the Pro profile, applied to every goal: 'off' (no audit),"
|
|
" 'final' (exit review only, the default), 'checkpoints' (every plan"
|
|
" checkpoint + exit review), 'strict' (checkpoints with a larger repair"
|
|
" budget; a failed assert halts)."
|
|
),
|
|
),
|
|
] = None,
|
|
explorer_pro_mode: Annotated[
|
|
str | None,
|
|
typer.Option(
|
|
"--explorer-pro-mode",
|
|
help="Explorer tier behind ask_explorer under the Pro profile ('flash', 'pro', 'ultra').",
|
|
),
|
|
] = None,
|
|
) -> None:
|
|
"""Execute multiple automation tasks in sequence."""
|
|
task_list: list[str] = []
|
|
|
|
if tasks_file:
|
|
if not tasks_file.exists():
|
|
typer.secho(f"Error: Tasks file '{tasks_file}' not found.", fg=typer.colors.RED)
|
|
raise typer.Exit(1)
|
|
|
|
content = tasks_file.read_text(encoding="utf-8").strip()
|
|
if tasks_file.suffix.lower() == ".json":
|
|
try:
|
|
parsed = json.loads(content)
|
|
if isinstance(parsed, list):
|
|
task_list.extend(str(item) for item in parsed)
|
|
else:
|
|
typer.secho(
|
|
"Error: JSON file must contain a list of goals.", fg=typer.colors.RED
|
|
)
|
|
raise typer.Exit(1)
|
|
except Exception as e:
|
|
typer.secho(f"Error parsing JSON file: {e}", fg=typer.colors.RED)
|
|
raise typer.Exit(1)
|
|
else:
|
|
task_list.extend(
|
|
[
|
|
line.strip()
|
|
for line in content.splitlines()
|
|
if line.strip() and not line.startswith("#")
|
|
]
|
|
)
|
|
|
|
if goals:
|
|
task_list.extend(goals)
|
|
|
|
if not task_list:
|
|
typer.secho(
|
|
"Error: No tasks provided. Use --file or pass goals as arguments.", fg=typer.colors.RED
|
|
)
|
|
raise typer.Exit(1)
|
|
|
|
is_standalone = standalone or os.environ.get("ARTEMIS_STANDALONE") == "1"
|
|
if not is_standalone:
|
|
try:
|
|
from artemis.runtime import (
|
|
ensure_daemon_running,
|
|
submit_batch_to_daemon,
|
|
wait_for_daemon_task,
|
|
)
|
|
|
|
console = Console()
|
|
console.print("[dim]Connecting to Artemis Daemon scheduler...[/dim]")
|
|
is_running, base_url = ensure_daemon_running(timeout=8.0, wait_ready=True)
|
|
if is_running and base_url:
|
|
console.print(
|
|
f"[bold green]✓[/bold green] Artemis Daemon active at [cyan]{base_url}[/cyan]"
|
|
)
|
|
resp = submit_batch_to_daemon(
|
|
task_list,
|
|
profile=profile,
|
|
verification_level=verification_level,
|
|
explorer_mode=explorer_pro_mode,
|
|
base_url=base_url,
|
|
)
|
|
if resp and resp.get("tasks"):
|
|
console.print(
|
|
f"[bold green]✓[/bold green] Enqueued {len(task_list)} batch tasks in Daemon scheduler.\n"
|
|
)
|
|
results = []
|
|
for idx, t_item in enumerate(resp["tasks"], 1):
|
|
sid = t_item.get("session_id")
|
|
goal = t_item.get("goal")
|
|
console.print(
|
|
f"[cyan]Waiting for Task {idx}/{len(task_list)}: '{goal}' (Session: {sid})...[/cyan]"
|
|
)
|
|
final_sess = wait_for_daemon_task(sid, base_url=base_url, timeout=1800.0)
|
|
st = final_sess.get("status")
|
|
err = final_sess.get("error") or ""
|
|
results.append(
|
|
{
|
|
"task": goal,
|
|
"status": "SUCCESS" if st in ("completed", "success") else "FAILED",
|
|
"error": err,
|
|
}
|
|
)
|
|
|
|
table = Table(title="Batch Execution Summary")
|
|
table.add_column("#", justify="right", style="cyan")
|
|
table.add_column("Task Goal", style="white")
|
|
table.add_column("Status", justify="center")
|
|
table.add_column("Notes", style="red")
|
|
|
|
for i, res in enumerate(results, 1):
|
|
status_styled = (
|
|
"[bold green]PASS[/bold green]"
|
|
if res["status"] == "SUCCESS"
|
|
else "[bold red]FAIL[/bold red]"
|
|
)
|
|
table.add_row(str(i), res["task"], status_styled, res["error"])
|
|
|
|
console.print(table)
|
|
return
|
|
except Exception as exc:
|
|
logger.debug(f"Batch daemon routing notice: {exc}")
|
|
|
|
asyncio.run(
|
|
run_batch_tasks(
|
|
task_list,
|
|
profile_name=profile,
|
|
delay_seconds=delay,
|
|
verification_level=verification_level,
|
|
explorer_pro_mode=explorer_pro_mode,
|
|
)
|
|
)
|