https://github.com/yangkyeongmo/mcp-server-apache-airflow
https://github.com/yangkyeongmo/mcp-server-apache-airflow
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/yangkyeongmo/mcp-server-apache-airflow
- Owner: yangkyeongmo
- License: mit
- Created: 2025-02-13T04:09:06.000Z (8 months ago)
- Default Branch: main
- Last Pushed: 2025-03-20T13:35:07.000Z (7 months ago)
- Last Synced: 2025-03-20T14:25:33.547Z (7 months ago)
- Language: Python
- Homepage: https://pypi.org/project/mcp-server-apache-airflow/
- Size: 222 KB
- Stars: 3
- Watchers: 1
- Forks: 3
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
- Codeowners: .github/CODEOWNERS
Awesome Lists containing this project
- awesome-mcp-list - @yangkyeongmo/mcp-server-apache-airflow - server-apache-airflow?style=social)](https://github.com/yangkyeongmo/mcp-server-apache-airflow): Connects to Apache Airflow using the official client. (Uncategorized / Uncategorized)
- awesome-mcp-servers - **mcp-server-apache-airflow** - Python-based (actively maintained) `python` `http` `ai` `git` `github` `pip install git+https://github.com/yangkyeongmo/mcp-server-apache-airflow` (AI/ML)
- awesome-mcp-servers - **mcp-server-apache-airflow** - Python-based (actively maintained) `python` `http` `ai` `git` `github` `pip install git+https://github.com/yangkyeongmo/mcp-server-apache-airflow` (AI/ML)
- best-of-mcp-servers - GitHub - 40% open Β· β±οΈ 13.09.2025) (Developer Tools)
- awesome-mcp-servers - @yangkyeongmo@/mcp-server-apache-airflow - MCP server that connects to [Apache Airflow](https://airflow.apache.org/) using official client. (Legend / π» <a name="developer-tools"></a>Developer Tools)
- Awesome-MCP-Servers-directory - Airflow - A MCP Server that connects to Apache Airflow using official python client (Workflow Automation)
- awesome-mcp-servers - Airflow MCP Server - Provides a Model Context Protocol (MCP) server for standardized Apache Airflow interaction (Table of Contents / System Automation)
- awesome-mcp-servers - Airflow MCP Server - Provides a Model Context Protocol (MCP) server for standardized Apache Airflow interaction (Table of Contents / System Automation)
- mcp-index - Apache Airflow MCP Server - Integrates Apache Airflow's REST API with the Model Context Protocol, enabling MCP clients to interact with Airflow tasks and workflows effectively. Facilitates standardized communication between AI models and Apache Airflow for job orchestration. (Business Tools)
- awesome-mcp-servers - @yangkyeongmo@/mcp-server-apache-airflow - MCP server that connects to [Apache Airflow](https://airflow.apache.org/) using official client. (Legend / π» <a name="developer-tools"></a>Developer Tools)
- awesome-mcp - @yangkyeongmo@/mcp-server-apache-airflow - MCP server that connects toΒ [Apache Airflow](https://airflow.apache.org/)Β using official client. (MCP Servers / π» Developer Tools)
README
# mcp-server-apache-airflow
[](https://smithery.ai/server/@yangkyeongmo/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](https://modelcontextprotocol.io/introduction) 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=
AIRFLOW_USERNAME=
AIRFLOW_PASSWORD=
```### Usage with Claude Desktop
Add to your `claude_desktop_config.json`:
```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`:
```json
{
"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.
```bash
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:
```bash
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:
```bash
make run-sse
```### Installing via Smithery
To install Apache Airflow MCP Server for Claude Desktop automatically via [Smithery](https://smithery.ai/server/@yangkyeongmo/mcp-server-apache-airflow):
```bash
npx -y @smithery/cli install @yangkyeongmo/mcp-server-apache-airflow --client claude
```## Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
## License
[MIT License](LICENSE)