MCP 服务器

Prefect MCP Server

一个模型上下文协议（Model Context Protocol）服务器，它允许人工智能助手通过自然语言与 Prefect 的工作流自动化平台进行交互，从而使用户能够通过对话命令来管理流程（flows）、部署（deployments）、任务（tasks）以及其他 Prefect 资源。

README

Prefect MCP 服务器

一个用于 Prefect 的模型上下文协议 (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 添加新函数：

将函数添加到 src/mcp_prefect 中的相应模块
将函数添加到模块中的 get_all_functions() 列表中

要添加新的 API 类型：

将新类型添加到 enums.py 中的 APIType
在 src/prefect/ 中创建一个新模块
更新 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"
    }
  }
}