HuggingFaceのSmolAgents、複数AI連携を簡単に実装

HuggingFaceのSmolAgents、複数AI連携を簡単に実装 おすすめAIツール

SmolAgentsとは何か

SmolAgentsは、HuggingFaceが開発している軽量なAIエージェント・フレームワークです。現在の安定版は1.24.0で、OpenAIのgpt-4o-miniなどのLLMを活用して動作します。「Smol」という名前が示す通り、最小限の構成で必要な機能を実現することを目指しています。

このフレームワークの特徴は、複数のAIエージェントにそれぞれ専門的な役割を持たせて、一つの管理エージェントがそれらを調整できる点です。例えば、数学計算が得意なエージェントと、Web検索が得意なエージェントを用意して、マネージャー役のエージェントが適切に仕事を振り分ける、といった使い方ができます。

従来のAI実装では、こうした複数エージェントの協調動作を実現するには、かなり複雑なコードを書く必要がありました。SmolAgentsはこのハードルを下げて、Pythonの基本的な知識があれば構築できるように設計されています。

実装できる機能とカスタムツール

SmolAgentsでは、AIエージェントに様々なツールを持たせることができます。公開されているチュートリアルでは、5つのカスタムツールが実装例として紹介されています。

温度変換ツール(celsius_to_fahrenheit)は、摂氏から華氏への変換を行います。素数チェッカー(PrimeTool)は、指定された数が素数かどうかを判定し、素数でない場合は最小の素因数を返します。メモリ・ストレージ・ツール(MemoTool)は、キーと値のペアでデータを保存・取得できる機能を提供します。Web検索ツール(DuckDuckGoTool)は、DuckDuckGoのAPIを使って情報を検索します。階乗計算ツール(factorial)は、数値の階乗を計算します。

これらのツールは、デコレータベース(関数に@マークで機能を追加する方式)とクラスベース(Pythonのクラスとして定義する方式)の両方で実装できます。どちらの方法を選ぶかは、ツールの複雑さや再利用性に応じて決められます。

エージェントのタイプと動作の仕組み

SmolAgentsには、主に2つのエージェントタイプがあります。CodeAgentは、与えられたタスクに対して動的にPythonコードを書いて実行し、複数ステップの問題を解決します。最大6ステップまで繰り返し処理を行い、途中で得られた情報を次のステップに活かします。

もう一つのToolCallingAgentは、ReAct(Reasoning and Acting)と呼ばれる推論スタイルを実装しています。これは「考える→行動する→結果を観察する」というサイクルを繰り返すアプローチで、最大5ステップまで実行します。

実際の動作の流れは次のようになります。まずタスクが与えられると、LLMがPythonコードを記述します。そのコードがサンドボックス環境で実行され、結果や例外が観察されます。この観察結果を元に、次のステップのコードが生成されます。このループが最大ステップ数に達するか、最終的な答えが得られるまで続きます。

マルチエージェント・オーケストレーションの実装

バージョン1.8以降のSmolAgentsでは、複数のエージェントを管理する仕組みが導入されました。この機能を使うと、専門性を持った複数のエージェントを一つのマネージャー・エージェントが調整できます。

チュートリアルで示されている例では、数学専門のエージェント(math_agent)と研究専門のエージェント(research_agent)が管理エージェントとして用意されています。数学エージェントは素数判定ツールを使い、研究エージェントはWeb検索とメモリ保存のツールを使います。これらはそれぞれ最大4ステップまで実行できます。

マネージャー・エージェントはCodeAgentとして実装され、最大8ステップの実行が可能です。このマネージャーが、タスクの内容に応じて適切な専門エージェントに仕事を委任します。例えば「Pythonの初版リリース年を調べて、その年が素数かどうか判定する」というタスクが与えられた場合、まず研究エージェントがWeb検索でリリース年を調べ、次に数学エージェントがその年の素数判定を行う、という流れになります。

実装方法としては、管理エージェントをmanaged_agentsパラメータを通じてマネージャーに渡します。各管理エージェントには、name(名前)とdescription(説明)を初期化時に設定する必要があります。この説明文を元に、マネージャー・エージェントがどのエージェントに仕事を振るべきか判断します。

技術的な実装のポイント

SmolAgentsの実装で押さえておくべき技術的なポイントがいくつかあります。LLMのバックエンドにはLiteLLMModelが使われており、モデルIDとして「openai/gpt-4o-mini」を指定します。OpenAI以外のLLMも使用可能です。

ツールの管理は、agent.toolsという通常のPython辞書を使って行います。バージョン1.x以前にあったToolboxクラスは削除されているので、古いチュートリアルを参考にする際は注意が必要です。実行時に新しいツールを追加したい場合は、このagent.tools辞書に直接追加できます。

CodeAgentの実行ループは明確に定義されています。タスクが与えられると、LLMがPythonコードを生成し、サンドボックスで実行されます。実行結果や例外が観察され、それを元に次のコードが生成されます。このループが最大ステップ数に達するか、final_answer()関数が呼び出されるまで続きます。

実際のタスク実行例

チュートリアルでは、いくつかの実行例が示されています。温度変換タスクでは、100℃と37℃をそれぞれ華氏に変換し、その差を計算させています。複合計算タスクでは、17×19を計算し、その結果と7919が素数かどうかを判定させています。階乗計算では、10の階乗(10!)を計算させています。

ToolCallingAgentを使ったタスクでは、ReActスタイルの推論を実行しています。マネージャー・エージェントを使った調整タスクでは、前述の「Pythonの初版リリース年を調べて素数判定する」という、複数の専門エージェントの協調が必要なタスクを実行しています。

フリーランスへの影響

この技術は、特にフリーランスのエンジニアやデータサイエンティストにとって、クライアントワークの幅を広げる可能性があります。「複数のAIを連携させて自動化したい」というニーズは増えていますが、実装の複雑さがハードルになっていました。SmolAgentsを使えば、このハードルが下がります。

例えば、クライアントの業務で「Web上の情報を収集して、その内容を分析し、レポートを作成する」という一連の流れを自動化したい場合、検索エージェント、分析エージェント、レポート生成エージェントを用意して、マネージャー・エージェントに調整させることができます。従来なら数週間かかっていた開発が、数日で完了する可能性があります。

ただし、この技術を活用するにはPythonの基本的な知識と、LLMのAPI利用経験が必要です。完全な初心者がいきなり使いこなすのは難しいでしょう。また、LLMのAPI利用にはコストがかかるため、クライアントワークで使う場合は、実行回数やステップ数を適切に設定して、コストを管理する必要があります。

作業時間への影響としては、マルチエージェントシステムの構築時間が大幅に短縮される点が大きいです。収益面では、これまで「難しすぎて受けられなかった案件」に対応できるようになる可能性があります。特に、AI自動化を求める企業クライアントとの取引で、より高度な提案ができるようになるでしょう。

まとめ

SmolAgentsは、マルチエージェントシステムの構築ハードルを下げる技術です。すでにPythonとLLM APIの経験がある方なら、GitHubで公開されているチュートリアルを試してみる価値があります。完全実装コードとノートブックは、Marktechpostのリポジトリで公開されています。ただし、実際のクライアントワークで使う前に、コスト管理と実行時間の最適化について検証しておくことをお勧めします。まずは小規模なタスクで動作を確認してから、徐々に複雑なシステムに適用していくのが現実的なアプローチでしょう。

コメント

タイトルとURLをコピーしました