airflow-mcp-server: 用于控制 Airflow 的 MCP 服务器

在 Glama 上查找

概述

一个通过 Airflow API 控制 Airflow 的模型上下文协议服务器。

演示视频

https://github.com/user-attachments/assets/f3e60fff-8680-4dd9-b08e-fa7db655a705

设置

与 Claude Desktop 一起使用

{
    "mcpServers": {
        "airflow-mcp-server": {
            "command": "uvx",
            "args": [
                "airflow-mcp-server",
                "--base-url",
                "http://localhost:8080",
                "--auth-token",
                "<jwt_token>"
            ]
        }
    }
}

注意：

• 将 base_url 设置为 Airflow 的根 URL（例如，http://localhost:8080）。
• 不要在 base URL 中包含 /api/v2。服务器将自动从 ${base_url}/openapi.json 获取 OpenAPI 规范。
• 仅需要 JWT 令牌进行身份验证。Airflow 3.0 不再支持 cookie 和基本身份验证。

操作模式

服务器支持两种操作模式：

• 安全模式 (--safe): 仅允许只读操作（GET 请求）。这在您希望防止对 Airflow 实例进行任何修改时非常有用。
• 非安全模式 (--unsafe): 允许所有操作，包括修改。这是默认模式。

要以安全模式启动：

airflow-mcp-server --safe

要显式地以非安全模式启动（尽管这是默认模式）：

airflow-mcp-server --unsafe

工具发现模式

服务器支持两种工具发现方法：

• 分层发现（默认）：工具按类别（DAGs、Tasks、Connections 等）组织。先浏览类别，然后选择特定工具。对于大型 API 更易于管理。
• 静态工具 (--static-tools): 所有工具立即可用。更适合编程访问，但可能会让人不知所措。

要使用静态工具：

airflow-mcp-server --static-tools

注意事项

身份验证

• 在 Airflow 3.0 中仅支持 JWT 身份验证。您必须提供有效的 AUTH_TOKEN。

页面限制

默认值为 100 个项目，但您可以在 airflow.cfg 文件的 [api] 部分中使用 maximum_page_limit 选项更改它。

任务

• [x] Airflow 3 准备就绪
• [x] 解析 OpenAPI 规范
• [x] 安全/非安全模式实现
• [x] 使用 list_tools 解析正确的描述。
• [x] 获取 Airflow 配置（特别是页面限制）
• [ ] 环境变量可选（环境变量可能不适合 Airflow 插件）