Hướng dẫn xây dựng hệ thống đa tác tử (Multi-Agent) với Google ADK: Tạo công cụ săn tìm xu hướng AI

Chào mừng các bạn đã quay trở lại với loạt bài hướng dẫn xây dựng trợ lý nghiên cứu AI tối ưu cho podcast về AI agent của chúng tôi! Trong bài viết đầu tiên, chúng ta đã xây dựng một agent tuyệt vời có khả năng tìm kiếm trên web để cập nhật những tin tức mới nhất về AI agent cho podcast agent factory. Nhưng nếu chúng ta muốn bổ sung các kỹ năng chuyên sâu hơn, chẳng hạn như nắm bắt "nhịp đập" thực tế từ các cộng đồng lập trình viên trên Reddit thì sao? Để làm được điều đó, chúng ta cần nâng cấp thiết kế cho agent của mình.

Trong hướng dẫn này, chúng ta sẽ nâng tầm kỹ năng và tái cấu trúc agent đơn giản hiện tại thành một hệ thống đa agent (multi-agent system) mạnh mẽ. Chúng ta sẽ xây dựng một agent "Quản lý" (Manager) để điều phối một nhóm các chuyên gia, bao gồm cả một agent sử dụng công cụ Reddit tùy chỉnh, nhằm thu thập những thông tin chuyên sâu, đa dạng và phong phú hơn.

Đến cuối bài viết này, bạn sẽ sở hữu một agent Trend Spotter mạnh mẽ hơn nữa, có khả năng lấy thông tin từ nhiều nguồn khác nhau. Quan trọng hơn, bạn sẽ học được các kỹ năng nâng cao cần thiết để xây dựng những agent phức tạp với ADK. Bạn sẽ biết cách:

Kiến trúc này chính là chìa khóa để khai phá toàn bộ tiềm năng của agent. Hãy cùng bắt đầu nào!

Bước 1: Lấy thông tin xác thực Reddit API & Cài đặt thư viện

Để cho phép agent truy cập Reddit theo lập trình, chúng ta cần lấy thông tin xác thực API. Việc này hoàn toàn miễn phí và chỉ mất một phút.

5. Thiết lập biến môi trường: Để đảm bảo tính bảo mật, chúng ta sẽ lưu trữ các thông tin xác thực này dưới dạng biến môi trường.

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

6. Cập nhật các phụ thuộc: Thêm thư viện praw vào tệp requirements.txt của bạn và cài đặt nó.

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Cấu hình môi trường Cloud của bạn

Nếu bạn chưa thực hiện, trong bài blog trước, chúng tôi đã hướng dẫn cách xác định các cài đặt để chỉ dẫn ADK kết nối an toàn với tài khoản Google Cloud của bạn nhằm sử dụng các dịch vụ như Vertex AI và Google Search.

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Chạy lệnh một lần này. Nó sẽ mở trình duyệt để bạn đăng nhập, cho phép ADK thực hiện các yêu cầu được ủy quyền thay cho bạn.

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Bước 2: Tạo thư mục dự án

Để tổ chức nhóm agent mới, chúng ta sẽ tạo một thư mục sub_agents bên trong gói trend_spotter chính.

Đừng lo lắng nếu bạn chưa xem qua bài blog đầu tiên, chúng tôi đã hỗ trợ bạn! Dưới đây là cách tạo cấu trúc thư mục cho agent chính của bạn:

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

(Trên Windows, hãy sử dụng python -m venv venv &&.\venv\Scripts\activate)

Sau khi đã có agent chính và cấu trúc thư mục được xác định, từ thư mục gốc trend-spotter, hãy chạy:

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Bước 3: Tạo các Sub-Agent chuyên gia

Bây giờ chúng ta sẽ xây dựng hai agent chuyên gia bằng cách tạo trực tiếp các instance của lớp Agent.

1. Chuyên gia Google Search:

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

2. Chuyên gia Reddit:

Đầu tiên, hãy tạo một tệp mới cho mã nguồn công cụ tùy chỉnh của chúng ta: trend_spotter/tools.py. Thêm hàm sau vào tệp đó.

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Bây giờ, hãy tạo chính agent Reddit tại trend_spotter/sub_agents/reddit_agent.py:

Chuyển sang chế độ toàn màn hình Thoát chế độ toàn màn hình

Bước 4: Xây dựng Agent điều phối chính (Orchestrator)

Bây giờ chúng ta sẽ sửa đổi agent chính từ Phần 1 để trở thành "người quản lý" cho nhóm chuyên gia mới của chúng ta.

Mở trend_spotter/prompt.py và thay thế nội dung của nó bằng prompt điều phối mới này: