AWS Bedrock KB Retrieval

Сервер 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.

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

• Создание базы знаний
• Управление разрешениями для баз знаний 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 и доступен только в определенных регионах.