Hướng Dẫn Cấu Hình Amazon Bedrock

Kiến Trúc và Tổng Quan

Trong bài viết này, chúng ta sẽ tìm hiểu cách xây dựng một nền tảng tìm kiếm tài liệu mạnh mẽ bằng cách sử dụng Amazon Bedrock. Một trong những chức năng quan trọng bao gồm lưu trữ và tìm kiếm nội dung tài liệu dưới dạng nhúng (embedding), đồng thời sử dụng mô hình học sâu (LLM) để tạo ra câu trả lời tự động dựa trên nội dung tài liệu được truy xuất (RAG).

Amazon Bedrock cung cấp khả năng tạo dựng một Knowledge Base hoàn chỉnh mà bạn có thể dễ dàng sử dụng. Bằng cách kết nối nguồn dữ liệu như S3, Bedrock sẽ nhúng nội dung lưu trữ và cho phép tìm kiếm hiệu quả qua các API như Retrieve API và RetrieveAndGenerate API.

Tạo S3 Bucket Làm Nguồn Dữ Liệu

⚠️ Lưu ý rằng Data Source (S3) phải được tạo trong cùng khu vực với Knowledge Base. Nếu Amazon Bedrock được triển khai ở us-east-1, thì S3 bucket cũng cần phải được tạo ở khu vực này.

Tạo S3 Bucket
1. Nhấn vào đây để bắt đầu tạo bucket.
2. Chọn khu vực (region) và nhập tên bucket theo ý muốn.
Giữ nguyên các cài đặt khác và nhấn Create bucket để hoàn tất.

Cấu Hình Amazon Bedrock Knowledge Base

Trong bảng điều khiển Amazon Bedrock, chọn Knowledge bases, sau đó nhấn Create knowledge base.
Nhập tên Knowledge Base, chọn Create and use a new service role trong phần IAM permissions, và điền tên cho vai trò này để dễ dàng chỉnh sửa sau này.
Nhấn Next và nhập Data Source Name cùng S3 URI. S3 URI sẽ có dạng s3://tên_bucket_đã_tạo.

⚠️ Quá trình đồng bộ (Sync) sẽ dựa trên S3 URI đã được cấu hình. Nếu bạn muốn đồng bộ hóa một thư mục cụ thể, bạn có thể chỉ định thêm đường dẫn thư mục.

Chọn mô hình Embeddings model - Titan Embeddings G1 Text và Vector Database.
Xem lại các tham số cấu hình và nhấn Create knowledge base để hoàn tất quá trình tạo (quá trình này có thể mất từ 10 đến 20 phút).

Thiết Lập Vai Trò IAM Cho Knowledge Base

Truy cập vào Bảng điều khiển IAM Role.
Nhập tên của Knowledge Base Role đã được tạo để tìm kiếm.
Nhấp vào Role và trong tab Permissions, chọn Add permissions - Create inline policy.
Chọn JSON và cấu hình quyền truy cập S3 cho Knowledge Base như sau:

⚠️ Hãy thay đổi YOUR_BUCKET_NAME và YOUR_ACCOUNT_ID (12 chữ số, không có dấu ‘-’ nằm ở góc trên bên phải bảng điều khiển).

json Copy

{
    "Version": "2012-10-17",
    "Statement": [{
        "Effect": "Allow",
        "Action": [
            "s3:GetObject",
            "s3:ListBucket"
        ],
        "Resource": [
            "arn:aws:s3:::YOUR_BUCKET_NAME",
            "arn:aws:s3:::YOUR_BUCKET_NAME/*"
        ],
        "Condition": {
            "StringEquals": {
                "aws:PrincipalAccount": "YOUR_ACCOUNT_ID"
            }
        }
    }]
}

Đặt tên cho Policy là access-to-s3-policy và nhấn Create policy để hoàn tất.
Kiểm tra Knowledge Base trên bảng điều khiển:
1. Tải tài liệu có định dạng phù hợp lên S3 bucket đã tạo.
2. Vào Bảng điều khiển Bedrock - Knowledge Base và kiểm tra Knowledge Base đã được tạo.
3. Trong Data source, chọn Data Source đã tạo và nhấn nút Sync.
4. Khi quá trình Sync hoàn tất, tại bảng điều khiển Test knowledge base, nhấn vào Select model để lựa chọn mô hình tạo phản hồi.
5. Ví dụ, bạn có thể tải lên tài liệu PDF liên quan đến luật đất đai 2024 và thực hiện truy vấn về quy định này.

Tích Hợp Dữ Liệu

Tạo các tài liệu mẫu và lưu trữ chúng trong S3 bucket đã tạo.
Cấu hình Knowledge Base để đồng bộ dữ liệu từ S3.

Kết Luận

Chúc mừng bạn! Giờ đây bạn có thể thực hiện truy vấn trên Knowledge Base mà bạn đã thiết lập. Hãy tìm hiểu thêm để tùy chỉnh tham số và tài liệu truy vấn một cách hiệu quả.

Nếu bạn cần thêm bất kỳ hỗ trợ nào khác hoặc có câu hỏi, hãy cho tôi biết!
source: viblo

Hướng Dẫn Cấu Hình Amazon Bedrock: Xây Dựng Knowledge Base và Tích Hợp Tìm Kiếm Tài Liệu