AWS Bedrock KB Retrieval

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

Máy chủ MCP để truy cập Amazon Bedrock Knowledge Bases

Tính năng

Khám phá các knowledge base và nguồn dữ liệu của chúng

• Tìm và khám phá tất cả các knowledge base có sẵn
• Tìm kiếm knowledge base theo tên hoặc thẻ
• Liệt kê các nguồn dữ liệu liên kết với từng knowledge base

Truy vấn knowledge base 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ừ knowledge base 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ể

Truy xuất theo tác nhân trên các knowledge base được quản lý

• Lập kế hoạch chiến lược truy xuất nhiều bước và tổng hợp câu trả lời có trích dẫn
• Tìm kiếm nhiều knowledge base trong một lần gọi
• Tùy chọn tóm tắt ngắn gọn về các bước lập kế hoạch và truy xuất của tác nhân
• Chỉ dành cho knowledge base được quản lý; công cụ từ chối các loại khác với thông báo rõ ràng

Truy cập nội dung được bảo vệ bởi ACL

• Truyền user_id để truy xuất nội dung từ các nguồn dữ liệu hỗ trợ ACL (SharePoint, OneDrive, Confluence với ACL theo tài liệu)
• Nếu không có, nội dung đó không thể truy cập được và bước mở rộng toàn bộ tài liệu của truy xuất theo tác nhân sẽ thất bại với thông báo "UserContext is required for ACL-aware data sources"
• Kết quả được lọc theo những gì người dùng đó được phép xem

Hỗ trợ cả knowledge base được quản lý và vector

• Hoạt động với knowledge base vector (type: VECTOR) và knowledge base được quản lý (type: MANAGED)
• Loại knowledge base được phát hiện tự động và cấu hình Retrieve chính xác được gửi (vectorSearchConfiguration hoặc managedSearchConfiguration)
• Lọc nguồn dữ liệu sử dụng khóa siêu dữ liệu phù hợp với loại knowledge base
• Công cụ ListKnowledgeBases báo cáo type của từng knowledge base

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à Knowledge Bases
2. Amazon Bedrock Knowledge Base: Bạn phải có ít nhất một Amazon Bedrock Knowledge Base 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ó quyền phù hợp để:
   • Liệt kê và mô tả knowledge base
   • Truy cập nguồn dữ liệu
   • Truy vấn knowledge base

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, Bedrock Knowledge Base của bạn cần các quyền bổ sung:

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ụ Amazon Bedrock Knowledge Bases cũng phải có các quyền này
3. Tính khả dụng của xếp hạng lại khác nhau theo từng mô hình: amazon.rerank-v1:0 không được cung cấp trong us-east-1, trong khi cohere.rerank-v3-5:0 thì có. Máy chủ xác thực cặp (vùng, mô hình) và thất bại nhanh với thông báo rõ ràng. Vui lòng tham khảo tài liệu chính thức để có danh sách cập nhật các vùng được hỗ trợ.
4. Bật quyền truy cập mô hình cho các mô hình xếp hạng lại có sẵn trong vùng được chỉ định.

Yêu cầu truy xuất theo tác nhân

Công cụ AgenticQueryKnowledgeBases gọi AgenticRetrieveStream, được hỗ trợ cho chỉ knowledge base được quản lý (type: MANAGED). Nó lập kế hoạch chiến lược truy xuất và, trừ khi bạn truyền generate_response=false, gọi một mô hình nền tảng để viết câu trả lời có trích dẫn.

1. Vai trò IAM của bạn cần bedrock:AgenticRetrieveStream trên knowledge base, ngoài các quyền được liệt kê ở trên
2. Vì nó gọi một mô hình nền tảng, chi phí mỗi lần gọi cao hơn đáng kể so với QueryKnowledgeBases. Truyền generate_response=false để truy xuất mà không cần tổng hợp
3. RetrieveAndGenerate không được hỗ trợ cho knowledge base được quản lý, vì vậy truy xuất theo tác nhân với generate_response=true là cách để có câu trả lời được tạo từ một trong số đó

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 bị 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.

Để có hướng dẫn chi tiết về thiết lập knowledge base, xem:

• Tạo knowledge base
• Quản lý quyền cho Amazon Bedrock knowledge bases
• Quyền 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 một chút:

{
  "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 liên tục 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 quyền bổ sung, quyền truy cập mô hình Amazon Bedrock và chỉ khả dụng ở các vùng cụ thể.