AWS Bedrock KB Retrieval MCP Server

официальный

Запрашивайте базы знаний Amazon Bedrock на естественном языке для получения релевантной информации из ваших источников данных.

GitHub
9.3k

Документация

Сервер MCP для извлечения данных из базы знаний Amazon Bedrock

Сервер MCP для доступа к базам знаний Amazon Bedrock

Возможности

Обнаружение баз знаний и их источников данных

  • Поиск и изучение всех доступных баз знаний
  • Поиск баз знаний по имени или тегу
  • Список источников данных, связанных с каждой базой знаний

Запросы к базам знаний на естественном языке

  • Извлечение информации с помощью диалоговых запросов
  • Получение релевантных отрывков из ваших баз знаний
  • Доступ к информации о цитировании для всех результатов

Фильтрация результатов по источнику данных

  • Фокусировка запросов на конкретных источниках данных
  • Включение или исключение определенных источников данных
  • Приоритизация результатов из конкретных источников данных

Ранжирование результатов

  • Повышение релевантности результатов извлечения
  • Использование возможностей ранжирования Amazon Bedrock
  • Сортировка результатов по релевантности вашему запросу

Предварительные требования

Требования к установке

  1. Установите uv с Astral или из README на GitHub
  2. Установите Python с помощью uv python install 3.10

Требования AWS

  1. Настройка AWS CLI: У вас должен быть настроен AWS CLI с учетными данными и AWS_PROFILE, имеющим доступ к Amazon Bedrock и базам знаний
  2. База знаний Amazon Bedrock: У вас должна быть хотя бы одна база знаний Amazon Bedrock с ключом тега mcp-multirag-kb и значением true
  3. Разрешения IAM: Ваша роль/пользователь IAM должны иметь соответствующие разрешения для:
    • Получения списка и описания баз знаний
    • Доступа к источникам данных
    • Выполнения запросов к базам знаний

Требования к ранжированию

Если вы планируете использовать функцию ранжирования, вашей базе знаний Bedrock потребуются дополнительные разрешения:

  1. Ваша роль IAM должна иметь разрешения на действия bedrock:Rerank и bedrock:InvokeModel
  2. Сервисная роль Amazon Bedrock Knowledge Bases также должна иметь эти разрешения
  3. Ранжирование доступно только в определенных регионах. Актуальный список поддерживаемых регионов смотрите в официальной документации.
  4. Включите доступ к моделям для доступных моделей ранжирования в указанном регионе.

Управление ранжированием

Ранжирование можно глобально включить или отключить с помощью переменной окружения BEDROCK_KB_RERANKING_ENABLED:

  • Установите значение false (по умолчанию): отключает ранжирование для всех запросов, если оно не включено явно
  • Установите значение true: включает ранжирование для всех запросов, если оно не отключено явно

Переменная окружения принимает различные форматы:

  • Для включения: 'true', '1', 'yes' или 'on' (без учета регистра)
  • Для отключения: любое другое значение или если переменная не установлена (поведение по умолчанию)

Эта настройка задает глобальное значение по умолчанию, в то время как отдельные вызовы API могут переопределить его, явно указав параметр reranking.

Подробные инструкции по настройке баз знаний см. в:

Установка

KiroCursorVS Code
Add to KiroInstall MCP ServerInstall on VS Code

Настройте сервер 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 не включаются в ответ на запрос к базе знаний.
  • Параметр reranking требует дополнительных разрешений, доступа к модели Amazon Bedrock и доступен только в определенных регионах.