Axe

A CLI tool for managing and running LLM-powered agents.

Why Axe?

Most AI tooling assumes you want a chatbot. A long-running session with a massive context window doing everything at once. But that’s not how good software works. Good software is small, focused, and composable.

Axe treats LLM agents the same way Unix treats programs. Each agent does one thing well. You define it in a TOML file, give it a focused skill, and run it from the command line. Pipe data in, get results out. Chain agents together. Trigger them from cron, git hooks, or CI. Whatever you already use. No daemon, no GUI, no framework to buy into. Just a binary and your configs.

Overview

Axe orchestrates LLM-powered agents defined via TOML configuration files. Each agent has its own system prompt, model selection, skill files, context files, working directory, persistent memory, and the ability to delegate to sub-agents.

Axe is the executor, not the scheduler. It is designed to be composed with standard Unix tools — cron, git hooks, pipes, file watchers — rather than reinventing scheduling or workflow orchestration.

Features

• Multi-provider support — Anthropic, OpenAI, Ollama (local models), OpenCode, and AWS Bedrock
• TOML-based agent configuration — declarative, version-controllable agent definitions
• Sub-agent delegation — agents can call other agents via LLM tool use, with depth limiting and parallel execution
• Persistent memory — timestamped markdown logs that carry context across runs
• Memory garbage collection — LLM-assisted pattern analysis and trimming
• Skill system — reusable instruction sets that can be shared across agents
• Stdin piping — pipe any output directly into an agent (git diff | axe run reviewer)
• Local agent directories — auto-discovers agents from /axe/agents/ before the global config, or use --agents-dir to point anywhere
• Dry-run mode — inspect resolved context without calling the LLM
• JSON output — structured output with metadata for scripting
• Built-in tools — file operations (read, write, edit, list) sandboxed to working directory; shell command execution; URL fetching; web search
• Output allowlist — restrict url_fetch and web_search to specific hostnames; private/reserved IPs are always blocked (SSRF protection)
• Token budgets — cap cumulative token usage per agent run via [budget] config or --max-tokens flag
• MCP tool support — connect to external MCP servers for additional tools via SSE or streamable-HTTP transport
• Configurable retry — exponential, linear, or fixed backoff for transient provider errors (429, 5xx, timeouts)
• Minimal dependencies — four direct dependencies (cobra, toml, mcp-go-sdk, x/net); all LLM calls use the standard library

Installation

Requires Go 1.25+.

Pre-built binaries (no Go required) are available for Linux, macOS, and Windows on the GitHub Releases page.

Install via Go:

go install github.com/jrswab/axe@latest

If this fails with invalid go version, your Go toolchain is older than 1.25. Upgrade from go.dev/dl or download a pre-built binary instead.

Or build from source:

git clone https://github.com/jrswab/axe.git
cd axe
go build .

Quick Start

Initialize the configuration directory:

axe config init

This creates the directory structure at $XDG_CONFIG_HOME/axe/ with a sample skill and a default config.toml for provider credentials.

Scaffold a new agent:

axe agents init my-agent

Edit its configuration:

axe agents edit my-agent

Run the agent:

axe run my-agent

Pipe input from other tools:

git diff --cached | axe run pr-reviewer
cat error.log | axe run log-analyzer

Examples

The examples/ directory contains ready-to-run agents you can copy into your config and use immediately. Includes a code reviewer, commit message generator, and text summarizer — each with a focused SKILL.md.

# Copy an example agent into your config
cp examples/code-reviewer/code-reviewer.toml "$(axe config path)/agents/"
cp -r examples/code-reviewer/skills/ "$(axe config path)/skills/"

# Set your API key and run
export ANTHROPIC_API_KEY="your-key-here"
git diff | axe run code-reviewer

See examples/README.md for full setup instructions.

Docker

Axe provides a Docker image for running agents in an isolated, hardened container.

Build the Image

docker build -t axe .

