From c745fe156ab4163a3508564cf5dd30d6f995cb9d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mart=C3=ADn=20Bravo?=
 <123977407+MartinEBravo@users.noreply.github.com>
Date: Tue, 18 Mar 2025 09:55:23 +0100
Subject: [PATCH] feat: add documentation for agent visualization using
 Graphviz

---
 docs/{visualizations.md => visualization.md} | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)
 rename docs/{visualizations.md => visualization.md} (97%)

diff --git a/docs/visualizations.md b/docs/visualization.md
similarity index 97%
rename from docs/visualizations.md
rename to docs/visualization.md
index 2a8bb28..6bb5ee5 100644
--- a/docs/visualizations.md
+++ b/docs/visualization.md
@@ -7,7 +7,7 @@ Agent visualization allows you to generate a structured graphical representation
 The visualization functionality relies on the **Graphviz** package. To use it, ensure you have Graphviz installed and add it as a dependency in `pyproject.toml`. Alternatively, install it directly via pip:
 
 ```bash
-pip install graphviz
+pip install openai-agents[visualization]
 ```
 
 ## Generating a Graph
@@ -22,7 +22,7 @@ You can generate an agent visualization using the `draw_graph` function. This fu
 
 ```python
 from agents import Agent, function_tool
-from agents.visualizations import draw_graph
+from agents.extensions.visualization import draw_graph
 
 @function_tool
 def get_weather(city: str) -> str: