RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add visualization functions for agent graphs

This commit is contained in:
Martín Bravo 2025-03-18 09:53:48 +01:00
parent dba2d4bd23
commit c7805669bc
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