mcp-rag-server

一个基于模型上下文协议（MCP）的服务器，支持检索增强生成（RAG）。它能够索引文档，并通过MCP协议为大型语言模型提供相关信息。

集成示例

通用 MCP 客户端配置

{
  "mcpServers": {
    "rag": {
      "command": "npx",
      "args": ["-y", "mcp-rag-server"],
      "env": {
        "BASE_LLM_API": "http://localhost:11434/v1",
        "EMBEDDING_MODEL": "nomic-embed-text",
        "VECTOR_STORE_PATH": "./vector_store",
        "CHUNK_SIZE": "500"
      }
    }
  }
}

示例交互

# 索引文档
>> tool:embedding_documents {"path":"./docs"}

# 检查状态
>> resource:embedding-status

<< rag://embedding/status
当前完成数：123
失败数：45
总数量：678

特性

• 文档索引：读取文件，根据 CHUNK_SIZE 分割文本，并进行排队以供嵌入处理。
• 嵌入处理：按顺序将每个分块发送到嵌入 API，结果存储在 SQLite 数据库中。
• 查询功能：对查询内容进行嵌入处理，从向量存储库中检索最相关的文本分块，并将其返回给客户端。

安装

npm install mcp-rag-server

快速开始

1. 安装依赖

   npm install
   

2. 启动服务器

   npm start
   

3. 使用示例命令进行交互

   node example.js
   

配置选项

使用方法

MCP 工具

# 索引文档
>> tool:embedding_documents {"path":"./docs"}

# 查询文档内容
>> tool:query_document {"query":"查询内容", "numberOfChunks":3}

# 获取所有文档 URI
>> resource:rag://documents

# 获取单个文档内容
>> resource:rag://document/path/to/document.txt

MCP 资源

• rag://documents：列出所有文档的 URI。
• rag://document/{path}：获取指定路径的文档内容。
• rag://query-document/{numberOfChunks}/{query}：按数量获取与查询相关的文档内容。
• rag://embedding/status：查看当前索引状态（已完成、失败、总数）。

开发

npm install
npm run build      # 编译 TypeScript 代码
npm start          # 启动服务器
npm run watch      # 监控文件变化

贡献

欢迎贡献！请在 GitHub 上打开问题或拉取请求。

许可证

MIT 2025 Quan Le