AWS Bedrock KB Retrieval

Serveur MCP de récupération Amazon Bedrock Knowledge Base

Serveur MCP pour accéder aux bases de connaissances Amazon Bedrock

Fonctionnalités

Découvrir les bases de connaissances et leurs sources de données

• Trouver et explorer toutes les bases de connaissances disponibles
• Rechercher des bases de connaissances par nom ou par balise
• Lister les sources de données associées à chaque base de connaissances

Interroger les bases de connaissances en langage naturel

• Récupérer des informations à l’aide de requêtes conversationnelles
• Obtenir des passages pertinents depuis vos bases de connaissances
• Accéder aux informations de citation pour tous les résultats

Filtrer les résultats par source de données

• Concentrer vos requêtes sur des sources de données spécifiques
• Inclure ou exclure des sources de données spécifiques
• Prioriser les résultats provenant de sources de données spécifiques

Réordonner les résultats

• Améliorer la pertinence des résultats de récupération
• Utiliser les capacités de réordonnancement d’Amazon Bedrock
• Trier les résultats par pertinence par rapport à votre requête

Prérequis

Configuration requise pour l’installation

1. Installer uv depuis Astral ou le README GitHub
2. Installer Python à l’aide de uv python install 3.10

Configuration AWS requise

1. Configuration de l’AWS CLI : Vous devez avoir configuré l’AWS CLI avec des identifiants et un AWS_PROFILE ayant accès à Amazon Bedrock et aux bases de connaissances
2. Base de connaissances Amazon Bedrock : Vous devez disposer d’au moins une base de connaissances Amazon Bedrock avec la clé de balise mcp-multirag-kb ayant pour valeur true
3. Permissions IAM : Votre rôle/utilisateur IAM doit disposer des permissions appropriées pour :
   • Lister et décrire les bases de connaissances
   • Accéder aux sources de données
   • Interroger les bases de connaissances

Configuration requise pour le réordonnancement

Si vous avez l’intention d’utiliser la fonctionnalité de réordonnancement, votre base de connaissances Bedrock nécessite des permissions supplémentaires :

1. Votre rôle IAM doit avoir les permissions pour les actions bedrock:Rerank et bedrock:InvokeModel
2. Le rôle de service Amazon Bedrock Knowledge Bases doit également disposer de ces permissions
3. Le réordonnancement n’est disponible que dans certaines régions. Veuillez consulter la documentation officielle pour une liste à jour des régions prises en charge.
4. Activer l’accès au modèle pour les modèles de réordonnancement disponibles dans la région spécifiée.

Contrôle du réordonnancement

Le réordonnancement peut être activé ou désactivé globalement à l’aide de la variable d’environnement BEDROCK_KB_RERANKING_ENABLED :

• Définie sur false (par défaut) : Désactive le réordonnancement pour toutes les requêtes, sauf activation explicite
• Définie sur true : Active le réordonnancement pour toutes les requêtes, sauf désactivation explicite

La variable d’environnement accepte différents formats :

• Pour activer : 'true', '1', 'yes' ou 'on' (insensible à la casse)
• Pour désactiver : toute autre valeur ou non définie (comportement par défaut)

Ce paramètre fournit une valeur par défaut globale, tandis que les appels API individuels peuvent toujours la remplacer en définissant explicitement le paramètre reranking.

Pour des instructions détaillées sur la configuration des bases de connaissances, consultez :

• Créer une base de connaissances
• Gestion des permissions pour les bases de connaissances Amazon Bedrock
• Permissions pour le réordonnancement dans Amazon Bedrock

Installation

Kiro	Cursor	VS Code

Configurez le serveur MCP dans la configuration de votre client MCP (par exemple, pour Kiro, modifiez ~/.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": []
    }
  }
}

Installation sous Windows

Pour les utilisateurs de Windows, le format de configuration du serveur MCP est légèrement différent :

{
  "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 après un docker build -t awslabs/bedrock-kb-retrieval-mcp-server . réussi :

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

REMARQUE : Vos identifiants devront être régulièrement rafraîchis depuis votre hôte

Limitations

• Les résultats avec le type de contenu IMAGE ne sont pas inclus dans la réponse de requête KB.
• Le paramètre reranking nécessite des permissions supplémentaires, l’accès au modèle Amazon Bedrock, et n’est disponible que dans certaines régions.