AWS Bedrock KB Retrieval

Server MCP Pengambilan Basis Pengetahuan Amazon Bedrock

Server MCP untuk mengakses Basis Pengetahuan Amazon Bedrock

Fitur

Temukan basis pengetahuan dan sumber datanya

• Temukan dan jelajahi semua basis pengetahuan yang tersedia
• Cari basis pengetahuan berdasarkan nama atau tag
• Daftar sumber data yang terkait dengan setiap basis pengetahuan

Kueri basis pengetahuan dengan bahasa alami

• Ambil informasi menggunakan kueri percakapan
• Dapatkan bagian yang relevan dari basis pengetahuan Anda
• Akses informasi kutipan untuk semua hasil

Filter hasil berdasarkan sumber data

• Fokuskan kueri Anda pada sumber data tertentu
• Sertakan atau kecualikan sumber data tertentu
• Prioritaskan hasil dari sumber data tertentu

Peringkat ulang hasil

• Tingkatkan relevansi hasil pengambilan
• Gunakan kemampuan peringkat ulang Amazon Bedrock
• Urutkan hasil berdasarkan relevansi dengan kueri Anda

Prasyarat

Persyaratan Instalasi

1. Instal uv dari Astral atau README GitHub
2. Instal Python menggunakan uv python install 3.10

Persyaratan AWS

1. Konfigurasi AWS CLI: Anda harus memiliki AWS CLI yang dikonfigurasi dengan kredensial dan AWS_PROFILE yang memiliki akses ke Amazon Bedrock dan Basis Pengetahuan
2. Basis Pengetahuan Amazon Bedrock: Anda harus memiliki setidaknya satu Basis Pengetahuan Amazon Bedrock dengan kunci tag mcp-multirag-kb dengan nilai true
3. Izin IAM: Peran/pengguna IAM Anda harus memiliki izin yang sesuai untuk:
   • Mencantumkan dan mendeskripsikan basis pengetahuan
   • Mengakses sumber data
   • Mengkueri basis pengetahuan

Persyaratan Peringkat Ulang

Jika Anda bermaksud menggunakan fungsionalitas peringkat ulang, Basis Pengetahuan Bedrock Anda memerlukan izin tambahan:

1. Peran IAM Anda harus memiliki izin untuk tindakan bedrock:Rerank dan bedrock:InvokeModel
2. Peran layanan Basis Pengetahuan Amazon Bedrock juga harus memiliki izin ini
3. Peringkat ulang hanya tersedia di wilayah tertentu. Silakan merujuk ke dokumentasi resmi untuk daftar wilayah yang didukung terkini.
4. Aktifkan akses model untuk model peringkat ulang yang tersedia di wilayah yang ditentukan.

Mengontrol Peringkat Ulang

Peringkat ulang dapat diaktifkan atau dinonaktifkan secara global menggunakan variabel lingkungan BEDROCK_KB_RERANKING_ENABLED:

• Atur ke false (default): Menonaktifkan peringkat ulang untuk semua kueri kecuali diaktifkan secara eksplisit
• Atur ke true: Mengaktifkan peringkat ulang untuk semua kueri kecuali dinonaktifkan secara eksplisit

Variabel lingkungan menerima berbagai format:

• Untuk mengaktifkan: 'true', '1', 'yes', atau 'on' (tidak peka huruf besar/kecil)
• Untuk menonaktifkan: nilai lain apa pun atau tidak diatur (perilaku default)

Pengaturan ini menyediakan default global, sementara panggilan API individual masih dapat menimpanya dengan mengatur parameter reranking secara eksplisit.

Untuk petunjuk terperinci tentang menyiapkan basis pengetahuan, lihat:

• Buat basis pengetahuan
• Mengelola izin untuk basis pengetahuan Amazon Bedrock
• Izin untuk peringkat ulang di Amazon Bedrock

Instalasi

Kiro	Cursor	VS Code

Konfigurasikan server MCP di konfigurasi klien MCP Anda (misalnya, untuk Kiro, edit ~/.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": []
    }
  }
}

Instalasi Windows

Untuk pengguna Windows, format konfigurasi server MCP sedikit berbeda:

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

atau docker setelah docker build -t awslabs/bedrock-kb-retrieval-mcp-server . berhasil:

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

CATATAN: Kredensial Anda perlu terus disegarkan dari host Anda

Batasan

• Hasil dengan tipe konten IMAGE tidak disertakan dalam respons kueri KB.
• Parameter reranking memerlukan izin tambahan, akses model Amazon Bedrock, dan hanya tersedia di wilayah tertentu.