feat: Add Graphviz-based agent visualization functionality

[MartinEBravo]

This pull request introduces functionality for visualizing agent structures using Graphviz. The changes include adding a new dependency, implementing functions to generate and draw graphs, and adding tests for these functions.

New functionality for visualizing agent structures:

• Added graphviz as a new dependency in pyproject.toml.
• Implemented functions in src/agents/visualizations.py to generate and draw graphs for agents using Graphviz. These functions include get_main_graph, get_all_nodes, get_all_edges, and draw_graph.

Testing the new visualization functionality:

• Added tests in tests/test_visualizations.py to verify the correctness of the graph generation and drawing functions. The tests cover get_main_graph, get_all_nodes, get_all_edges, and draw_graph.

For example, given the following code:

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

Generates the following image:

[rm-openai]

This is such a great PR/idea. Couple of small things to get it in:

1. Could you please add this to the agents/src/extensions directory? So the import would be from agents.extensions import visualization
2. Could you please add an optional dependency group? So you can do pip install openai-agents[visualization] to install it, and if you don't, there's an import error.

Thank you!

[rm-openai]

Suggested change

	from src.agents.agent import Agent
	from agents import Agent

[MartinEBravo]

Fixed!

[rm-openai]

Looks like tests are failing. I think you can add an optional dependency to pyproject.toml

[project.optional-dependencies]
visualization=["graphviz"]

[rm-openai]

I think you want handoff.agent_name not handoff.name, if it's a Handoff instance

[MartinEBravo]

Added isinstance check.

[rm-openai]

I don't think this is right - the handoffs list is list[Handoff | Agent]. So the node might be an Agent, but it might also be a Handoff object, where you don't statically know which agent is being handed off to. You can use handoff.agent_name for that

[MartinEBravo]

Added isinstance check.

[rm-openai]

THink you just need a rebase and should be good to go

[rm-openai]

you might need a rebase, as [project.optional-dependencies] exists already. Also perhaps viz is a shorter optional dependency heh

[rm-openai]

Yeah looks like this lack of rebase is causing all the checks to fail.

[rm-openai]

Oof unfortunately checks still failing. I tried to update them but you may need to run make tests, make mypy etc locally.

[MartinEBravo]

Ok, now it cannot fail.