Bedrock Kb Retrieval MCP Server

亚马逊基岩知识库检索MCP服务器

用于访问亚马逊基岩知识库的MCP服务器

特性

发现知识库及其数据源

• 查找并探索所有可用的知识库
• 按名称或标签搜索知识库
• 列出与每个知识库关联的数据源

用自然语言查询知识库

• 使用会话查询检索信息
• 从你的知识库中获取相关段落
• 访问所有结果的引用信息

按数据源筛选结果

• 将您的查询集中在特定的数据源上
• 包含或排除特定数据源
• 优先考虑特定数据源的结果

重新排列结果

• 提高检索结果的相关性
• 使用Amazon Bedrock重新分级功能
• 按与查询的相关性对结果进行排序

先决条件

安装要求

1. 安装 uv 从 星体 或
2. 使用以下命令安装Python uv python install 3.10

AWS要求

1. AWS CLI配置：您必须将AWS CLI配置为具有凭据和AWS_PROFILE，该AWS_PROFILE可以访问Amazon Bedrock和知识库
2. 亚马逊基岩知识库：您必须至少有一个带有标签密钥的亚马逊基岩知识库 mcp-multirag-kb 值为 true
3. IAM权限：您的IAM角色/用户必须具有适当的权限：

- 列出并描述知识库 - 访问数据源 - 查询知识库

重新排序要求

如果您打算使用重新分级功能，您的基岩知识库需要额外的权限：

1. 您的IAM角色必须同时具有这两个权限 bedrock:Rerank 和 bedrock:InvokeModel 行动
2. 亚马逊基岩知识库服务角色也必须具有这些权限
3. 重新排名仅在特定地区可用。请咨询官方 文档 获取最新的支持区域列表。
4. 为指定区域中的可用重新分级模型启用模型访问。

控制重新排名

可以使用全局启用或禁用重新排名 BEDROCK_KB_RERANKING_ENABLED 环境变量：

• 设置为 false （默认）：除非明确启用，否则禁用所有查询的重新分级
• 设置为 true：启用所有查询的重新分级，除非明确禁用

环境变量接受各种格式：

• 用于启用：“true”、“1”、“yes”或“on”（不区分大小写）
• 用于禁用：任何其他值或未设置（默认行为）

此设置提供全局默认值，而单个API调用仍然可以通过显式设置 reranking 参数。

有关设置知识库的详细说明，请参阅：

• 创建知识库
• 管理亚马逊基岩知识库的权限
• Amazon Bedrock的重新登录权限

安装

在MCP客户端配置中配置MCP服务器（例如，对于Kiro，编辑 ~/.kiro/settings/mcp.json):

{
  "mcpServers": {
    "awslabs.bedrock-kb-retrieval-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.bedrock-kb-retrieval-mcp-server@latest"],
      "env": {
        "AWS_PROFILE": "your-profile-name",
        "AWS_REGION": "us-east-1",
        "FASTMCP_LOG_LEVEL": "ERROR",
        "KB_INCLUSION_TAG_KEY": "optional-tag-key-to-filter-kbs",
        "BEDROCK_KB_RERANKING_ENABLED": "false"
      },
      "disabled": false,
      "autoApprove": []
    }
  }
}

Windows安装

对于Windows用户，MCP服务器配置格式略有不同：

{
  "mcpServers": {
    "awslabs.bedrock-kb-retrieval-mcp-server": {
      "disabled": false,
      "timeout": 60,
      "type": "stdio",
      "command": "uv",
      "args": [
        "tool",
        "run",
        "--from",
        "awslabs.bedrock-kb-retrieval-mcp-server@latest",
        "awslabs.bedrock-kb-retrieval-mcp-server.exe"
      ],
      "env": {
        "FASTMCP_LOG_LEVEL": "ERROR",
        "AWS_PROFILE": "your-aws-profile",
        "AWS_REGION": "us-east-1"
      }
    }
  }
}

或docker成功后 docker build -t awslabs/bedrock-kb-retrieval-mcp-server .:

# fictitious `.env` file with AWS temporary credentials
AWS_ACCESS_KEY_ID=ASIAIOSFODNN7EXAMPLE
AWS_SECRET_ACCESS_KEY=wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY
AWS_SESSION_TOKEN=AQoEXAMPLEH4aoAH0gNCAPy...truncated...zrkuWJOgQs8IZZaIv2BXIa2R4Olgk

  {
    "mcpServers": {
      "awslabs.bedrock-kb-retrieval-mcp-server": {
        "command": "docker",
        "args": [
          "run",
          "--rm",
          "--interactive",
          "--env",
          "FASTMCP_LOG_LEVEL=ERROR",
          "--env",
          "KB_INCLUSION_TAG_KEY=optional-tag-key-to-filter-kbs",
          "--env",
          "BEDROCK_KB_RERANKING_ENABLED=false",
          "--env",
          "AWS_REGION=us-east-1",
          "--env-file",
          "/full/path/to/file/above/.env",
          "awslabs/bedrock-kb-retrieval-mcp-server:latest"
        ],
        "env": {},
        "disabled": false,
        "autoApprove": []
      }
    }
  }

注意：您的凭据需要从主机刷新

局限性

• 结果与 IMAGE KB查询响应中不包括内容类型。
• 这 reranking 参数需要额外的权限，亚马逊基岩模型访问权限，并且仅在特定地区可用。