このページに掲載されているコードサンプルは、すべて Python です。
weave.init() を呼び出し、通常どおりライブラリを使用します。
-
Smolagents を Weave で使用する前に、必要なライブラリをインストールするか、最新バージョンにアップグレードしてください。次のコマンドは、
smolagents、openai、weave をインストールまたはアップグレードし、出力を抑制します。
pip install -U smolagents openai weave -qqq
-
Smolagents は、OpenAI、Hugging Face Transformers、Anthropic など、複数の LLM プロバイダをサポートしています。選択したプロバイダに対応する環境変数を設定して、APIキーを指定してください。
import os
import getpass
os.environ["OPENAI_API_KEY"] = getpass.getpass("Enter your OpenAI API key: ")
言語モデルアプリケーションのトレースを一元管理された場所に保存することは、開発時にも本番環境でも不可欠です。これらのトレースはデバッグに役立つだけでなく、アプリケーション改善のための貴重なデータセットにもなります。
Weave は Smolagents のトレースを自動的にキャプチャします。トラッキングを開始するには、weave.init() を呼び出して Weave を初期化し、その後は通常どおりライブラリを使用します。
次の例では、ツールを使用する LLM エージェントの推論 call を Weave でログする方法を示します。このシナリオでは、次のことを行います。
- Smolagents の
OpenAIServerModel を使用して、言語モデル (OpenAI の gpt-4o) を定義します。
- 必要に応じてエージェントが呼び出せる検索ツール (
DuckDuckGoSearchTool) を設定します。
- ツールとモデルを渡して、
ToolCallingAgent を構築します。
- 検索ツールをトリガーするクエリをエージェント経由で実行します。
- Weave は各関数とモデルの呼び出しをログし、Web インターフェースで確認できるようにします。
import weave
from smolagents import DuckDuckGoSearchTool, OpenAIServerModel, ToolCallingAgent
# Weaveを初期化する
weave.init(project_name="smolagents")
# SmolagentsがサポートするLLMプロバイダーを定義する
model = OpenAIServerModel(model_id="gpt-4o")
# クエリに基づいてDuckDuckGoウェブ検索ツールを定義する
search_tool = DuckDuckGoSearchTool()
# ツール呼び出しエージェントを定義する
agent = ToolCallingAgent(tools=[search_tool], model=model)
answer = agent.run(
"Get me just the title of the page at url 'https://wandb.ai/geekyrakshit/story-illustration/reports/Building-a-GenAI-assisted-automatic-story-illustrator--Vmlldzo5MTYxNTkw'?"
)
smolagents の @tool で関数をデコレートするか、smolagents.Tool クラスを継承することで、agentic ワークフロー用のカスタムツールを定義できます。
Weave は、Smolagents ワークフロー内のカスタムツールの call を自動的にトラッキングします。次の例では、Weave でカスタム Smolagents ツールの call をログする方法を示します。
- カスタムの
get_weather 関数を定義し、Smolagents の @tool でデコレートすることで、エージェントが推論プロセスの一部としてこの関数を呼び出せるようにします。
- この関数は、location と、摂氏で出力するための optional フラグを受け取ります。
OpenAIServerModel を使用して言語モデルをインスタンス化します。- カスタムツールとモデルを使って
ToolCallingAgent を作成します。
- エージェントがクエリを実行すると、
get_weather ツールを選択して呼び出します。
- Weave は、モデルの推論とカスタムツールの呼び出しの両方を、argument と Return Value を含めてログします。
from typing import Optional
import weave
from smolagents import OpenAIServerModel, ToolCallingAgent, tool
weave.init(project_name="smolagents")
@tool
def get_weather(location: str, celsius: Optional[bool] = False) -> str:
"""
指定した場所の今後数日間の天気を取得します。
引数:
location: 場所。
celsius: 温度にセルシウスを使用するかどうか。
"""
return f"{location}の天気は晴れで、気温は7°C前後です。"
model = OpenAIServerModel(model_id="gpt-4o")
agent = ToolCallingAgent(tools=[get_weather], model=model)
answer = agent.run("東京の天気はどうですか？")
