コンテンツにスキップ

エージェント可視化

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

インストール

任意の viz 依存関係グループをインストールします:

pip install "openai-agents[viz]"

グラフの生成

draw_graph 関数を使用してエージェントの可視化を生成できます。この関数は、次のような有向グラフを作成します:

  • Agents は黄色のボックスで表されます。
  • MCP Servers は灰色のボックスで表されます。
  • Tools は緑の楕円で表されます。
  • Handoffs は、あるエージェントから別のエージェントへの有向エッジです。

使用例

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 node (__start__)。
  • 黄色で塗りつぶされた rectangles で表されるエージェント。
  • 緑で塗りつぶされた ellipses で表されるツール。
  • 灰色で塗りつぶされた MCP Servers を表す rectangles
  • 相互作用を示す有向エッジ:
  • エージェント間のハンドオフには Solid arrows
  • ツール呼び出しには Dotted arrows
  • MCP サーバー呼び出しには Dashed arrows
  • 実行が終了する場所を示す end node (__end__)。

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

グラフのカスタマイズ

グラフの表示

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

draw_graph(triage_agent).view()

グラフの保存

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

draw_graph(triage_agent, filename="agent_graph")

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