智能体可视化

智能体可视化功能允许你使用 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 服务器和下游任务转移。如果自定义 Handoff 没有可用的目标 Agent，它将仅呈现为一个具名目标，因此图形无法展开该目标背后的资源。

图中的节点由底层智能体、工具、MCP 服务器或自定义任务转移对象来标识，而非按显示名称标识。即使不同对象使用相同名称，它们仍会作为具有相同可见标签的独立节点，每条边也会连接到相应的对象。

可视化图解读

生成的图中包括：

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

注意：在较新版本的 agents 包中，MCP 服务器会显示在图中，包括已验证此行为的 v0.2.8。如果可视化图中没有显示 MCP 方框，请升级到最新版本。

图形自定义

图形显示

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

draw_graph(triage_agent).view()

图形保存

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

draw_graph(triage_agent, filename="agent_graph")

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