Multi-architecture builds (linux/amd64, linux/arm64) are supported via buildx:

docker buildx build --platform linux/amd64,linux/arm64 -t axe:latest .

Run an Agent

Mount your config directory and pass API keys as environment variables:

docker run --rm \
  -v ./my-config:/home/axe/.config/axe \
  -e ANTHROPIC_API_KEY \
  axe run my-agent

Pipe stdin with the -i flag:

git diff | docker run --rm -i \
  -v ./my-config:/home/axe/.config/axe \
  -e ANTHROPIC_API_KEY \
  axe run pr-reviewer

Without a config volume mounted, axe exits with code 2 (config error) because no agent TOML files exist.

Running a Single Agent

The examples above mount the entire config directory. If you only need to run one agent with one skill, mount just those files to their expected XDG paths inside the container. No config.toml is needed when API keys are passed via environment variables.

docker run --rm -i \
  -e ANTHROPIC_API_KEY \
  -v ./agents/reviewer.toml:/home/axe/.config/axe/agents/reviewer.toml:ro \
  -v ./skills/code-review/:/home/axe/.config/axe/skills/code-review/:ro \
  axe run reviewer

The agent’s skill field resolves automatically against the XDG config path inside the container, so no --skill flag is needed.

To use a different skill than the one declared in the agent’s TOML, use the --skill flag to override it. In this case you only mount the replacement skill — the original skill declared in the TOML is ignored entirely:

docker run --rm -i \
  -e ANTHROPIC_API_KEY \
  -v ./agents/reviewer.toml:/home/axe/.config/axe/agents/reviewer.toml:ro \
  -v ./alt-review.md:/home/axe/alt-review.md:ro \
  axe run reviewer --skill /home/axe/alt-review.md

If the agent declares sub_agents, all referenced agent TOMLs and their skills must also be mounted.

Persistent Data

Agent memory persists across runs when you mount a data volume:

docker run --rm \
  -v ./my-config:/home/axe/.config/axe \
  -v axe-data:/home/axe/.local/share/axe \
  -e ANTHROPIC_API_KEY \
  axe run my-agent

Docker Compose

A docker-compose.yml is included for running axe alongside a local Ollama instance.

Cloud provider only (no Ollama):

docker compose run --rm axe run my-agent

With Ollama sidecar:

docker compose --profile ollama up -d ollama
docker compose --profile cli run --rm axe run my-agent

Pull an Ollama model:

docker compose --profile ollama exec ollama ollama pull llama3

Note: The compose axe service declares depends_on: ollama. Docker Compose will attempt to start the Ollama service whenever axe is started via compose, even for cloud-only runs. For cloud-only usage without Ollama, use docker run directly instead of docker compose run.

Ollama on the Host

If Ollama runs directly on the host (not via compose), point to it with:

• Linux: --add-host=host.docker.internal:host-gateway -e AXE_OLLAMA_BASE_URL=http://host.docker.internal:11434
• macOS / Windows (Docker Desktop): -e AXE_OLLAMA_BASE_URL=http://host.docker.internal:11434

Security

The container runs with the following hardening by default (via compose):

• Non-root user — UID 10001
• Read-only root filesystem — writable locations are the config mount, data mount, and /tmp/axe tmpfs
• All capabilities dropped — cap_drop: ALL
• No privilege escalation — no-new-privileges:true

These settings do not restrict outbound network access. To isolate an agent that only talks to a local Ollama instance, add --network=none and connect it to the shared Docker network manually.

Volume Mounts

Config is read-write because axe config init and axe agents init write into it. Mount as :ro if you only run agents.

Environment Variables

CLI Reference

Commands

Run Flags

User Message Precedence

The user message sent to the LLM is resolved in this order:

1. -p / --prompt flag — If provided with a non-empty, non-whitespace value, it is used as the user message.
2. Piped stdin — If -p is absent or empty/whitespace-only, piped stdin is used.
3. Built-in default — If neither -p nor stdin provides content, the default message "Execute the task described in your instructions." is used.

