AWS 知识库检索 MCP 服务器

一个 MCP 服务器实现，用于使用 Bedrock Agent Runtime 从 AWS 知识库检索信息。

功能

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

工具

• retrieve_from_aws_kb
  • 使用 AWS 知识库执行检索操作。
  • 输入：
    • query (字符串)：用于检索的搜索查询。
    • knowledgeBaseId (字符串)：AWS 知识库的 ID。
    • n (数字，可选)：要检索的结果数量（默认值：3）。
  • 响应格式：
    • 响应现在返回两个单独的内容项：
      • 一个文本项，包含来自知识库的原始上下文。
      • 一个 JSON 项，包含带有元数据的结构化 RAG 来源（id、fileName、snippet 和 score）。
    • 这种分离允许更灵活地处理结果。

配置

设置 AWS 凭证

您有两种配置 AWS 凭证的选项：

选项 1：IAM 访问密钥

1. 从 AWS 管理控制台获取 AWS 访问密钥 ID、秘密访问密钥和区域。
2. 确保这些凭证具有 Bedrock Agent Runtime 操作的适当权限。
3. 如下面的配置示例所示，设置环境变量。
4. 对于临时凭证，您还可以使用 AWS_SESSION_TOKEN 环境变量提供会话令牌。

选项 2：AWS SSO（单点登录）

服务器现在支持 AWS SSO 凭证：

1. 使用您的 SSO 配置文件配置 AWS CLI：aws configure sso
2. 仅在 MCP 服务器配置中设置 AWS_REGION 环境变量。
3. 服务器将使用默认的凭证提供程序链，其中包括 SSO 凭证。

可选：配置默认知识库 ID

您可以选择指定一个或多个要默认使用的知识库 ID：

1. 创建一个 JSON 格式的知识库 ID 数组。
2. 将其设置为配置中的 AWS_KB_IDS 环境变量。
3. 配置此项后，knowledgeBaseId 参数在工具中变为可选。

与 Claude Desktop 一起使用

将其添加到您的 claude_desktop_config.json：

Docker with IAM Access Keys

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "docker",
      "args": [ "run", "-i", "--rm", "-e", "AWS_ACCESS_KEY_ID", "-e", "AWS_SECRET_ACCESS_KEY", "-e", "AWS_REGION", "-e", "AWS_KB_IDS", "mcp/aws-kb-retrieval-server" ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

Docker with AWS SSO

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "docker",
      "args": [ "run", "-i", "--rm", "-e", "AWS_REGION", "-e", "AWS_KB_IDS", "-v", "${HOME}/.aws:/root/.aws", "mcp/aws-kb-retrieval-server" ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

NPX with IAM Access Keys

{
  "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_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

NPX with AWS SSO

{
  "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_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

Local Repository (from cloned/built repo)

{
  "mcpServers": {
    "aws-kb": {
      "command": "node",
      "args": [
        "/path/to/mcp-aws-kb/dist/index.js"
      ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      },
      "disabled": false,
      "autoApprove": [
        "retrieve_from_aws_kb"
      ],
      "timeout": 120
    }
  }
}

构建

Docker:

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

许可证

此 MCP 服务器在 MIT 许可证下获得许可。 这意味着您可以自由使用、修改和分发该软件，但须遵守 MIT 许可证的条款和条件。 有关更多详细信息，请参阅项目存储库中的 LICENSE 文件。

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