エージェント可視化
エージェントの可視化では、 Graphviz を使用してエージェントとその関係の構造化されたグラフィカル表現を生成できます。これは、アプリケーション内でエージェント、ツール、ハンドオフがどのように連携するかを理解するのに役立ちます。
インストール
任意の viz 依存関係グループをインストールします:
グラフ生成
draw_graph 関数を使用してエージェントの可視化を生成できます。この関数は有向グラフを作成し、以下のように表現します:
- エージェント は黄色のボックスで表されます。
- MCP サーバー は灰色のボックスで表されます。
- ツール は緑色の楕円で表されます。
- ハンドオフ は、あるエージェントから別のエージェントへの有向エッジで表されます。
使用例
import os
from agents import Agent, function_tool
from agents.mcp.server import MCPServerStdio
from agents.extensions.visualization import draw_graph
@function_tool
def get_weather(city: str) -> str:
return f"The weather in {city} is sunny."
spanish_agent = Agent(
name="Spanish agent",
instructions="You only speak Spanish.",
)
english_agent = Agent(
name="English agent",
instructions="You only speak English",
)
current_dir = os.path.dirname(os.path.abspath(__file__))
samples_dir = os.path.join(current_dir, "sample_files")
mcp_server = MCPServerStdio(
name="Filesystem Server, via npx",
params={
"command": "npx",
"args": ["-y", "@modelcontextprotocol/server-filesystem", samples_dir],
},
)
triage_agent = Agent(
name="Triage agent",
instructions="Handoff to the appropriate agent based on the language of the request.",
handoffs=[spanish_agent, english_agent],
tools=[get_weather],
mcp_servers=[mcp_server],
)
draw_graph(triage_agent)
これは、 トリアージ エージェント とそのサブエージェントおよびツールへの接続を視覚的に表すグラフを生成します。
可視化の理解
生成されるグラフには以下が含まれます:
- エントリーポイントを示す 開始ノード (
__start__) - 黄色で塗りつぶされた 長方形 のエージェント
- 緑で塗りつぶされた 楕円 のツール
- 灰色で塗りつぶされた 長方形 の MCP サーバー
- 相互作用を示す有向エッジ:
- エージェント間のハンドオフは 実線の矢印
- ツールの呼び出しは 点線の矢印
- MCP サーバーの呼び出しは 破線の矢印
- 実行の終了地点を示す 終了ノード (
__end__)
注意: MCP サーバーは、最近の agents パッケージのバージョンでレンダリングされます（ v0.2.8 で確認済み）。可視化に MCP のボックスが表示されない場合は、最新リリースにアップグレードしてください。
グラフのカスタマイズ
グラフの表示
デフォルトでは、draw_graph はグラフをインライン表示します。別ウィンドウで表示するには、次のように記述します:
グラフの保存
デフォルトでは、draw_graph はグラフをインライン表示します。ファイルとして保存するには、ファイル名を指定します:
これにより、作業ディレクトリに agent_graph.png が生成されます。