When -p is provided alongside piped stdin, the piped stdin is silently ignored (no warning is emitted). An empty or whitespace-only -p value is treated as absent and falls through to stdin, then the default.

Example:

axe run my-agent -p "Summarize the README"

Exit Codes

Agent Configuration

Agents are defined as TOML files in $XDG_CONFIG_HOME/axe/agents/.

name = "pr-reviewer"
description = "Reviews pull requests for issues and improvements"
model = "anthropic/claude-sonnet-4-20250514"
system_prompt = "You are a senior code reviewer. Be concise and actionable."
skill = "skills/code-review/SKILL.md"
files = ["src/**/*.go", "CONTRIBUTING.md"]
workdir = "/home/user/projects/myapp"
tools = ["read_file", "list_directory", "run_command"]
sub_agents = ["test-runner", "lint-checker"]
allowed_hosts = ["api.example.com", "docs.example.com"]

[sub_agents_config]
max_depth = 3       # maximum nesting depth (hard max: 5)
parallel = true     # run sub-agents concurrently
timeout = 120       # per sub-agent timeout in seconds

[memory]
enabled = true
last_n = 10         # load last N entries into context
max_entries = 100   # warn when exceeded

[[mcp_servers]]
name = "my-tools"
url = "https://my-mcp-server.example.com/sse"
transport = "sse"
headers = { Authorization = "Bearer ${MY_TOKEN}" }

[params]
temperature = 0.3
max_tokens = 4096
# response_format = "json_object"   # text, json_object, or json_schema

[budget]
max_tokens = 50000    # 0 = unlimited (default)

[retry]
max_retries = 3           # retry up to 3 times on transient errors
backoff = "exponential"   # "exponential", "linear", or "fixed"
initial_delay_ms = 500    # base delay before first retry
max_delay_ms = 30000      # maximum delay cap

[[mcp_servers]]
name = "filesystem"
transport = "stdio"
command = "/usr/local/bin/mcp-server-filesystem"
args = ["--root", "/home/user/projects"]

All fields except name and model are optional.

Retry

Agents can retry on transient LLM provider errors — rate limits (429), server errors (5xx), and timeouts. Retry is opt-in and disabled by default.

Only transient errors are retried. Authentication errors (401/403) and bad requests (400) are never retried. When --verbose is enabled, each retry attempt is logged to stderr. The --json envelope includes a retry_attempts field for observability.

Output Allowlist

Agents that use url_fetch or web_search can be restricted to specific hostnames with the allowed_hosts field:

allowed_hosts = ["api.example.com", "docs.example.com"]

Token Budget

Cap cumulative token usage (input + output, across all turns and sub-agent calls) for a single run:

[budget]
max_tokens = 50000   # 0 = unlimited (default)

Or override via flag:

axe run my-agent --max-tokens 10000

The flag takes precedence over TOML when set to a value greater than zero.

When the budget is exceeded, the current response is returned but no further tool calls execute. The process exits with code 4. Memory is not appended on a budget-exceeded run.

With --verbose, each turn logs cumulative usage to stderr. With --json, the output envelope includes budget_max_tokens, budget_used_tokens, and budget_exceeded fields (omitted when unlimited).

Response Format

Constrain LLM output to a specific format via the [params] section. Supported by OpenAI and Ollama providers; other providers ignore this setting.

# String shorthand
[params]
response_format = "json_object"

# Table form (required for json_schema)
[params.response_format]
type = "json_schema"
[params.response_format.schema]
name = "my_schema"
type = "object"

Tools

Agents can use built-in tools to interact with the filesystem and run commands. When tools are enabled, the agent enters a conversation loop — the LLM can make tool calls, receive results, and continue reasoning for up to 50 turns.

Built-in Tools

