NiFi MCP 服务器

NiFi MCP 服务器是一个基于自然语言的模型上下文协议(MCP)服务器,用于与 Apache NiFi 进行交互和流程生成。它允许用户通过自然语言查询与 Apache NiFi 交互,处理用户查询,提取意图,并执行适当的 NiFi API 操作。

  • by kimcaprio
  • 更新于
在 GitHub 查看 kimcaprio/NIFI_MCP_SERVER

使用说明

基本能力

产品定位

NiFi MCP 服务器是一个基于自然语言的模型上下文协议(MCP)服务器,用于与 Apache NiFi 进行交互和流程生成。

核心功能

  • 自然语言查询处理与意图检测
  • 对流程组、处理器、连接和模板的操作
  • NiFi 组件的流程控制(启动/停止/状态)
  • 对所有 NiFi 组件类型的搜索能力
  • 处理器和其他组件的文档访问
  • Streamlit 聊天界面,便于交互
  • 配置 NiFi 连接和认证的选项

适用场景

  • 通过自然语言查询与 Apache NiFi 交互
  • 生成和管理 NiFi 流程
  • 监控和控制 NiFi 组件的状态
  • 搜索和访问 NiFi 组件的文档

使用教程

使用依赖

  • Python 3.8 或更高版本
  • Apache NiFi 实例

安装教程

  1. 克隆仓库
  2. 创建并激活虚拟环境
  3. 安装依赖:pip install -r requirements.txt
  4. 配置:cp config.example.yaml config.yaml

调试方式

快速启动:

python run.py

选项:

  • --server-only: 仅运行服务器组件
  • --ui-only: 仅运行 UI 组件
  • --server-host: 指定服务器主机
  • --server-port: 指定服务器端口
  • --ui-port: 指定 UI 端口
  • --log-level: 设置日志级别(debug, info, warning, error, critical)

手动启动: 启动服务器:python -m uvicorn nifi_mcp_server.server:app --host 0.0.0.0 --port 8000

启动聊天 UI:streamlit run nifi_chat_ui/app.py