AWS Bedrock KB Retrieval

Servidor MCP de Recuperação de Base de Conhecimento do Amazon Bedrock

Servidor MCP para acessar Bases de Conhecimento do Amazon Bedrock

Funcionalidades

Descobrir bases de conhecimento e suas fontes de dados

• Encontrar e explorar todas as bases de conhecimento disponíveis
• Pesquisar bases de conhecimento por nome ou tag
• Listar fontes de dados associadas a cada base de conhecimento

Consultar bases de conhecimento com linguagem natural

• Recuperar informações usando consultas conversacionais
• Obter trechos relevantes de suas bases de conhecimento
• Acessar informações de citação para todos os resultados

Filtrar resultados por fonte de dados

• Focar suas consultas em fontes de dados específicas
• Incluir ou excluir fontes de dados específicas
• Priorizar resultados de fontes de dados específicas

Reordenar resultados

• Melhorar a relevância dos resultados de recuperação
• Usar capacidades de reordenação do Amazon Bedrock
• Ordenar resultados por relevância para sua consulta

Pré-requisitos

Requisitos de Instalação

1. Instale o uv da Astral ou do README do GitHub
2. Instale o Python usando uv python install 3.10

Requisitos da AWS

1. Configuração da AWS CLI: Você deve ter a AWS CLI configurada com credenciais e um AWS_PROFILE que tenha acesso ao Amazon Bedrock e às Bases de Conhecimento
2. Base de Conhecimento do Amazon Bedrock: Você deve ter pelo menos uma Base de Conhecimento do Amazon Bedrock com a chave de tag mcp-multirag-kb com o valor true
3. Permissões IAM: Sua função/usuário IAM deve ter permissões apropriadas para:
   • Listar e descrever bases de conhecimento
   • Acessar fontes de dados
   • Consultar bases de conhecimento

Requisitos de Reordenação

Se você pretende usar a funcionalidade de reordenação, sua Base de Conhecimento do Bedrock precisa de permissões adicionais:

1. Sua função IAM deve ter permissões para as ações bedrock:Rerank e bedrock:InvokeModel
2. A função de serviço das Bases de Conhecimento do Amazon Bedrock também deve ter essas permissões
3. A reordenação está disponível apenas em regiões específicas. Consulte a documentação oficial para obter uma lista atualizada das regiões suportadas.
4. Habilite o acesso ao modelo para os modelos de reordenação disponíveis na região especificada.

Controlando a Reordenação

A reordenação pode ser habilitada ou desabilitada globalmente usando a variável de ambiente BEDROCK_KB_RERANKING_ENABLED:

• Definida como false (padrão): Desabilita a reordenação para todas as consultas, a menos que explicitamente habilitada
• Definida como true: Habilita a reordenação para todas as consultas, a menos que explicitamente desabilitada

A variável de ambiente aceita vários formatos:

• Para habilitar: 'true', '1', 'yes' ou 'on' (não diferencia maiúsculas de minúsculas)
• Para desabilitar: qualquer outro valor ou não definida (comportamento padrão)

Esta configuração fornece um padrão global, enquanto chamadas de API individuais ainda podem substituí-la definindo explicitamente o parâmetro reranking.

Para instruções detalhadas sobre como configurar bases de conhecimento, consulte:

• Criar uma base de conhecimento
• Gerenciando permissões para bases de conhecimento do Amazon Bedrock
• Permissões para reordenação no Amazon Bedrock

Instalação

Kiro	Cursor	VS Code

Configure o servidor MCP na configuração do seu cliente MCP (por exemplo, para Kiro, edite ~/.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": []
    }
  }
}

Instalação no Windows

Para usuários do Windows, o formato de configuração do servidor MCP é ligeiramente diferente:

{
  "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"
      }
    }
  }
}

ou docker após um docker build -t awslabs/bedrock-kb-retrieval-mcp-server . bem-sucedido:

# 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": []
      }
    }
  }

NOTA: Suas credenciais precisarão ser mantidas atualizadas a partir do seu host

Limitações

• Resultados com o tipo de conteúdo IMAGE não são incluídos na resposta da consulta KB.
• O parâmetro reranking requer permissões adicionais, acesso ao modelo Amazon Bedrock e está disponível apenas em regiões específicas.