智能体可视化

智能体可视化允许你使用 Graphviz 生成智能体及其关系的结构化图形表示。这有助于理解在应用中智能体、工具与任务转移如何交互。

安装

安装可选的 viz 依赖组：

pip install "openai-agents[viz]"

生成图形

你可以使用 draw_graph 函数生成智能体可视化。该函数会创建一个有向图，其中：

智能体 用黄色方框表示。
MCP Servers 用灰色方框表示。
工具用绿色椭圆表示。
任务转移 是从一个智能体指向另一个智能体的有向边。

示例用法

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)

Agent Graph

这将生成一个图形，直观展示了分诊智能体及其与子智能体和工具的连接结构。

理解可视化

生成的图形包含：

一个表示入口的起始节点（__start__）。
用矩形且填充黄色表示的智能体。
用椭圆且填充绿色表示的工具。
用矩形且填充灰色表示的 MCP Servers。
表示交互的有向边：
实线箭头 表示智能体之间的任务转移。
虚线点状箭头 表示工具调用。
虚线箭头 表示 MCP 服务调用。
一个表示执行终止位置的结束节点（__end__）。

注意： 在较新的 agents 包版本（已在 v0.2.8 验证）中会渲染 MCP servers。若你的可视化中未显示 MCP 方框，请升级到最新版本。

自定义图形

显示图形

默认情况下，draw_graph 会内联展示图形。若要在单独窗口中显示，请编写如下代码：

draw_graph(triage_agent).view()

保存图形

默认情况下，draw_graph 会内联展示图形。若要将其保存为文件，请指定文件名：

draw_graph(triage_agent, filename="agent_graph")

这将在工作目录中生成 agent_graph.png。