MulaiMulai sekarang secara gratis

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 adalah bagian dari kursus

Agen AI dengan Hugging Face smolagents

Lihat Kursus

Petunjuk latihan

  • Lengkapi tanda tangan fungsi dengan menambahkan parameter agent_step.
  • Periksa apakah langkah saat ini menghasilkan jawaban final menggunakan atribut .is_final_answer dari agent_step.
  • Jika ini jawaban final, ambil total jumlah token dari total_tokens dan cetak nilainya.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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: {____}")
Edit dan Jalankan Kode