Skip to main content

Apache Airflow MCP Server

MCP PyPI Python Airflow CI License Ruff

Independent community project; not affiliated with or endorsed by the Apache Software Foundation.

Connect Claude, Cursor, and any MCP client to your Apache Airflow deployments — and let AI agents debug failed DAGs for you.

Paste an Airflow UI link from a PagerDuty/Datadog alert and ask "why did this fail?" — the agent resolves the URL, finds the failed tasks, pulls the error lines from the logs (server-side filtered so it doesn't blow the context window), and can re-trigger or clear runs. Write tools carry destructive-operation annotations that MCP clients can use to request confirmation.

Highlights

  • 🔍 Incident-response first — resolve Airflow UI URLs straight to the failing task, filter logs by error level with context lines, follow try_number semantics correctly (sensors included)
  • 🏢 Multi-instance — one server for same-API-family dev/staging/prod targets, with per-instance credentials and an SSRF guard that rejects unknown hosts
  • 🔒 Safety controls — opt-in read-only mode (AIRFLOW_MCP_READ_ONLY=true) that never registers write tools; write tools annotated as destructive; configured credentials are redacted from instance responses and operation logs
  • 📉 Token-efficient — log tailing, level filtering, byte caps, and truncation metadata designed for LLM context windows
  • 🧭 Airflow 2 and 3 — API v1/v2 adapters with live E2E against Airflow 2.11 and 3.3, including JWT auth
  • 📎 Traceable — every response carries a request_id that matches the structured server logs

Quickstart

1. Install

uv tool install apache-airflow-mcp-server \
  --with 'apache-airflow-client==3.3.0'  # replace 3.3.0 with your Airflow version

Airflow 2.11? Use --with 'apache-airflow-client==2.10.0' instead; 2.10.0 is the final generated v1 client and targets Airflow 2's stable API. See Airflow compatibility before using a different release.

2. Connect your MCP client

The fastest path is a single instance configured entirely with environment variables — no config file needed.

Claude Code
claude mcp add airflow \
  --env AIRFLOW_MCP_HOST=https://airflow.example.com \
  --env AIRFLOW_MCP_USERNAME=admin \
  --env AIRFLOW_MCP_PASSWORD=your-password \
  -- uvx --from apache-airflow-mcp-server \
  --with apache-airflow-client==3.3.0 airflow-mcp --transport stdio
Claude Desktop

Add to claude_desktop_config.json (Settings → Developer → Edit Config):

{
  "mcpServers": {
    "airflow": {
      "command": "uvx",
      "args": ["--from", "apache-airflow-mcp-server", "--with", "apache-airflow-client==3.3.0", "airflow-mcp", "--transport", "stdio"],
      "env": {
        "AIRFLOW_MCP_HOST": "https://airflow.example.com",
        "AIRFLOW_MCP_USERNAME": "admin",
        "AIRFLOW_MCP_PASSWORD": "your-password"
      }
    }
  }
}
Cursor

Add to ~/.cursor/mcp.json:

{
  "mcpServers": {
    "airflow": {
      "command": "uvx",
      "args": ["--from", "apache-airflow-mcp-server", "--with", "apache-airflow-client==3.3.0", "airflow-mcp", "--transport", "stdio"],
      "env": {
        "AIRFLOW_MCP_HOST": "https://airflow.example.com",
        "AIRFLOW_MCP_USERNAME": "admin",
        "AIRFLOW_MCP_PASSWORD": "your-password"
      }
    }
  }
}
VS Code (Copilot)

Add to .vscode/mcp.json:

{
  "servers": {
    "airflow": {
      "type": "stdio",
      "command": "uvx",
      "args": ["--from", "apache-airflow-mcp-server", "--with", "apache-airflow-client==3.3.0", "airflow-mcp", "--transport", "stdio"],
      "env": {
        "AIRFLOW_MCP_HOST": "https://airflow.example.com",
        "AIRFLOW_MCP_USERNAME": "admin",
        "AIRFLOW_MCP_PASSWORD": "your-password"
      }
    }
  }
}
Any client, over HTTP

Run the server yourself and point the client at the endpoint:

AIRFLOW_MCP_HOST=https://airflow.example.com \
AIRFLOW_MCP_USERNAME=admin AIRFLOW_MCP_PASSWORD=your-password \
airflow-mcp --transport http --host 127.0.0.1 --port 8765
{ "mcpServers": { "airflow": { "url": "http://127.0.0.1:8765/mcp" } } }

Health check: GET /health200 OK.

3. Ask your agent something

"Why did the latest run of etl_pipeline fail?"

"https://airflow.example.com/dags/etl_pipeline/grid — what happened here, and is it safe to clear?"

"Preview clearing the failed task from that alert, then explain what would rerun."

Airflow compatibility

The server talks to Airflow through the generated apache-airflow-client. For Airflow 3, match the client release to your Airflow release: generated models can change within a major version, and a newer client is not guaranteed to deserialize an older server's responses correctly. Airflow 2.11 uses the final v1 client release, 2.10.0, against Airflow 2's stable API.

Your Airflow REST API Install Live E2E status
3.3 v2 uv tool install apache-airflow-mcp-server --with 'apache-airflow-client==3.3.0' ✅ 3.3.0
2.11 v1 uv tool install apache-airflow-mcp-server --with 'apache-airflow-client==2.10.0' ✅ Airflow 2.11 + final v1 client 2.10.0
3.0–3.2 v2 Pin the client to the deployed Airflow 3 version 🧪 Not in the current live matrix
2.5–2.10 v1 Use the final v1 client, apache-airflow-client==2.10.0 🧪 Not in the current live matrix

When api_version isn't set, the server assumes the API matching the installed client (v1 for a 2.x client, v2 for 3.x). Set AIRFLOW_MCP_API_VERSION (or api_version: in the registry) explicitly to catch a major-version mismatch early.

One server process can currently load only one generated client major. All instances in a registry must therefore use the same API family; run separate MCP server processes for Airflow 2 and Airflow 3. Mixed-version support requires a future client-adapter change and is not advertised as working today.

Airflow 3 notes:

  • Auth: bearer tokens are passed through as JWTs; basic credentials are automatically exchanged for a JWT via POST /auth/token and refreshed periodically (AIRFLOW_MCP_TOKEN_REFRESH_SECONDS, default 3600 — keep it below your deployment's JWT expiry, and note there is no automatic re-auth on 401 yet).
  • execution_date ordering maps to logical_date, datasets map to assets, and UI links use the Airflow 3 route scheme. Tool names and the core workflow stay stable; documented fields and options can differ by API family.
  • Clear options that no longer exist in Airflow 3 (include_subdags/include_parentdag, and the include_*/reset_dag_runs options of airflow_clear_dag_run) are rejected with INVALID_INPUT rather than silently narrowing a destructive operation.

Both client families are exercised on relevant pull requests and main pushes. Bug reports from real Airflow deployments are very welcome!

Configuration

Single instance (env vars only)

Variable Required Description
AIRFLOW_MCP_HOST Airflow base URL, e.g. https://airflow.example.com
AIRFLOW_MCP_USERNAME / AIRFLOW_MCP_PASSWORD ✅* Basic auth credentials
AIRFLOW_MCP_TOKEN ✅* Bearer/JWT token (used instead of basic auth)
AIRFLOW_MCP_API_VERSION v1 (Airflow 2) or v2 (Airflow 3); defaults to whichever matches the installed apache-airflow-client
AIRFLOW_MCP_VERIFY_SSL Verify TLS certificates (default true)

* provide either username+password or a token.

Multiple instances (registry YAML)

Point AIRFLOW_MCP_INSTANCES_FILE at a YAML registry (it takes precedence over the single-instance env vars). Values may reference environment variables with ${VAR}:

data-stg:
  host: https://airflow.data-stg.example.com/
  api_version: v1        # Airflow 2
  verify_ssl: true
  auth:
    type: basic
    username: ${AIRFLOW_DATA_STG_USERNAME}
    password: ${AIRFLOW_DATA_STG_PASSWORD}

data-prod:
  host: https://airflow.data-prod.example.com/
  api_version: v1        # Keep one client/API family per server process
  auth:
    type: bearer
    token: ${AIRFLOW_DATA_PROD_TOKEN}

Every tool accepts either an instance key (data-stg) or a ui_url — a full http(s) Airflow UI URL whose host is resolved against the registry, with unknown hosts rejected (SSRF guard). ui_url also auto-fills dag_id/dag_run_id/task_id when the link contains them. If both instance and ui_url are passed and disagree, the call fails with INSTANCE_MISMATCH rather than guessing.

Kubernetes tip: mount the registry from a Secret at /config/instances.yaml and set AIRFLOW_MCP_INSTANCES_FILE=/config/instances.yaml.

Server options

Variable Default Description
AIRFLOW_MCP_DEFAULT_INSTANCE Default instance key (also names the env-var instance)
AIRFLOW_MCP_READ_ONLY false Don't register write tools at all
AIRFLOW_MCP_HTTP_HOST / AIRFLOW_MCP_HTTP_PORT 127.0.0.1 / 8765 HTTP transport bind
AIRFLOW_MCP_TIMEOUT_SECONDS 30 Airflow API timeout
AIRFLOW_MCP_TOKEN_REFRESH_SECONDS 3600 Airflow 3: JWT refresh interval for basic-auth instances
AIRFLOW_MCP_LOG_FILE Optional log file path
AIRFLOW_MCP_ENABLE_EXTENDED_CLEAR_PARAMS false Enable include_* clear params (Airflow ≥2.6)
AIRFLOW_MCP_HTTP_BLOCK_GET_ON_MCP true Return 405 for GET /mcp (SSE reads) on HTTP deployments

Read-only mode

Pointing an AI agent at production? Set AIRFLOW_MCP_READ_ONLY=true and the write tools (trigger, clear, pause/unpause) are never registered. This prevents MCP mutations, but read tools can still disclose sensitive logs, configuration, rendered fields, and DAG-run data; use least-privilege Airflow credentials. With writes enabled, tools carry MCP destructiveHint annotations that clients can use when deciding whether to request confirmation.

Tools

Discovery & URL utilities

Tool Description
airflow_list_instances List configured instance keys and the default
airflow_describe_instance Host, API version, auth type (secrets redacted)
airflow_resolve_url Parse an Airflow UI URL into instance + dag/run/task identifiers

Read

Tool Description
airflow_list_dags DAGs with pause state and UI links
airflow_get_dag DAG details
airflow_list_dag_runs Runs with state filters and ordering (latest first by default)
airflow_get_dag_run Single run details
airflow_list_task_instances Task attempts for a run; filter by state / task_ids server-side
airflow_get_task_instance Task metadata, retries, timings, optional rendered template fields
airflow_get_task_instance_logs Logs with level filtering, tailing, context lines, and byte caps
airflow_dataset_events Dataset (Airflow 2) / asset (Airflow 3) events

Write (annotated destructive so clients can require approval; hidden entirely in read-only mode)

Tool Description
airflow_trigger_dag Trigger a run with optional conf/logical date/note
airflow_clear_task_instances Clear task instances across runs (dry_run=true by default)
airflow_clear_dag_run Clear a whole run (dry_run=true by default)
airflow_pause_dag / airflow_unpause_dag Toggle DAG scheduling

Every success payload includes a request_id for log correlation; failures raise a structured ToolError with {code, message, request_id, context}.

The incident workflow

This is the flow the tools were designed around — going from an alert link to a diagnosis in four calls:

# 1. Alert contains an Airflow UI link → resolve it
airflow_resolve_url("https://airflow.example.com/dags/etl_pipeline/grid?dag_run_id=...")
#    → {instance, dag_id, dag_run_id, ...}

# 2. Which tasks failed in this run?
airflow_list_task_instances(dag_id="etl_pipeline", dag_run_id="scheduled__2026-01-01",
                            state=["failed"])

# 3. Get attempt metadata (authoritative try_number, retries, timings)
ti = airflow_get_task_instance(dag_id="etl_pipeline",
                               dag_run_id="scheduled__2026-01-01",
                               task_id="transform_data")

# 4. Pull only the error lines, with context, capped for the LLM
airflow_get_task_instance_logs(dag_id="etl_pipeline",
                               dag_run_id="scheduled__2026-01-01",
                               task_id="transform_data",
                               try_number=ti["attempts"]["try_number"],
                               tail_lines=500, filter_level="error", context_lines=5)

Log responses include truncated, auto_tailed (logs >100MB tail automatically), match_count, and byte/line stats so the agent knows exactly what it's looking at. Host-segmented logs are flattened with --- [worker-1] --- headers; Airflow 3 structured logs are rendered as plain lines.

Note on try_number: reschedule-mode sensors could increment it on every reschedule through Airflow 2.9; Airflow 2.10+ no longer does. Always read it from airflow_get_task_instance rather than guessing—the derived retries_consumed/retries_remaining fields are heuristics.

Deployment

Docker

docker run -p 127.0.0.1:8765:8765 \
  -e AIRFLOW_MCP_HOST=https://airflow.example.com \
  -e AIRFLOW_MCP_USERNAME=admin \
  -e AIRFLOW_MCP_PASSWORD=your-password \
  ghcr.io/madamak/apache-airflow-mcp-server:latest

Or build locally with docker build -t airflow-mcp .

The release image contains the lockfile's Airflow 3.3 client and serves streamable HTTP on :8765 (/mcp endpoint, /health for probes). The MCP HTTP endpoint has no built-in caller authentication: keep it loopback-bound or place it behind an authenticated private proxy. Mount a same-API-family registry YAML for multi-instance setups. Airflow 2 deployments should use the pinned local installation path above until a separate v1 image is published.

FastMCP tooling

A fastmcp.json is included so FastMCP-aware tooling can auto-discover the entrypoint and deployment defaults.

Development

uv sync                 # install dependencies
uv run pytest           # unit tests (no real network; the Airflow client is mocked)
uv run ruff check .     # lint
uv run ruff format .    # format
uv run airflow-mcp --transport stdio   # run locally
./scripts/e2e.sh af2    # end-to-end against Airflow 2.11
./scripts/e2e.sh af3    # end-to-end against Airflow 3.3
                        # Both seed failures/noisy logs and drive every tool
                        # through MCP. Set E2E_KEEP=1 to keep the instance up.

CI runs the unit suite against both apache-airflow-client families on Python 3.10–3.13. Relevant pull requests, main pushes, nightly runs, and releases also exercise live dockerized Airflow 2.11 and 3.3. See CONTRIBUTING.md for guidelines and AGENTS.md if you're pointing a coding agent at this repo (it's written for that).

Contributing

Issues and PRs are welcome — especially:

  • Reports from real Airflow incident-response workflows and version combinations
  • Bounded-log, diagnosis-safety, and URL-first workflow improvements
  • Client setup recipes for more MCP hosts and deployment types

If this server saves you a debugging session, a ⭐ helps other Airflow teams find it.

License

Apache 2.0 — see LICENSE.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

apache_airflow_mcp_server-1.0.0.tar.gz (206.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

apache_airflow_mcp_server-1.0.0-py3-none-any.whl (61.3 kB view details)

Uploaded Python 3

File details

Details for the file apache_airflow_mcp_server-1.0.0.tar.gz.

File metadata

File hashes

Hashes for apache_airflow_mcp_server-1.0.0.tar.gz
Algorithm Hash digest
SHA256 5f74df51e11740432d4c790851cea0fdfca81f2dca17cbf8b349e5bffc60ae2c
MD5 4f933f6d4b083d4d61181e954ea74cab
BLAKE2b-256 2edb40db0dfe5cf694efe8a5cbd0509451f67131367191117486212393264917

See more details on using hashes here.

Provenance

The following attestation bundles were made for apache_airflow_mcp_server-1.0.0.tar.gz:

Publisher: publish.yml on madamak/apache-airflow-mcp-server

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file apache_airflow_mcp_server-1.0.0-py3-none-any.whl.

File metadata

File hashes

Hashes for apache_airflow_mcp_server-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 7fa139dcb708f7caccec753135eeb52089ebcc3bbd0599d0a4a36321ff8c1f44
MD5 1becaf1f983eff648583e614bc510eeb
BLAKE2b-256 1e5ef6f5d722067805ca9d60331d7035c7b51274b3ce2f4efa296ef911ee49c1

See more details on using hashes here.

Provenance

The following attestation bundles were made for apache_airflow_mcp_server-1.0.0-py3-none-any.whl:

Publisher: publish.yml on madamak/apache-airflow-mcp-server

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

1.0.2

2 files

1.0.1

2 files

This release

1.0.0 This release

2 files

0.1.1

2 files

0.1.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page