File size: 599 Bytes
79d0909 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | """Script to generate graph visualization."""
import os
import sys
# Add project root to path for direct script execution
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from src.core import setup_logging
from src.graph.builder import save_graph_visualization
setup_logging("INFO")
def main() -> None:
"""Generate and save graph visualization."""
output_dir = "data"
os.makedirs(output_dir, exist_ok=True)
output_path = os.path.join(output_dir, "rag_graph.png")
save_graph_visualization(output_path)
if __name__ == "__main__":
main()
|