AWS Knowledge Base Retrieval MCP Server

AWS Knowledge Base Retrieval MCP Server

site icon
2025.03.13 0
JavaScriptAWS 知识库检索检索增强生成搜索工具开发效率
mcp-aws-kb 是一个基于 AWS Knowledge Base 的检索服务,通过 Bedrock Agent Runtime 实现信息的检索和生成。该服务支持检索增强生成(RAG)技术,能够从 AWS Knowledge Base 中获取上下文信息,并支持多结果检索。适用于需要快速获取 AWS 知识库信息的场景,如开发支持、技术文档查询等。
View on GitHub

Overview

基本能力

产品定位

mcp-aws-kb 是一个用于从 AWS Knowledge Base 检索信息的 MCP 服务器实现,主要用于开发支持和技术文档查询。

核心功能

  • RAG (Retrieval-Augmented Generation): 基于查询和知识库 ID 从 AWS Knowledge Base 检索上下文信息。
  • 多结果检索: 支持自定义检索结果数量。

适用场景

  • 开发支持:快速获取 AWS 相关技术文档。
  • 技术文档查询:从知识库中检索特定问题的解决方案。

工具列表

  • retrieve_from_aws_kb
  • 功能:使用 AWS Knowledge Base 执行检索操作。
  • 输入参数:
    • query (string): 检索查询。
    • knowledgeBaseId (string): AWS Knowledge Base 的 ID。
    • n (number, 可选): 检索结果数量(默认: 3)。
  • 响应格式:
    • 文本内容:知识库的原始上下文。
    • JSON 内容:结构化的 RAG 源数据(包含 id、fileName、snippet 和 score)。

常见问题解答

  • 如何配置 AWS 凭证?
  • 支持 IAM 访问密钥和 AWS SSO 两种方式。
  • 如何设置默认知识库 ID?
  • 通过环境变量 AWS_KB_IDS 配置。

使用教程

使用依赖

  1. 确保已安装 Docker 或 Node.js。
  2. 获取 AWS 访问凭证(IAM 或 SSO)。

安装教程

Docker 安装

docker build -t mcp/aws-kb-retrieval -f src/aws-kb-retrieval-server/Dockerfile .

NPX 安装

npx -y @modelcontextprotocol/server-aws-kb-retrieval

本地仓库安装

node /path/to/mcp-aws-kb/dist/index.js

调试方式

  1. 确保环境变量已正确设置(如 AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION)。
  2. 使用 retrieve_from_aws_kb 工具测试检索功能。
{
  "query": "AWS Lambda 最佳实践",
  "knowledgeBaseId": "kb-12345",
  "n": 3
}

许可证

该项目遵循 MIT 开源许可条款。