| """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() | |