Prefect MCP 服务器

这是针对 Prefect 的 Model Context Protocol (MCP) 服务器实现，允许AI助手通过自然语言与Prefect进行交互。

功能

此MCP服务器提供了对以下Prefect API的访问：

• 流程管理：列出、获取和删除流程
• 流程运行管理：创建、监控和控制流程运行
• 部署管理：管理和调度部署
• 任务运行管理：监控和控制任务运行
• 工作队列管理：创建和管理工作队列
• 块管理：访问块类型和文档
• 变量管理：创建和管理变量
• 工作区管理：获取关于工作区的信息

配置

设置以下环境变量：

export PREFECT_API_URL="http://localhost:4200/api"  # Prefect API的URL
export PREFECT_API_KEY="your_api_key"               # Prefect API密钥（如果使用Prefect Cloud）

使用方法

运行MCP服务器和Prefect：

docker compose up

示例输入

连接后，AI助手可以协助用户通过自然语言与Prefect进行交互。示例：

• "显示我所有的流程"
• "列出昨天所有失败的流程运行"
• "触发'data-processing'部署"
• "暂停'daily-reporting'部署的调度"
• "我的最后一个ETL流程运行的状态是什么？"

开发

几个端点尚未实现

添加新功能

要向现有API添加新功能：

1. 将该功能添加到src/mcp_prefect中的适当模块中
2. 将该功能添加到模块中的get_all_functions()列表中

要添加新的API类型：

1. 在enums.py中的APIType添加新的类型
2. 在src/prefect/中创建一个新的模块
3. 更新main.py以包含新的API类型

示例用法：

{
  "mcpServers": {
    "mcp-prefect": {
      "command": "mcp-prefect",
      "args": [
        "--transport", "sse"
      ],
      "env": {
        "PYTHONPATH": "/path/to/your/project/directory"
      },
      "cwd": "/path/to/your/project/directory"
    }
  }
}