Airflow MCP Server

🚀 用於控制Airflow的MCP服務器

本項目是一個用於通過Airflow API控制Airflow的Model Context Protocol（MCP）服務器，藉助它可以方便地對Airflow進行操作與管理。

🚀 快速開始

本項目可以通過配置Claude Desktop來使用，同時支持安全和不安全兩種運行模式。

📦 安裝指南

使用Claude Desktop的配置方法

{
  "mcpServers": {
    "airflow-mcp-server": {
      "command": "uvx",
      "args": [
        "airflow-mcp-server"
      ],
      "env": {
        "AIRFLOW_BASE_URL": "http://<host:port>/api/v1",
        // 使用AUTH_TOKEN進行基本認證
        "AUTH_TOKEN": "<base64_encoded_username_password>",
        // 或使用COOKIE進行基於Cookie的認證
        "COOKIE": "<session_cookie>"
      }
    }
  }
}

運行模式

服務器支持兩種運行模式：

• 安全模式 (--safe)：只允許讀取操作（GET請求）。當你希望防止對Airflow實例進行任何修改時，這非常有用。
• 不安全模式 (--unsafe)：允許所有操作，包括修改。這是默認模式。

以安全模式啟動：

airflow-mcp-server --safe

為了明確地以不安全模式啟動（儘管這是默認設置）：

airflow-mcp-server --unsafe

考慮事項

MCP服務器期望以下環境變量已配置：

• AIRFLOW_BASE_URL：Airflow API的基礎URL
• AUTH_TOKEN：用於基本認證的base64編碼用戶名:密碼（如果提供COOKIE則可選）
• COOKIE：用於基於Cookie認證的會話Cookie（如果提供AUTH_TOKEN則可選）
• OPENAPI_SPEC：OpenAPI規格文件的路徑（默認為最新穩定版本）

認證方法 服務器支持兩種認證方式：

• 基本認證：使用base64編碼的用戶名:密碼通過AUTH_TOKEN環境變量
• 基於Cookie的認證：使用會話Cookie通過COOKIE環境變量

必須至少提供一種認證方法。

頁面限制 默認顯示100個項目，但你可以通過在Airflow的airflow.cfg文件中[api]部分使用maximum_page_limit選項來更改此設置。

📚 詳細文檔

演示視頻

可通過以下鏈接查看演示視頻： https://github.com/user-attachments/assets/f3e60fff-8680-4dd9-b08e-fa7db655a705

任務列表

• [x] 首個API實現
• [x] 解析OpenAPI規格
• [x] 實現安全/不安全模式
• [x] 允許會話認證
• [ ] 使用list_tools解析正確的描述
• [ ] 獲取Airflow配置（特別是頁面限制）
• [ ] 環境變量的可選性（環境變量可能不適合Airflow插件）