エージェントの可視化
エージェントの可視化では、Graphviz を使用してエージェントとその関係の構造化されたグラフィカル表現を生成できます。これは、アプリケーション内でエージェント、ツール、ハンドオフがどのように相互作用するかを理解するのに役立ちます。
インストール
オプションのviz 依存関係グループをインストールします:
グラフの生成
draw_graph 関数を使ってエージェントの可視化を生成できます。この関数は次のような有向グラフを作成します:
- エージェント は黄色のボックスで表されます。
- MCP サーバー は灰色のボックスで表されます。
- ツール は緑色の楕円で表されます。
- ハンドオフ はあるエージェントから別のエージェントへの有向エッジです。
使用例
importosfromagentsimportAgent,function_toolfromagents.mcp.serverimportMCPServerStdiofromagents.extensions.visualizationimportdraw_graph@function_tooldefget_weather(city:str)->str:returnf"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__)
Note: MCP サーバーは最近のagents パッケージのバージョン(v0.2.8 で検証)でレンダリングされます。可視化に MCP ボックスが表示されない場合は、最新リリースにアップグレードしてください。
グラフのカスタマイズ
グラフの表示
デフォルトでは、draw_graph はグラフをインライン表示します。別ウィンドウで表示するには、次のように記述します:
グラフの保存
デフォルトでは、draw_graph はグラフをインライン表示します。ファイルとして保存するには、ファイル名を指定します:
これにより、作業ディレクトリにagent_graph.png が生成されます。