mcp-server-apache-airflow
A Model Context Protocol (MCP) server implementation for Apache Airflow, enabling seamless integration with MCP clients. This project provides a standardized way to interact with Apache Airflow through the Model Context Protocol.
About
This project implements a Model Context Protocol server that wraps Apache Airflow's REST API, allowing MCP clients to interact with Airflow in a standardized way. It uses the official Apache Airflow client library to ensure compatibility and maintainability.
Feature Implementation Status
| Feature | API Path | Status |
|---|---|---|
| DAG Management | ||
| List DAGs | /api/v1/dags |
✅ |
| Get DAG Details | /api/v1/dags/{dag_id} |
✅ |
| Pause DAG | /api/v1/dags/{dag_id} |
✅ |
| Unpause DAG | /api/v1/dags/{dag_id} |
✅ |
| Update DAG | /api/v1/dags/{dag_id} |
✅ |
| Delete DAG | /api/v1/dags/{dag_id} |
✅ |
| Get DAG Source | /api/v1/dagSources/{file_token} |
✅ |
| Patch Multiple DAGs | /api/v1/dags |
✅ |
| Reparse DAG File | /api/v1/dagSources/{file_token}/reparse |
✅ |
| DAG Runs | ||
| List DAG Runs | /api/v1/dags/{dag_id}/dagRuns |
✅ |
| Create DAG Run | /api/v1/dags/{dag_id}/dagRuns |
✅ |
| Get DAG Run Details | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id} |
✅ |
| Update DAG Run | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id} |
✅ |
| Delete DAG Run | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id} |
✅ |
| Get DAG Runs Batch | /api/v1/dags/~/dagRuns/list |
✅ |
| Clear DAG Run | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/clear |
✅ |
| Set DAG Run Note | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/setNote |
✅ |
| Get Upstream Dataset Events | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/upstreamDatasetEvents |
✅ |
| Tasks | ||
| List DAG Tasks | /api/v1/dags/{dag_id}/tasks |
✅ |
| Get Task Details | /api/v1/dags/{dag_id}/tasks/{task_id} |
✅ |
| Get Task Instance | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/taskInstances/{task_id} |
✅ |
| List Task Instances | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/taskInstances |
✅ |
| Update Task Instance | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/taskInstances/{task_id} |
✅ |
| Clear Task Instances | /api/v1/dags/{dag_id}/clearTaskInstances |
✅ |
| Set Task Instances State | /api/v1/dags/{dag_id}/updateTaskInstancesState |
✅ |
| Variables | ||
| List Variables | /api/v1/variables |
✅ |
| Create Variable | /api/v1/variables |
✅ |
| Get Variable | /api/v1/variables/{variable_key} |
✅ |
| Update Variable | /api/v1/variables/{variable_key} |
✅ |
| Delete Variable | /api/v1/variables/{variable_key} |
✅ |
| Connections | ||
| List Connections | /api/v1/connections |
✅ |
| Create Connection | /api/v1/connections |
✅ |
| Get Connection | /api/v1/connections/{connection_id} |
✅ |
| Update Connection | /api/v1/connections/{connection_id} |
✅ |
| Delete Connection | /api/v1/connections/{connection_id} |
✅ |
| Test Connection | /api/v1/connections/test |
✅ |
| Pools | ||
| List Pools | /api/v1/pools |
✅ |
| Create Pool | /api/v1/pools |
✅ |
| Get Pool | /api/v1/pools/{pool_name} |
✅ |
| Update Pool | /api/v1/pools/{pool_name} |
✅ |
| Delete Pool | /api/v1/pools/{pool_name} |
✅ |
| XComs | ||
| List XComs | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/taskInstances/{task_id}/xcomEntries |
✅ |
| Get XCom Entry | /api/v1/dags/{dag_id}/dagRuns/{dag_run_id}/taskInstances/{task_id}/xcomEntries/{xcom_key} |
✅ |
| Datasets | ||
| List Datasets | /api/v1/datasets |
✅ |
| Get Dataset | /api/v1/datasets/{uri} |
✅ |
| Get Dataset Events | /api/v1/datasetEvents |
✅ |
| Create Dataset Event | /api/v1/datasetEvents |
✅ |
| Get DAG Dataset Queued Event | /api/v1/dags/{dag_id}/dagRuns/queued/datasetEvents/{uri} |
✅ |
| Get DAG Dataset Queued Events | /api/v1/dags/{dag_id}/dagRuns/queued/datasetEvents |
✅ |
| Delete DAG Dataset Queued Event | /api/v1/dags/{dag_id}/dagRuns/queued/datasetEvents/{uri} |
✅ |
| Delete DAG Dataset Queued Events | /api/v1/dags/{dag_id}/dagRuns/queued/datasetEvents |
✅ |
| Get Dataset Queued Events | /api/v1/datasets/{uri}/dagRuns/queued/datasetEvents |
✅ |
| Delete Dataset Queued Events | /api/v1/datasets/{uri}/dagRuns/queued/datasetEvents |
✅ |
| Monitoring | ||
| Get Health | /api/v1/health |
✅ |
| DAG Stats | ||
| Get DAG Stats | /api/v1/dags/statistics |
✅ |
| Config | ||
| Get Config | /api/v1/config |
✅ |
| Plugins | ||
| Get Plugins | /api/v1/plugins |
✅ |
| Providers | ||
| List Providers | /api/v1/providers |
✅ |
| Event Logs | ||
| List Event Logs | /api/v1/eventLogs |
✅ |
| Get Event Log | /api/v1/eventLogs/{event_log_id} |
✅ |
| System | ||
| Get Import Errors | /api/v1/importErrors |
✅ |
| Get Import Error Details | /api/v1/importErrors/{import_error_id} |
✅ |
| Get Health Status | /api/v1/health |
✅ |
| Get Version | /api/v1/version |
✅ |
Setup
Dependencies
This project depends on the official Apache Airflow client library (apache-airflow-client). It will be automatically installed when you install this package.
Environment Variables
Set the following environment variables:
AIRFLOW_HOST=<your-airflow-host>
AIRFLOW_USERNAME=<your-airflow-username>
AIRFLOW_PASSWORD=<your-airflow-password>
Usage with Claude Desktop
Add to your claude_desktop_config.json:
{
"mcpServers": {
"mcp-server-apache-airflow": {
"command": "uvx",
"args": ["mcp-server-apache-airflow"],
"env": {
"AIRFLOW_HOST": "https://your-airflow-host",
"AIRFLOW_USERNAME": "your-username",
"AIRFLOW_PASSWORD": "your-password"
}
}
}
}
Alternative configuration using uv:
{
"mcpServers": {
"mcp-server-apache-airflow": {
"command": "uv",
"args": [
"--directory",
"/path/to/mcp-server-apache-airflow",
"run",
"mcp-server-apache-airflow"
],
"env": {
"AIRFLOW_HOST": "https://your-airflow-host",
"AIRFLOW_USERNAME": "your-username",
"AIRFLOW_PASSWORD": "your-password"
}
}
}
}
Replace /path/to/mcp-server-apache-airflow with the actual path where you've cloned the repository.
Selecting the API groups
You can select the API groups you want to use by setting the --apis flag.
uv run mcp-server-apache-airflow --apis "dag,dagrun"
The default is to use all APIs.
Allowed values are:
- config
- connections
- dag
- dagrun
- dagstats
- dataset
- eventlog
- importerror
- monitoring
- plugin
- pool
- provider
- taskinstance
- variable
- xcom
Manual Execution
You can also run the server manually:
make run
make run accepts following options:
Options:
--port: Port to listen on for SSE (default: 8000)--transport: Transport type (stdio/sse, default: stdio)
Or, you could run the sse server directly, which accepts same parameters:
make run-sse
Installing via Smithery
To install Apache Airflow MCP Server for Claude Desktop automatically via Smithery:
npx -y @smithery/cli install @yangkyeongmo/mcp-server-apache-airflow --client claude
Contributing
Contributions are welcome! Please feel free to submit a Pull Request.