feat: add visualization functions for agent graphs

Author: MartinEBravo

src/agents/visualizations.py renamed to src/agents/extensions/visualizations.py

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

tests/test_visualizations.py

@@ -3,8 +3,8 @@
 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