MCP Pinecone Vector Database Server

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：字符串（搜索查询文本）
   • topK：数字（1-100，默认值：5）
   • filter：对象（可选筛选条件）

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

   • text：字符串（要向量化的内容）
   • metadata：对象（向量元数据）
   • id：字符串（可选自定义 ID）

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

   • filePath：字符串（JSON 文件路径）
   • namespace：字符串（可选，默认为 "capella-document-search"）

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

   • ids：字符串[]（向量 ID 列表）
   • namespace：字符串（可选，默认为 "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. 提交拉取请求

许可证

MIT