Theo dõi các bước của Agent, giám sát hành động
Bạn đang xây dựng một trợ lý smolagents cho huấn luyện viên bóng rổ, người cần phân tích các báo cáo PDF chứa thống kê cầu thủ, thông tin do thám và chiến thuật thi đấu.
Huấn luyện viên dựa vào agent để trả lời các câu hỏi như: "Chúng ta nên dùng chiến thuật phòng ngự nào trước đội hình hai của họ?"
Nhưng huấn luyện viên không chỉ muốn câu trả lời — họ còn muốn thấy agent đang làm gì ở hậu trường.
Trong bài tập này, bạn sẽ viết một action callback chạy mỗi khi agent thực hiện một bước, như gọi một công cụ hoặc sử dụng mô hình. Callback này sẽ:
- Hiển thị số lượng bước,
- Và nếu agent đã hoàn thành, hiển thị số token đã dùng.
Điều này sẽ giúp huấn luyện viên (và cả bạn!) theo dõi tiến độ của agent và biết được agent đã làm bao nhiêu việc để đi đến kết luận.
Bài tập này là một phần của khóa học
AI Agents với Hugging Face smolagents
Hướng dẫn bài tập
- Hoàn thiện chữ ký hàm bằng cách thêm tham số
agent_step. - Kiểm tra liệu bước hiện tại có tạo ra câu trả lời cuối cùng hay không bằng thuộc tính
.is_final_answercủaagent_step. - Nếu là câu trả lời cuối cùng, lấy tổng số token từ
total_tokensvà in ra.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Define an action callback that accepts the agent step and the agent
def action_callback(____, agent):
step_num = agent_step.step_number
print(f"Step {step_num}: Analyzing basketball data!")
# Check if the agent step includes token usage
if agent_step.____:
total_tokens = agent_step.token_usage.total_tokens
# Print how many tokens were used
print(f"Analysis complete! Total tokens used: {____}")