Enable tools by adding them to the agent’s tools field:

tools = ["read_file", "list_directory", "run_command"]

The call_agent tool is not listed in tools — it is automatically available when sub_agents is configured and the depth limit has not been reached.

Path Security

All file tools (list_directory, read_file, write_file, edit_file) are sandboxed to the agent’s working directory. Absolute paths, .. traversal, and symlink escapes are rejected.

Parallel Execution

When an LLM returns multiple tool calls in a single turn, they run concurrently by default. This applies to both built-in tools and sub-agent calls. Disable with parallel = false in [sub_agents_config].

MCP Tools

Agents can use tools from external MCP servers. Declare servers in the agent TOML with [[mcp_servers]]:

[[mcp_servers]]
name = "my-tools"
url = "https://my-mcp-server.example.com/sse"
transport = "sse"
headers = { Authorization = "Bearer ${MY_TOKEN}" }

At startup, axe connects to each declared server, discovers available tools via tools/list, and makes them available to the LLM alongside built-in tools.

MCP tools are controlled entirely by [[mcp_servers]] — they are not listed in the tools field. If an MCP tool has the same name as an enabled built-in tool, the built-in takes precedence.

Skills

Skills are reusable instruction sets that provide an agent with domain-specific knowledge and workflows. They are defined as SKILL.md files following the community SKILL.md format.

Skill Resolution

The skill field in an agent TOML is resolved in order:

1. Absolute path — used as-is (e.g. /home/user/skills/SKILL.md)
2. Relative to config dir — e.g. skills/code-review/SKILL.md resolves to $XDG_CONFIG_HOME/axe/skills/code-review/SKILL.md
3. Bare name — e.g. code-review resolves to $XDG_CONFIG_HOME/axe/skills/code-review/SKILL.md

Script Paths

Skills often reference helper scripts. Since run_command executes in the agent’s workdir (not the skill directory), script paths in SKILL.md must be absolute. Relative paths will fail because the scripts don’t exist in the agent’s working directory.

# Correct — absolute path
/home/user/.config/axe/skills/my-skill/scripts/fetch.sh 

# Wrong — relative path won't resolve from the agent's workdir
scripts/fetch.sh 

Directory Structure

$XDG_CONFIG_HOME/axe/
├── config.toml
├── agents/
│   └── my-agent.toml
└── skills/
    └── my-skill/
        ├── SKILL.md
        └── scripts/
            └── fetch.sh

Local Agent Directories

By default, agents are loaded from $XDG_CONFIG_HOME/axe/agents/. Axe also supports project-local agent directories for per-repo agent definitions.

Auto-Discovery

If /axe/agents/ exists, axe searches it before the global config directory. A local agent with the same name as a global agent shadows the global one.

my-project/
└── axe/
    └── agents/
        └── my-agent.toml   ← found automatically

Explicit Override

Use --agents-dir to point to any directory:

axe run my-agent --agents-dir ./custom/agents

This flag is available on all commands: run, agents list, agents show, agents init, agents edit, and gc.

Resolution Order

1. --agents-dir (if provided)
2. /axe/agents/ (auto-discovered)
3. $XDG_CONFIG_HOME/axe/agents/ (global fallback)

The first directory containing a matching .toml wins.

Smart Scaffolding

axe agents init writes to /axe/agents/ if that directory already exists, otherwise falls back to the global config directory.

Providers

AWS Bedrock Configuration:

• Region: Set via AWS_REGION environment variable or [providers.bedrock] region = "us-east-1" in config.toml
• Credentials: Uses environment variables (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_SESSION_TOKEN) or ~/.aws/credentials file (supports AWS_PROFILE and AWS_SHARED_CREDENTIALS_FILE)
• Model IDs: Use full Bedrock model IDs (e.g., bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0)

Base URLs can be overridden with AXE__BASE_URL environment variables or in config.toml.

License

Apache-2.0. See LICENSE.