AWS知识库检索MCP服务器

一个用于从AWS知识库中检索信息的MCP服务器实现，使用Bedrock代理运行时。

功能

• RAG（基于检索的增强生成）：根据查询和知识库ID从AWS知识库中检索上下文。
• 支持多结果检索：可以选择检索自定义数量的结果。

工具

• retrieve_from_aws_kb
  • 执行使用AWS知识库的检索操作。
  • 输入：
    • query（字符串）：检索查询。
    • knowledgeBaseId（字符串）：AWS知识库ID。
    • n（数字，可选）：要检索的结果数量，默认为10。
  • 输出：
    • 返回与查询相关的上下文段落列表，每个结果包含相关性评分。

配置

AWS凭证配置

使用IAM访问密钥：

在环境变量中设置以下参数：

AWS_ACCESS_KEY_ID=your_access_key_id
AWS_SECRET_ACCESS_KEY=your_secret_access_key

使用AWS SSO：

在环境变量中设置以下参数：

AWS_PROFILE=your_sso_profile_name

知识库配置

在环境变量中设置知识库ID：

AWS_KB_ID=your_knowledge_base_id

示例使用

从本地存储的JSON文件加载知识库

将知识库内容存储为knowledge.json，然后使用以下命令进行检索：

node index.js --file knowledge.json --query "你的查询问题"

使用Docker运行服务器

构建镜像：

docker build -t mcp-aws-kb .

运行容器并挂载知识库文件夹：

docker run -it --mount type=bind,source=/path/to/knowledge,target=/app/knowledge mcp-aws-kb

本地开发

安装依赖：

npm install

启动开发服务器：

npm start

许可证

本MCP服务器根据MIT许可证发布。这意味着您可以自由使用、修改和分发软件，前提是您遵守MIT许可证的条款和条件。有关详细信息，请参阅项目存储库中的LICENSE文件。

此README假设您的服务器包名为@modelcontextprotocol/server-aws-kb-retrieval。如果您的设置不同，请调整包名和安装详细信息。同时，请确保您的服务器脚本正确构建，并且所有依赖项在package.json中正确管理。