技术文档摘要

AWS 知识库检索 MCP 服务器

这是一个使用 Bedrock Agent 运行时从 AWS 知识库检索信息的 MCP 服务器实现。

特性

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

工具

• retrieve_from_aws_kb
  • 使用 AWS 知识库执行检索操作。
  • 输入参数：
    • query（字符串）：用于检索的搜索查询。
    • knowledgeBaseId（字符串）：AWS 知识库的ID。
    • n（数字，可选）：要检索的结果数量（默认值：3）。

配置

设置 AWS 凭证

1. 从 AWS 管理控制台获取 AWS 访问密钥ID、秘密访问密钥和区域。
2. 确保这些凭证具有进行 Bedrock Agent 运行时操作所需的适当权限。

在 Claude Desktop 中使用

在你的 claude_desktop_config.json 文件中添加以下内容：

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": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE"
      }
    }
  }
}

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "npx",
      "args": ["-y", "@modelcontextprotocol/server-aws-kb-retrieval"],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE"
      }
    }
  }
}

在 VS Code 中使用

为了快速安装，请使用下面的一个一键安装按钮...

手动安装

对于手动安装，在 VS Code 的用户设置（JSON）文件中添加以下 JSON 块。你可以通过按 Ctrl + Shift + P 并输入 Preferences: Open Settings (JSON) 来完成此操作。

可选地，你可以将其添加到工作区中的一个名为 .vscode/mcp.json 的文件中。这将允许你与他人共享配置。

注意：.vscode/mcp.json 文件中不需要 mcp 键。

{
  "mcp": {
    "inputs": [
      {
        "type": "promptString",
        "id": "aws_access_key",
        "description": "AWS Access Key ID",
        "password": true
      },
      {
        "type": "promptString",
        "id": "aws_secret_key",
        "description": "AWS Secret Access Key",
        "password": true
      },
      {
        "type": "promptString",
        "id": "aws_region",
        "description": "AWS Region"
      }
    ],
    "servers": {
      "aws-kb-retrieval": {
        "command": "npx",
        "args": ["-y", "@modelcontextprotocol/server-aws-kb-retrieval"],
        "env": {
          "AWS_ACCESS_KEY_ID": "${input:aws_access_key}",
          "AWS_SECRET_ACCESS_KEY": "${input:aws_secret_key}",
          "AWS_REGION": "${input:aws_region}"
        }
      }
    }
  }
}

对于 Docker 安装：

{
  "mcp": {
    "inputs": [
      {
        "type": "promptString",
        "id": "aws_access_key",
        "description": "AWS Access Key ID",
        "password": true
      },
      {
        "type": "promptString",
        "id": "aws_secret_key",
        "description": "AWS Secret Access Key",
        "password": true
      },
      {
        "type": "promptString",
        "id": "aws_region",
        "description": "AWS Region"
      }
    ],
    "servers": {
      "aws-kb-retrieval": {
        "command": "docker",
        "args": ["run", "-i", "--rm", "mcp/aws-kb-retrieval-server"],
        "env": {
          "AWS_ACCESS_KEY_ID": "${input:aws_access_key}",
          "AWS_SECRET_ACCESS_KEY": "${input:aws_secret_key}",
          "AWS_REGION": "${input:aws_region}"
        }
      }
    }
  }
}

构建

Docker：

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

许可证

此 MCP 服务器采用 MIT 许可证。这意味着你可以自由地使用、修改和分发该软件，但需遵守 MIT 许可证的条款和条件。更多详情，请参阅项目存储库中的 LICENSE 文件。