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

Connected

演習

状態を持つ設計:RAG 検索ツールを構築する

前のレッスンでは、家電製品のマニュアルをベクトルベースのナレッジベースとして構築し、埋め込みを使って検索する方法を学びました。

次は、このような検索ロジックをラップするカスタムツールを構築しましょう。エージェントがこのツールを使って、家電製品に関する質問に答えられるようにします。

作成するツールは Tool 基底クラスのサブクラスとして実装し、家電製品の操作に関する質問という1つの入力を受け付けます。

すでに以下のものが利用できます。

  • 事前に構築済みの FAISS インデックスを含む vector_store という変数
  • 家電製品マニュアルの内容を埋め込み済みのドキュメントチャンク(検索可能な状態)

ここでの課題は、このナレッジベースをエージェントから利用できるようにするツールの構造とロジックを実装することです。

指示

100 XP
  • __init__() メソッドで vector_store パラメータを受け取るようにします。
  • forward() メソッドにパラメータとして query を追加します。
  • self.k を使って、類似検索から返すドキュメント数を指定します。