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

Servidor MCP para acceder a las bases de conocimiento de Amazon Bedrock

Características

Descubrir bases de conocimiento y sus fuentes de datos

• Buscar y explorar todas las bases de conocimiento disponibles
• Búsqueda de bases de conocimiento por nombre o etiqueta
• Enumerar las fuentes de datos asociadas a cada base de conocimientos

Consulta de bases de conocimientos con lenguaje natural

• Recuperar información mediante consultas conversacionales
• Obtener pasajes relevantes de las bases de conocimientos
• Acceda a la información sobre citas de todos los resultados

Filtrar los resultados por fuente de datos

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

Volver a clasificar los resultados

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

Requisitos previos

Requisitos de instalación

1. Instale uv desde Astral o desde el LÉEME de GitHub
2. Instalar Python usando uv python install 3.10

Requisitos de AWS

1. Configuración de la CLI de AWS: Debes tener la CLI de AWS configurada con credenciales y un AWS_PROFILE que tenga acceso a Amazon Bedrock y Knowledge Bases
2. Base de conocimientos deAmazon 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 IAM: Su rol/usuario IAM debe tener los permisos apropiados para
   • Enumerar y describir bases de conocimiento
   • Acceder a fuentes de datos
   • Consultar bases de conocimiento

Requisitos de renumeración

Si tiene intención de utilizar la funcionalidad de renumeración, su base de conocimiento de Bedrock necesita permisos adicionales:

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

Control de la reevaluación

La reordenación puede activarse o desactivarse globalmente mediante la variable de entorno BEDROCK_KB_RERANKING_ENABLED:

• Establecer en false (por defecto): Desactiva la reordenación para todas las consultas a menos que se active explícitamente
• Definida como true: activa la reordenación para todas las consultas a menos que se desactive explícitamente

La variable de entorno acepta varios formatos:

• Para activar: "true", "1", "yes" o "on" (no distingue mayúsculas de minúsculas)
• Para desactivar: cualquier otro valor o no establecer (comportamiento por defecto)

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

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

• Crear una base de conocimientos
• Administración de permisos para las bases de conocimientos de Amazon Bedrock
• Permisos para reranking en Amazon Bedrock

Instalación

Configure el servidor MCP en su configuración de cliente MCP (por ejemplo, para Amazon Q Developer CLI, edite ~/.aws/amazonq/mcp.json):

{ "mcpServers": { "awslabs.bedrock-kb-retrieval-mcp-server": { "command": "uvx", "args": ["awslabs.bedrock-kb-retrieval-mcp-server@latest"], "env": {"AWS_PROFILE": "tu-nombre-de-perfil", "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": [] } } }

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

# archivo `.env` ficticio con credenciales temporales de AWS 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 la consulta KB.
• El parámetro reranking requiere permisos adicionales, acceso al modelo de Amazon Bedrock y sólo está disponible en regiones específicas.