mcp-vector-search 0.15.7__py3-none-any.whl

mcp_vector_search/cli/commands/visualize/exporters/__init__.py

@@ -0,0 +1,12 @@
+"""Export functionality for graph visualization data.
+
+This package contains modules for exporting graph data to various formats.
+"""
+
+from .html_exporter import export_to_html
+from .json_exporter import export_to_json
+
+__all__ = [
+    "export_to_html",
+    "export_to_json",
+]

mcp_vector_search/cli/commands/visualize/exporters/html_exporter.py

@@ -0,0 +1,33 @@
+"""HTML export functionality for visualization.
+
+This module handles creating standalone HTML files for the visualization.
+"""
+
+from pathlib import Path
+
+from rich.console import Console
+
+from ..templates.base import generate_html_template
+
+console = Console()
+
+
+def export_to_html(output_path: Path) -> None:
+    """Export visualization to standalone HTML file.
+
+    Args:
+        output_path: Path to output HTML file
+    """
+    # Generate HTML template
+    html_content = generate_html_template()
+
+    # Ensure output directory exists
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+
+    # Write to file
+    with open(output_path, "w") as f:
+        f.write(html_content)
+
+    console.print(
+        f"[green]✓[/green] Created visualization HTML at [cyan]{output_path}[/cyan]"
+    )

mcp_vector_search/cli/commands/visualize/exporters/json_exporter.py

@@ -0,0 +1,29 @@
+"""JSON export functionality for graph data.
+
+This module handles exporting graph data to JSON format.
+"""
+
+import json
+from pathlib import Path
+from typing import Any
+
+from rich.console import Console
+
+console = Console()
+
+
+def export_to_json(graph_data: dict[str, Any], output_path: Path) -> None:
+    """Export graph data to JSON file.
+
+    Args:
+        graph_data: Graph data dictionary containing nodes, links, and metadata
+        output_path: Path to output JSON file
+    """
+    # Ensure output directory exists
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+
+    # Write to file
+    with open(output_path, "w") as f:
+        json.dump(graph_data, f, indent=2)
+
+    console.print(f"[green]✓[/green] Exported graph data to [cyan]{output_path}[/cyan]")