Lacak Langkah Agen, Pantau Aksi
Anda sedang membangun asisten smolagents untuk seorang pelatih bola basket yang membutuhkan bantuan menganalisis laporan PDF yang berisi statistik pemain, wawasan scouting, dan strategi permainan.
Pelatih mengandalkan agen untuk menjawab pertanyaan seperti: "Strategi pertahanan apa yang sebaiknya kita jalankan melawan unit kedua mereka?"
Namun pelatih tidak hanya menginginkan jawaban — mereka juga ingin mengetahui apa yang dilakukan agen di belakang layar.
Dalam latihan ini, Anda akan menulis sebuah action callback yang berjalan setiap kali agen mengambil langkah, seperti memanggil tool atau menggunakan model. Callback ini akan:
- Menampilkan jumlah langkah,
- Dan jika agen sudah selesai, menampilkan berapa banyak token yang digunakan.
Ini akan membantu pelatih (dan Anda!) memantau bagaimana agen bergerak maju dan seberapa banyak pekerjaan yang dilakukan untuk mencapai sebuah kesimpulan.
Latihan ini merupakan bagian dari kursus
Agen AI dengan Hugging Face smolagents
Instruksi latihan
- Lengkapi tanda tangan fungsi dengan menambahkan parameter
agent_step. - Periksa apakah langkah saat ini menghasilkan jawaban final menggunakan atribut
.is_final_answerdariagent_step. - Jika ini jawaban final, ambil total jumlah token dari
total_tokensdan cetak nilainya.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# 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: {____}")