Serveur MCP pour l'accès aux bases de connaissances Amazon Bedrock

Serveur MCP pour l'accès 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
• Recherche de bases de connaissances par nom ou par étiquette
• Liste des 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
• Obtenez des passages pertinents de vos bases de connaissances
• Accéder aux informations de citation pour tous les résultats

Filtrer les résultats par source de données

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

Reclasser les résultats

• Améliorer la pertinence des résultats de la recherche
• Utiliser les capacités de reclassement d'Amazon Bedrock
• Trier les résultats en fonction de leur pertinence par rapport à votre requête

Conditions préalables

Conditions d'installation

1. Installer uv à partir d'Astral ou du README de GitHub
2. Installer Python en utilisant uv python install 3.10

Exigences AWS

1. Configuration de la CLI AWS: Vous devez avoir configuré le CLI AWS avec des informations d'identification et un AWS_PROFILE qui a 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 marquage mcp-multirag-kb et la valeur true
3. Permissions IAM: Votre rôle/utilisateur IAM doit avoir les permissions appropriées pour
   • Lister et décrire les bases de connaissances
   • Accéder aux sources de données
   • Interroger les bases de connaissances

Exigences en matière de reclassement

Si vous avez l'intention d'utiliser la fonctionnalité de reclassement, votre base de connaissances Bedrock a besoin de permissions supplémentaires :

1. Votre rôle IAM doit avoir des permissions pour les actions bedrock:Rerank et bedrock:InvokeModel
2. Le rôle de service Amazon Bedrock Knowledge Bases doit également avoir ces permissions
3. Le Reranking n'est disponible que dans certaines régions. Veuillez vous référer à la documentation officielle pour une liste à jour des régions supportées.
4. Activer l'accès au modèle pour les modèles de reclassement disponibles dans la région spécifiée.

Contrôle du reclassement

Le reclassement peut être activé ou désactivé globalement à l'aide de la variable d'environnement BEDROCK_KB_RERANKING_ENABLED:

• La valeur false (par défaut) : Désactive le reclassement pour toutes les requêtes, à moins qu'il ne soit explicitement activé
• Valeur vraie: Active le reclassement pour toutes les requêtes, à moins qu'il ne soit explicitement désactivé

La variable d'environnement accepte différents formats :

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

Ce paramètre fournit une valeur globale par défaut, mais les appels d'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, voir :

• Créer une base de connaissances
• Gestion des autorisations pour les bases de connaissances Amazon Bedrock
• Autorisations pour le reranking dans Amazon Bedrock

Installation de la base de connaissances

Configurez le serveur MCP dans la configuration de votre client MCP (par exemple, pour Amazon Q Developer CLI, modifiez ~/.aws/amazonq/mcp.json) :

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

ou docker après une construction réussie de docker -t awslabs/bedrock-kb-retrieval-mcp-server .. :

# fichier `.env` fictif avec les informations d'identification temporaires 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 informations d'identification devront être mises à jour régulièrement à partir de votre hôte

Limitations

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