AWS Bedrock KB Retrieval

Servidor MCP de Recuperación de Bases de Conocimiento de Amazon Bedrock

Servidor MCP para acceder a Bases de Conocimiento de Amazon Bedrock

Funcionalidades

Descubrir bases de conocimiento y sus fuentes de datos

• Encontrar y explorar todas las bases de conocimiento disponibles
• Buscar bases de conocimiento por nombre o etiqueta
• Listar las fuentes de datos asociadas a cada base de conocimiento

Consultar bases de conocimiento con lenguaje natural

• Recuperar información mediante consultas conversacionales
• Obtener pasajes relevantes de sus bases de conocimiento
• Acceder a información de citas para todos los resultados

Filtrar resultados por fuente de datos

• Enfocar sus consultas en fuentes de datos específicas
• Incluir o excluir fuentes de datos específicas
• Priorizar resultados de fuentes de datos específicas

Reordenar resultados

• Mejorar la relevancia de los resultados de recuperación
• Usar las capacidades de reordenamiento de Amazon Bedrock
• Ordenar resultados por relevancia para su consulta

Requisitos previos

Requisitos de instalación

1. Instalar uv desde Astral o el README de GitHub
2. Instalar Python usando uv python install 3.10

Requisitos de AWS

1. Configuración de AWS CLI: Debe tener la AWS CLI configurada con credenciales y un AWS_PROFILE que tenga acceso a Amazon Bedrock y Bases de Conocimiento
2. Base de Conocimiento de Amazon Bedrock: Debe tener al menos una Base de Conocimiento de Amazon Bedrock con la clave de etiqueta mcp-multirag-kb con un valor de true
3. Permisos de IAM: Su rol/usuario de IAM debe tener los permisos adecuados para:
   • Listar y describir bases de conocimiento
   • Acceder a fuentes de datos
   • Consultar bases de conocimiento

Requisitos de reordenamiento

Si tiene la intención de usar la funcionalidad de reordenamiento, su Base de Conocimiento de Bedrock necesita permisos adicionales:

1. Su rol de IAM debe tener permisos para las acciones bedrock:Rerank y bedrock:InvokeModel
2. El rol de servicio de Bases de Conocimiento de Amazon Bedrock también debe tener estos permisos
3. El reordenamiento solo está disponible en regiones específicas. Consulte la documentación oficial para obtener una lista actualizada de regiones compatibles.
4. Habilite el acceso al modelo para los modelos de reordenamiento disponibles en la región especificada.

Control del reordenamiento

El reordenamiento se puede habilitar o deshabilitar globalmente usando la variable de entorno BEDROCK_KB_RERANKING_ENABLED:

• Establecer en false (predeterminado): Deshabilita el reordenamiento para todas las consultas a menos que se habilite explícitamente
• Establecer en true: Habilita el reordenamiento para todas las consultas a menos que se deshabilite explícitamente

La variable de entorno acepta varios formatos:

• Para habilitar: 'true', '1', 'yes' u 'on' (sin distinción de mayúsculas/minúsculas)
• Para deshabilitar: cualquier otro valor o no establecida (comportamiento predeterminado)

Esta configuración proporciona un valor predeterminado global, mientras que las llamadas API individuales aún pueden anularlo estableciendo explícitamente el parámetro reranking.

Para obtener instrucciones detalladas sobre la configuración de bases de conocimiento, consulte:

• Crear una base de conocimiento
• Administrar permisos para bases de conocimiento de Amazon Bedrock
• Permisos para el reordenamiento en Amazon Bedrock

Instalación

Kiro	Cursor	VS Code

Configure el servidor MCP en la configuración de su cliente MCP (por ejemplo, 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": []
    }
  }
}

Instalación en Windows

Para usuarios de Windows, el formato de configuración del servidor MCP es ligeramente 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"
      }
    }
  }
}

o docker después de un docker build -t awslabs/bedrock-kb-retrieval-mcp-server . exitoso:

# 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: Sus credenciales deberán mantenerse actualizadas desde su host

Limitaciones

• Los resultados con tipo de contenido IMAGE no se incluyen en la respuesta de consulta de KB.
• El parámetro reranking requiere permisos adicionales, acceso al modelo de Amazon Bedrock y solo está disponible en regiones específicas.