AWS Bedrock KB Retrieval

MCP-сервер Amazon Bedrock Knowledge Base Retrieval

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

Возможности

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

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

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

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

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

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

Агентное извлечение в управляемых базах знаний

• Планирование многошаговой стратегии извлечения и синтез ответа с цитированием
• Поиск по нескольким базам знаний за один вызов
• Необязательная сжатая трассировка этапов планирования и извлечения агента
• Только для управляемых баз знаний; инструмент отклоняет другие типы с понятным сообщением

Доступ к контенту, защищённому ACL

• Передайте user_id, чтобы получить контент из источников данных с поддержкой ACL (SharePoint, OneDrive, Confluence с постраничными ACL)
• Без этого такой контент недоступен, а этап расширения полного документа при агентном извлечении завершается ошибкой «UserContext is required for ACL-aware data sources»
• Результаты фильтруются в соответствии с правами доступа пользователя

Поддержка управляемых и векторных баз знаний

• Работает с векторными базами знаний (type: VECTOR) и управляемыми базами знаний (type: MANAGED)
• Тип базы знаний определяется автоматически, и отправляется правильная конфигурация Retrieve (vectorSearchConfiguration или managedSearchConfiguration)
• Фильтрация источников данных использует ключ метаданных, соответствующий типу базы знаний
• Инструмент ListKnowledgeBases сообщает type каждой базы знаний

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

• Повышение релевантности результатов поиска
• Использование возможностей переранжирования 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. Доступность переранжирования различается в зависимости от модели: amazon.rerank-v1:0 не предлагается в us-east-1, а cohere.rerank-v3-5:0 предлагается. Сервер проверяет пару (регион, модель) и быстро завершает работу с понятным сообщением. Обратитесь к официальной документации за актуальным списком поддерживаемых регионов.
4. Включите доступ к моделям для доступных моделей переранжирования в указанном регионе.

Требования к агентному извлечению

Инструмент AgenticQueryKnowledgeBases вызывает AgenticRetrieveStream, который поддерживается только для управляемых баз знаний (type: MANAGED). Он планирует стратегию извлечения и, если вы не передадите generate_response=false, вызывает фундаментальную модель для написания ответа с цитированием.

1. Вашей роли IAM требуется bedrock:AgenticRetrieveStream на базе знаний в дополнение к разрешениям, перечисленным выше
2. Поскольку он вызывает фундаментальную модель, каждый вызов стоит значительно дороже, чем QueryKnowledgeBases. Передайте generate_response=false для извлечения без синтеза
3. RetrieveAndGenerate не поддерживается для управляемых баз знаний, поэтому агентное извлечение с generate_response=true — это способ получить сгенерированный ответ из такой базы

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

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

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

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

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

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

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

• Создание базы знаний
• Управление разрешениями для баз знаний Amazon Bedrock
• Разрешения для переранжирования в Amazon Bedrock

Установка

Kiro	Cursor	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 и доступен только в определённых регионах.