AWS Bedrock KB Retrieval

Amazon Bedrock Knowledge Base Retrieval MCP-Server

MCP-Server für den Zugriff auf Amazon Bedrock Knowledge Bases

Funktionen

Knowledge Bases und ihre Datenquellen entdecken

• Alle verfügbaren Knowledge Bases finden und erkunden
• Knowledge Bases nach Name oder Tag durchsuchen
• Mit jeder Knowledge Base verknüpfte Datenquellen auflisten

Knowledge Bases mit natürlicher Sprache abfragen

• Informationen über konversationelle Abfragen abrufen
• Relevante Passagen aus Ihren Knowledge Bases erhalten
• Quellenangaben für alle Ergebnisse abrufen

Ergebnisse nach Datenquelle filtern

• Ihre Abfragen auf bestimmte Datenquellen fokussieren
• Bestimmte Datenquellen ein- oder ausschließen
• Ergebnisse aus bestimmten Datenquellen priorisieren

Ergebnisse neu bewerten (Reranking)

• Relevanz der Abrufergebnisse verbessern
• Amazon Bedrock Reranking-Funktionen nutzen
• Ergebnisse nach Relevanz für Ihre Abfrage sortieren

Voraussetzungen

Installationsanforderungen

1. Installieren Sie uv von Astral oder der GitHub README
2. Installieren Sie Python mit uv python install 3.10

AWS-Anforderungen

1. AWS CLI-Konfiguration: Sie müssen die AWS CLI mit Anmeldeinformationen und einem AWS_PROFILE konfiguriert haben, das Zugriff auf Amazon Bedrock und Knowledge Bases hat
2. Amazon Bedrock Knowledge Base: Sie müssen mindestens eine Amazon Bedrock Knowledge Base mit dem Tag-Schlüssel mcp-multirag-kb und dem Wert true haben
3. IAM-Berechtigungen: Ihre IAM-Rolle/Ihr IAM-Benutzer muss über die entsprechenden Berechtigungen verfügen, um:
   • Knowledge Bases aufzulisten und zu beschreiben
   • Auf Datenquellen zuzugreifen
   • Knowledge Bases abzufragen

Reranking-Anforderungen

Wenn Sie die Reranking-Funktionalität nutzen möchten, benötigt Ihre Bedrock Knowledge Base zusätzliche Berechtigungen:

1. Ihre IAM-Rolle muss Berechtigungen für die Aktionen bedrock:Rerank und bedrock:InvokeModel haben
2. Die Servicerolle von Amazon Bedrock Knowledge Bases muss ebenfalls über diese Berechtigungen verfügen
3. Reranking ist nur in bestimmten Regionen verfügbar. Eine aktuelle Liste der unterstützten Regionen finden Sie in der offiziellen Dokumentation.
4. Aktivieren Sie den Modellzugriff für die verfügbaren Reranking-Modelle in der angegebenen Region.

Reranking steuern

Reranking kann global über die Umgebungsvariable BEDROCK_KB_RERANKING_ENABLED aktiviert oder deaktiviert werden:

• Auf false gesetzt (Standard): Deaktiviert Reranking für alle Abfragen, sofern nicht explizit aktiviert
• Auf true gesetzt: Aktiviert Reranking für alle Abfragen, sofern nicht explizit deaktiviert

Die Umgebungsvariable akzeptiert verschiedene Formate:

• Zum Aktivieren: 'true', '1', 'yes' oder 'on' (Groß-/Kleinschreibung wird nicht beachtet)
• Zum Deaktivieren: jeder andere Wert oder nicht gesetzt (Standardverhalten)

Diese Einstellung bietet einen globalen Standard, während einzelne API-Aufrufe ihn weiterhin überschreiben können, indem sie den Parameter reranking explizit setzen.

Detaillierte Anweisungen zum Einrichten von Knowledge Bases finden Sie unter:

• Eine Knowledge Base erstellen
• Berechtigungen für Amazon Bedrock Knowledge Bases verwalten
• Berechtigungen für Reranking in Amazon Bedrock

Installation

Kiro	Cursor	VS Code

Konfigurieren Sie den MCP-Server in Ihrer MCP-Client-Konfiguration (z. B. für Kiro bearbeiten Sie ~/.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": []
    }
  }
}

Windows-Installation

Für Windows-Benutzer ist das Konfigurationsformat des MCP-Servers etwas anders:

{
  "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"
      }
    }
  }
}

oder Docker nach einem erfolgreichen docker build -t awslabs/bedrock-kb-retrieval-mcp-server .:

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

HINWEIS: Ihre Anmeldeinformationen müssen von Ihrem Host aus regelmäßig aktualisiert werden.

Einschränkungen

• Ergebnisse mit dem Inhaltstyp IMAGE werden nicht in die KB-Abfrageantwort aufgenommen.
• Der Parameter reranking erfordert zusätzliche Berechtigungen, Amazon Bedrock-Modellzugriff und ist nur in bestimmten Regionen verfügbar.