1. 学ぶ
    2. /
  2. コース
    3. /
  3. Hugging Face smolagents で学ぶ AI エージェント
Connected

演習

エージェントのステップを追跡してアクションを監視する

バスケットボールのコーチを支援する smolagents アシスタントを構築しています。このアシスタントは、選手の統計データ、スカウティング情報、試合戦略などを含む PDF レポートの分析に活用されます。

コーチは「相手の第2ユニットに対してどのような守備戦略を取るべきか？」といった質問への回答をエージェントに求めています。

ただし、コーチが必要としているのは回答だけではありません。エージェントが裏側でどのような処理をしているか、その動作も把握したいと考えています。

この演習では、ツールの呼び出しやモデルの利用など、エージェントがステップを実行するたびに動作するアクションコールバックを実装します。このコールバックは以下の情報を表示します。

  • ステップ数、
  • エージェントが処理を完了した場合は、使用されたトークン数。

これにより、コーチ（そしてあなた自身も）がエージェントの進捗状況と、結論に至るまでの処理量を把握できるようになります。

指示

100 XP
  • agent_step パラメータを追加して、関数のシグネチャを完成させてください。
  • agent_step の .is_final_answer 属性を使って、現在のステップが最終回答を生成したかどうかを確認してください。
  • 最終回答であれば、total_tokens からトークンの合計数を取得して出力してください。