AWS Bedrock KB Retrieval

Máy chủ MCP Truy xuất Cơ sở Kiến thức Amazon Bedrock

Máy chủ MCP để truy cập Cơ sở Kiến thức Amazon Bedrock

Tính năng

Khám phá cơ sở kiến thức và nguồn dữ liệu của chúng

• Tìm và khám phá tất cả cơ sở kiến thức khả dụng
• Tìm kiếm cơ sở kiến thức theo tên hoặc thẻ
• Liệt kê các nguồn dữ liệu liên kết với mỗi cơ sở kiến thức

Truy vấn cơ sở kiến thức bằng ngôn ngữ tự nhiên

• Truy xuất thông tin bằng các truy vấn hội thoại
• Lấy các đoạn văn liên quan từ cơ sở kiến thức của bạn
• Truy cập thông tin trích dẫn cho tất cả kết quả

Lọc kết quả theo nguồn dữ liệu

• Tập trung truy vấn vào các nguồn dữ liệu cụ thể
• Bao gồm hoặc loại trừ các nguồn dữ liệu cụ thể
• Ưu tiên kết quả từ các nguồn dữ liệu cụ thể

Xếp hạng lại kết quả

• Cải thiện mức độ liên quan của kết quả truy xuất
• Sử dụng khả năng xếp hạng lại của Amazon Bedrock
• Sắp xếp kết quả theo mức độ liên quan đến truy vấn của bạn

Điều kiện tiên quyết

Yêu cầu cài đặt

1. Cài đặt uv từ Astral hoặc GitHub README
2. Cài đặt Python bằng uv python install 3.10

Yêu cầu AWS

1. Cấu hình AWS CLI: Bạn phải cấu hình AWS CLI với thông tin xác thực và AWS_PROFILE có quyền truy cập Amazon Bedrock và Cơ sở Kiến thức
2. Cơ sở Kiến thức Amazon Bedrock: Bạn phải có ít nhất một Cơ sở Kiến thức Amazon Bedrock với khóa thẻ mcp-multirag-kb có giá trị true
3. Quyền IAM: Vai trò/người dùng IAM của bạn phải có các quyền phù hợp để:
   • Liệt kê và mô tả cơ sở kiến thức
   • Truy cập nguồn dữ liệu
   • Truy vấn cơ sở kiến thức

Yêu cầu xếp hạng lại

Nếu bạn định sử dụng chức năng xếp hạng lại, Cơ sở Kiến thức Bedrock của bạn cần thêm quyền:

1. Vai trò IAM của bạn phải có quyền cho cả hành động bedrock:Rerank và bedrock:InvokeModel
2. Vai trò dịch vụ Cơ sở Kiến thức Amazon Bedrock cũng phải có các quyền này
3. Xếp hạng lại chỉ khả dụng ở các khu vực cụ thể. Vui lòng tham khảo tài liệu chính thức để biết danh sách cập nhật các khu vực được hỗ trợ.
4. Kích hoạt quyền truy cập mô hình cho các mô hình xếp hạng lại khả dụng trong khu vực được chỉ định.

Kiểm soát xếp hạng lại

Xếp hạng lại có thể được bật hoặc tắt toàn cục bằng biến môi trường BEDROCK_KB_RERANKING_ENABLED:

• Đặt thành false (mặc định): Tắt xếp hạng lại cho tất cả truy vấn trừ khi được bật rõ ràng
• Đặt thành true: Bật xếp hạng lại cho tất cả truy vấn trừ khi được tắt rõ ràng

Biến môi trường chấp nhận nhiều định dạng:

• Để bật: 'true', '1', 'yes', hoặc 'on' (không phân biệt chữ hoa chữ thường)
• Để tắt: bất kỳ giá trị nào khác hoặc không đặt (hành vi mặc định)

Cài đặt này cung cấp mặc định toàn cục, trong khi các lệnh gọi API riêng lẻ vẫn có thể ghi đè bằng cách đặt rõ ràng tham số reranking.

Để biết hướng dẫn chi tiết về thiết lập cơ sở kiến thức, xem:

• Tạo cơ sở kiến thức
• Quản lý quyền cho cơ sở kiến thức Amazon Bedrock
• Quyền cho xếp hạng lại trong Amazon Bedrock

Cài đặt

Kiro	Cursor	VS Code

Cấu hình máy chủ MCP trong cấu hình máy khách MCP của bạn (ví dụ: đối với Kiro, chỉnh sửa ~/.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": []
    }
  }
}

Cài đặt trên Windows

Đối với người dùng Windows, định dạng cấu hình máy chủ MCP hơi khác:

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

hoặc docker sau khi docker build -t awslabs/bedrock-kb-retrieval-mcp-server . thành công:

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

LƯU Ý: Thông tin xác thực của bạn sẽ cần được làm mới từ máy chủ của bạn

Hạn chế

• Kết quả có loại nội dung IMAGE không được bao gồm trong phản hồi truy vấn KB.
• Tham số reranking yêu cầu thêm quyền, quyền truy cập mô hình Amazon Bedrock và chỉ khả dụng ở các khu vực cụ thể.