コンテンツにスキップ

エージェントの可視化

エージェントの可視化では、 Graphviz を使用して、エージェント、および他のエージェント、ツール、MCPサーバーとの接続を構造化されたグラフィカル表現として生成できます。これは、アプリケーション内でエージェント、ツール、ハンドオフがどのように連携するかを理解するのに役立ちます。

インストール

オプションの viz 依存関係グループをインストールします。

pip install "openai-agents[viz]"

グラフの生成

draw_graph 関数を使用して、エージェントの可視化を生成できます。この関数は、以下の要素を含む有向グラフを作成します。

  • エージェント は黄色のボックスで表されます。
  • MCPサーバー は灰色のボックスで表されます。
  • ツール は緑色の楕円で表されます。
  • ハンドオフ は、あるエージェントから別のエージェントへの有向エッジとして表されます。

使用例

import os

from agents import Agent, handoff
from agents.decorators import tool
from agents.mcp.server import MCPServerStdio
from agents.extensions.visualization import draw_graph


@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=[handoff(spanish_agent), handoff(english_agent)],
    tools=[get_weather],
    mcp_servers=[mcp_server],
)

draw_graph(triage_agent)

エージェントグラフ

これにより、 トリアージエージェント の構造と、サブエージェントやツールとの接続を視覚的に表すグラフが生成されます。

draw_graph() は、handoffs で直接指定されたターゲットエージェント、または handoff(agent) を通じて登録されたターゲットエージェントを再帰的に展開します。どちらの形式でも、グラフには各ターゲットのツール、MCPサーバー、およびそれ以降のハンドオフが含まれます。利用可能なターゲット Agent がないカスタム Handoff は、名前付きの宛先としてのみレンダリングされるため、グラフはその宛先の背後にあるリソースを展開できません。

グラフノードは、表示名ではなく、基になるエージェント、ツール、MCPサーバー、またはカスタムハンドオフオブジェクトによって識別されます。同じ名前を共有する別個のオブジェクトは、同じ表示ラベルを持つ別々のノードとして維持され、各エッジは対応するオブジェクトに接続されます。

可視化の構成

生成されたグラフには、以下が含まれます。

  • エントリーポイントを示す 開始ノード（__start__）。
  • 黄色で塗りつぶされた 長方形 で表されるエージェント。
  • 緑色で塗りつぶされた 楕円 で表されるツール。
  • 灰色で塗りつぶされた 長方形 で表されるMCPサーバー。
  • インタラクションを示す有向エッジ。
  • エージェント間のハンドオフを示す 実線の矢印。
  • ツール呼び出しを示す 点線の矢印。
  • MCPサーバー呼び出しを示す 破線の矢印。
  • 実行が終了する位置を示す 終了ノード（__end__）。

注: MCPサーバーは、この動作が確認されている v0.2.8 を含む、最近のバージョンの agents パッケージでレンダリングされます。可視化にMCPボックスが表示されない場合は、最新リリースにアップグレードしてください。

グラフのカスタマイズ

グラフの表示

デフォルトでは、draw_graph はグラフをインラインで表示します。別のウィンドウにグラフを表示するには、以下のように記述します。

draw_graph(triage_agent).view()

グラフの保存

デフォルトでは、draw_graph はグラフをインラインで表示します。ファイルとして保存するには、ファイル名を指定します。

draw_graph(triage_agent, filename="agent_graph")

これにより、作業ディレクトリに agent_graph.png が生成されます。