Merge branch 'main' into glama-badge

Author: yangkyeongmo

Diff for: Makefile

@@ -1,8 +1,17 @@
+.PHONY: run build publish lint format
+
+PYTHON=uv run --env-file .env python
+
+PARAMS=
+run:
+	$(PYTHON) src $(PARAMS)
+run-sse:
+	$(PYTHON) src --transport sse $(PARAMS)
 build:
-	uv run python -m build
+	uv build
 publish:
-	uv run python -m twine upload --config-file .pypirc dist/*
+	uv publish
 lint:
-	uv run ruff check . --fix
+	$(PYTHON) -m ruff check . --fix
 format:
-	uv run ruff format .
+	$(PYTHON) -m ruff format .

Diff for: README.md

@@ -8,7 +8,7 @@ A Model Context Protocol (MCP) server implementation for Apache Airflow, enablin
 
 ## 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.
+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
 
@@ -19,40 +19,87 @@ This project implements a [Model Context Protocol](https://modelcontextprotocol.
 | 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}` | ❌ |
+| 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 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 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}` | ❌ |
+| 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` | ✅ |
-| **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}` | ❌ |
 
 ## 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:
@@ -107,21 +154,56 @@ Alternative configuration using `uv`:
 
 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
-python src/server.py
+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
+```
+
 ## Contributing
 
 Contributions are welcome! Please feel free to submit a Pull Request.
 
 ## License
 
-[Add your license information here]
+[MIT License](LICENSE)

Diff for: pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp-server-apache-airflow"
-version = "0.1.3"
+version = "0.2.1"
 description = "Model Context Protocol (MCP) server for Apache Airflow"
 authors = [
     { name = "Gyeongmo Yang", email = "[email protected]" }
@@ -9,6 +9,7 @@ dependencies = [
     "httpx>=0.24.1",
     "click>=8.1.7",
     "mcp>=0.1.0",
+    "apache-airflow-client>=2.7.0",
 ]
 requires-python = ">=3.10"
 readme = "README.md"
@@ -56,3 +57,8 @@ include = [
 [tool.ruff]
 select = ["E", "W", "F", "B", "I"]
 line-length = 120
+
+[dependency-groups]
+dev = [
+    "ruff>=0.11.0",
+]

Diff for: src/__main__.py

@@ -1,5 +1,5 @@
 import sys
 
-from src.server import main
+from src.main import main
 
 sys.exit(main())