智能体可视化
智能体可视化允许你使用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)
这将生成一个图，直观展示了分诊智能体（triage agent）的结构及其与子智能体和工具的连接关系。
可视化说明
生成的图包含：
- 一个表示入口点的起始节点（
__start__）。 - 用黄色填充并以矩形表示的智能体。
- 用绿色填充并以椭圆表示的工具。
- 用灰色填充并以矩形表示的 MCP 服务。
- 表示交互的有向边：
- 实线箭头：智能体到智能体的任务转移。
- 点线箭头：工具调用。
- 虚线箭头：MCP 服务调用。
- 一个表示执行终止位置的结束节点（
__end__）。
注意： 在较新的agents 包版本（在v0.2.8 验证）中会渲染 MCP 服务。如果你在可视化中未看到 MCP 方框，请升级到最新版本。
自定义图形
显示图形
默认情况下，draw_graph 会在内联显示图形。要在单独窗口中显示图形，请编写如下代码：
保存图形
默认情况下，draw_graph 会在内联显示图形。要将其保存为文件，请指定文件名：
这将在工作目录中生成agent_graph.png。