Đến giờ thi đấu: Chạy toàn bộ agent
Agent huấn luyện bóng rổ của bạn gần như sẵn sàng ra sân. Bạn đã viết một hàm callback để hiển thị số bước và mức sử dụng token.
Giờ là lúc ghép mọi thứ lại: bạn sẽ cấu hình một CodeAgent dùng callback đó.
Bạn đã có sẵn:
- Một
basketball_toolgiúp agent tìm kiếm các báo cáo trinh sát - Một biến
modelchứa mô hình ngôn ngữ của bạn - Một hàm callback:
action_callback
Mục tiêu của bạn là hoàn thiện việc kết nối agent.
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
- Import lớp
ActionStepđể có thể dùng nó đăng ký action callback. - Đăng ký hàm
action_callbackcho lớpActionStepđể agent hiển thị các hành động của mình trong khi thực thi.
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.
# Import ActionStep to register a callback for it
from smolagents import ____
coach_agent = CodeAgent(
tools=[basketball_tool],
model=model,
verbosity_level=0,
# Register a callback that runs when an ActionStep is triggered
step_callbacks={ActionStep: ____}
)
# Run a question through the agent
result = coach_agent.run("What defensive strategy should we use to stop their point guard who averages 25 points per game?")
print(result)