智能体可视化

代理可视化功能允许您使用Graphviz生成代理及其关系的结构化图形表示。这对于理解应用程序中代理、工具和交接如何交互非常有用。

安装

安装可选的 viz 依赖组：

pip install "openai-agents[viz]"

生成图表

您可以使用draw_graph函数生成代理可视化。该函数创建一个有向图，其中：

代理(Agents)以黄色方框表示。
工具以绿色椭圆表示。
交接是指从一个智能体指向另一个智能体的有向边。

使用示例

from agents import Agent, function_tool
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",
)

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],
)

draw_graph(triage_agent)

Agent Graph

这将生成一个图表，直观展示分诊代理的结构及其与子代理和工具之间的连接关系。

理解可视化

生成的图表包含：

一个起始节点 (__start__) 表示入口点。
智能体以矩形形式表示，填充色为黄色。
工具以椭圆形式表示，并填充绿色。
表示交互关系的有向边：
实线箭头表示代理间的交接。
虚线箭头 表示工具调用。
一个结束节点 (__end__) 表示执行终止的位置。

自定义图表

显示图表

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

draw_graph(triage_agent).view()

保存图表

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

draw_graph(triage_agent, filename="agent_graph.png")

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