MCP Pinecone 向量数据库服务器

本项目实现了一个模型上下文协议 (MCP) 服务器，允许向 Pinecone 向量数据库读写向量化信息。它被设计用于处理 RAG 处理过的 PDF 数据和 Confluence 数据。

功能

• 使用文本查询搜索相似文档
• 向数据库添加带有自定义元数据的新向量
• 批量处理和上传 Confluence 数据
• 通过 ID 删除向量
• 基本数据库统计信息（暂时禁用）

前提条件

• Bun 运行时
• Pinecone API 密钥
• OpenAI API 密钥（用于生成嵌入）

安装

1. 克隆此仓库

2. 安装依赖项：

   bun install
   

3. 创建一个 .env 文件，内容如下：

   PINECONE_API_KEY=your-pinecone-api-key
   OPENAI_API_KEY=your-openai-api-key
   PINECONE_HOST=your-pinecone-host
   PINECONE_INDEX_NAME=your-index-name
   DEFAULT_NAMESPACE=your-namespace
   

用法

运行 MCP 服务器

启动服务器：

bun src/index.ts

服务器将启动并通过 stdio 监听 MCP 命令。

运行示例客户端

使用示例客户端测试服务器：

bun examples/client.ts

处理 Confluence 数据

Confluence 处理脚本提供详细的日志记录和验证：

bun src/scripts/process-confluence.ts <file-path> [collection] [scope]

参数：

• file-path: Confluence JSON 文件的路径（必需）
• collection: 文档集合名称（默认为 "documentation"）
• scope: 文档范围（默认为 "documentation"）

示例：

bun src/scripts/process-confluence.ts ./data/confluence-export.json "tech-docs" "engineering"

该脚本将：

1. 验证输入参数
2. 处理和向量化内容
3. 批量上传向量
4. 验证上传成功
5. 提供该过程的详细日志

可用工具

服务器提供以下工具：

1. search-vectors - 搜索相似文档，参数：

   • query: string (搜索查询文本)
   • topK: number (1-100，默认值：5)
   • filter: object (可选的过滤条件)

2. add-vector - 添加单个文档，参数：

   • text: string (要向量化的内容)
   • metadata: object (向量元数据)
   • id: string (可选的自定义 ID)

3. process-confluence - 处理 Confluence JSON 数据，参数：

   • filePath: string (JSON 文件的路径)
   • namespace: string (可选，默认为 "capella-document-search")

4. delete-vectors - 删除向量，参数：

   • ids: string[] (向量 ID 列表)
   • namespace: string (可选，默认为 "capella-document-search")

5. get-stats - 获取数据库统计信息（暂时禁用）

数据库配置

服务器需要一个 Pinecone 向量数据库。在您的 .env 文件中配置连接详细信息：

PINECONE_API_KEY=your-api-key
PINECONE_HOST=your-host
PINECONE_INDEX_NAME=your-index
DEFAULT_NAMESPACE=your-namespace

元数据模式

Confluence 文档

ID: confluence-[page-id]-[item-id]
title: [title]
pageId: [page-id]
spaceKey: [space-key]
type: [type]
content: [text-content]
author: [author-name]
source: "confluence"
collection: "documentation"
scope: "documentation"
...

贡献

1. Fork 该仓库
2. 创建您的功能分支：git checkout -b feature/my-new-feature
3. 提交您的更改：git commit -am 'Add some feature'
4. 推送到该分支：git push origin feature/my-new-feature
5. 提交一个 pull request

许可证

MIT