Servidor MCP de recuperación de la base de conocimientos de AWS

Una implementación de servidor MCP para recuperar información de la base de conocimientos de AWS mediante Bedrock Agent Runtime.

Características

• RAG (Recuperación-Generación Aumentada) : recupera el contexto de la Base de conocimiento de AWS en función de una consulta y un ID de la Base de conocimiento.

• Admite la recuperación de múltiples resultados : opción para recuperar una cantidad personalizable de resultados.

Related MCP server: DuckDB-RAG-MCP-Sample

Herramientas

• recuperar_de_aws_kb

  • Realice operaciones de recuperación utilizando la base de conocimientos de AWS.

  • Entradas:

    • query (cadena): la consulta de búsqueda para la recuperación.

    • knowledgeBaseId (cadena): el ID de la base de conocimiento de AWS.

    • n (número, opcional): Número de resultados a recuperar (predeterminado: 3).

  • Formato de respuesta:

    • La respuesta ahora devuelve dos elementos de contenido separados:

      • Un elemento de texto que contiene el contexto sin procesar de la base de conocimientos.

      • Un elemento JSON que contiene las fuentes RAG estructuradas con metadatos (id, nombre de archivo, fragmento y puntuación).

    • Esta separación permite un procesamiento más flexible de los resultados.

Configuración

Configuración de credenciales de AWS

Tiene dos opciones para configurar las credenciales de AWS:

Opción 1: Claves de acceso de IAM

1. Obtenga la ID de clave de acceso de AWS, la clave de acceso secreta y la región desde la consola de administración de AWS.

2. Asegúrese de que estas credenciales tengan los permisos adecuados para las operaciones de Bedrock Agent Runtime.

3. Establezca las variables de entorno como se muestra en los ejemplos de configuración a continuación.

4. Para las credenciales temporales, también puede proporcionar un token de sesión utilizando la variable de entorno AWS_SESSION_TOKEN .

Opción 2: AWS SSO (inicio de sesión único)

El servidor ahora admite las credenciales SSO de AWS:

1. Configure AWS CLI con su perfil SSO: aws configure sso

2. Establezca solo la variable de entorno AWS_REGION en la configuración del servidor MCP.

3. El servidor utilizará la cadena de proveedor de credenciales predeterminada, que incluye credenciales SSO.

Opcional: Configurar los ID de la base de conocimientos predeterminados

Opcionalmente, puede especificar uno o más identificadores de base de conocimiento para usar de manera predeterminada:

1. Cree una matriz de identificadores de bases de conocimiento en formato JSON.

2. Establezca esto como la variable de entorno AWS_KB_IDS en su configuración.

3. Cuando esto está configurado, el parámetro knowledgeBaseId se vuelve opcional en la herramienta.

Uso con Claude Desktop

Agregue esto a su claude_desktop_config.json :

Docker con claves de acceso IAM

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "docker",
      "args": [ "run", "-i", "--rm", "-e", "AWS_ACCESS_KEY_ID", "-e", "AWS_SECRET_ACCESS_KEY", "-e", "AWS_REGION", "-e", "AWS_KB_IDS", "mcp/aws-kb-retrieval-server" ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

Docker con AWS SSO

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "docker",
      "args": [ "run", "-i", "--rm", "-e", "AWS_REGION", "-e", "AWS_KB_IDS", "-v", "${HOME}/.aws:/root/.aws", "mcp/aws-kb-retrieval-server" ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

NPX con claves de acceso IAM

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "npx",
      "args": [
        "-y",
        "@modelcontextprotocol/server-aws-kb-retrieval"
      ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

NPX con AWS SSO

{
  "mcpServers": {
    "aws-kb-retrieval": {
      "command": "npx",
      "args": [
        "-y",
        "@modelcontextprotocol/server-aws-kb-retrieval"
      ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      }
    }
  }
}

Repositorio local (del repositorio clonado/construido)

{
  "mcpServers": {
    "aws-kb": {
      "command": "node",
      "args": [
        "/path/to/mcp-aws-kb/dist/index.js"
      ],
      "env": {
        "AWS_ACCESS_KEY_ID": "YOUR_ACCESS_KEY_HERE",
        "AWS_SECRET_ACCESS_KEY": "YOUR_SECRET_ACCESS_KEY_HERE",
        "AWS_SESSION_TOKEN": "YOUR_OPTIONAL_SESSION_ID_FOR_SSO_TEMPORARY_CREDENTIALS_HERE",
        "AWS_REGION": "YOUR_AWS_REGION_HERE",
        "AWS_KB_IDS": "[\"kb-12345\", \"kb-67890\"]"
      },
      "disabled": false,
      "autoApprove": [
        "retrieve_from_aws_kb"
      ],
      "timeout": 120
    }
  }
}

Edificio

Estibador:

docker build -t mcp/aws-kb-retrieval -f src/aws-kb-retrieval-server/Dockerfile .

Licencia

Este servidor MCP cuenta con la licencia MIT. Esto significa que puede usar, modificar y distribuir el software libremente, sujeto a los términos y condiciones de la licencia MIT. Para más detalles, consulte el archivo de LICENCIA en el repositorio del proyecto.

Este archivo README asume que el nombre del paquete de su servidor es @modelcontextprotocol/server-aws-kb-retrieval . Ajuste el nombre del paquete y los detalles de instalación si difieren en su configuración. Además, asegúrese de que el script de su servidor esté correctamente compilado y de que todas las dependencias se administren correctamente en su package.json .