Aws Kb Retrieval Server

Aws Kb Retrieval Server

这是一个用于通过Bedrock代理运行时从AWS知识库检索信息的MCP Server实现。

简介

此 MCP 服务器通过 Bedrock Agent Runtime 从 AWS 知识库中检索信息,支持 RAG(检索增强生成)功能。

主要功能

  • 根据查询从 AWS 知识库中获取上下文
  • 支持自定义检索结果数量

快速设置

1. 准备 AWS 凭证

确保你有以下信息:

  • AWS 访问密钥 ID
  • AWS 秘密访问密钥
  • AWS 区域
  • 足够的权限执行 Bedrock Agent Runtime 操作

2. 配置 Claude Desktop

将以下配置添加到 claude_desktop_config.json 文件:

选项 1: 通过 Docker 使用

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "docker",
      "args": [ "run", "-i", "--rm", "-e", "AWS_ACCESS_KEY_ID", "-e", "AWS_SECRET_ACCESS_KEY", "-e", "AWS_REGION", "mcp/aws-kb-retrieval-server" ],
      "env": {
        "AWS_ACCESS_KEY_ID": "您的访问密钥",
        "AWS_SECRET_ACCESS_KEY": "您的秘密访问密钥",
        "AWS_REGION": "您的 AWS 区域"
      }
    }
  }
}

选项 2: 通过 NPX 使用

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "npx",
      "args": [
        "-y",
        "@modelcontextprotocol/server-aws-kb-retrieval"
      ],
      "env": {
        "AWS_ACCESS_KEY_ID": "您的访问密钥",
        "AWS_SECRET_ACCESS_KEY": "您的秘密访问密钥",
        "AWS_REGION": "您的 AWS 区域"
      }
    }
  }
}

3. 使用方式

服务器提供以下工具:

  • retrieve_from_aws_kb
    • 参数:
      • query (字符串): 检索查询
      • knowledgeBaseId (字符串): AWS 知识库 ID
      • n (数字, 可选): 检索结果数量 (默认: 3)

构建自己的镜像 (可选)

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

许可证

MIT 许可证