admin-valentin/openai-agents-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add visualization functions for agent graphs

Browse source
This commit is contained in:
Martín Bravo 2025-03-18 09:53:48 +01:00
parent aff1d60ea1
commit f7c594da08
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/agents/visualizations.py → src/agents/extensions/visualizations.py
View file

@ -1,6 +1,6 @@
import graphviz
from src.agents.agent import Agent
from agents import Agent
def get_main_graph(agent: Agent) -> str:

4
tests/test_visualizations.py
View file

@ -3,8 +3,8 @@ from unittest.mock import Mock
import graphviz
import pytest
from src.agents.agent import Agent
from src.agents.visualizations import draw_graph, get_all_edges, get_all_nodes, get_main_graph
from agents import Agent
from agents.extensions.visualizations import draw_graph, get_all_edges, get_all_nodes, get_main_graph
@pytest.fixture