mcp-vector-search 0.8.2__py3-none-any.whl → 0.8.4__py3-none-any.whl

mcp_vector_search/__init__.py

@@ -1,7 +1,7 @@
 """MCP Vector Search - CLI-first semantic code search with MCP integration."""
 
-__version__ = "0.8.2"
-__build__ = "31"
+__version__ = "0.8.4"
+__build__ = "33"
 __author__ = "Robert Matsuoka"
 __email__ = "bobmatnyc@gmail.com"
 

mcp_vector_search/cli/commands/visualize.py

@@ -166,9 +166,30 @@ def serve(
     """
     import http.server
     import os
+    import socket
     import socketserver
     import webbrowser
 
+    # Find free port in range 8080-8099
+    def find_free_port(start_port: int = 8080, end_port: int = 8099) -> int:
+        """Find a free port in the given range."""
+        for test_port in range(start_port, end_port + 1):
+            try:
+                with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+                    s.bind(("", test_port))
+                    return test_port
+            except OSError:
+                continue
+        raise OSError(f"No free ports available in range {start_port}-{end_port}")
+
+    # Use specified port or find free one
+    if port == 8080:  # Default port, try to find free one
+        try:
+            port = find_free_port(8080, 8099)
+        except OSError as e:
+            console.print(f"[red]✗ {e}[/red]")
+            raise typer.Exit(1)
+
     # Get visualization directory
     viz_dir = Path(__file__).parent.parent.parent / "visualization"
 
@@ -364,9 +385,8 @@ def _create_visualization_html(html_file: Path) -> None:
     <div id="controls">
         <h1>🔍 Code Graph</h1>
 
-        <div class="control-group">
-            <label>Load Graph Data:</label>
-            <input type="file" id="fileInput" accept=".json">
+        <div class="control-group" id="loading">
+            <label>⏳ Loading graph data...</label>
         </div>
 
         <h3>Legend</h3>
@@ -408,34 +428,79 @@ def _create_visualization_html(html_file: Path) -> None:
         const g = svg.append("g");
         const tooltip = d3.select("#tooltip");
         let simulation;
+        let allNodes = [];
+        let allLinks = [];
+        let visibleNodes = new Set();
+        let collapsedNodes = new Set();
 
         function visualizeGraph(data) {
             g.selectAll("*").remove();
 
-            simulation = d3.forceSimulation(data.nodes)
-                .force("link", d3.forceLink(data.links).id(d => d.id).distance(100))
+            allNodes = data.nodes;
+            allLinks = data.links;
+
+            // Find root nodes (nodes without parents or depth 0/1)
+            const rootNodes = allNodes.filter(n =>
+                !n.parent_id || n.depth === 0 || n.depth === 1 || n.type === 'module'
+            );
+
+            // Start with only root nodes visible
+            visibleNodes = new Set(rootNodes.map(n => n.id));
+            collapsedNodes = new Set(rootNodes.map(n => n.id));
+
+            renderGraph();
+        }
+
+        function renderGraph() {
+            const visibleNodesList = allNodes.filter(n => visibleNodes.has(n.id));
+            const visibleLinks = allLinks.filter(l =>
+                visibleNodes.has(l.source.id || l.source) &&
+                visibleNodes.has(l.target.id || l.target)
+            );
+
+            simulation = d3.forceSimulation(visibleNodesList)
+                .force("link", d3.forceLink(visibleLinks).id(d => d.id).distance(100))
                 .force("charge", d3.forceManyBody().strength(-400))
                 .force("center", d3.forceCenter(width / 2, height / 2))
                 .force("collision", d3.forceCollide().radius(40));
 
+            g.selectAll("*").remove();
+
             const link = g.append("g")
                 .selectAll("line")
-                .data(data.links)
+                .data(visibleLinks)
                 .join("line")
                 .attr("class", "link");
 
             const node = g.append("g")
                 .selectAll("g")
-                .data(data.nodes)
+                .data(visibleNodesList)
                 .join("g")
                 .attr("class", d => `node ${d.type}`)
                 .call(drag(simulation))
+                .on("click", toggleNode)
                 .on("mouseover", showTooltip)
                 .on("mouseout", hideTooltip);
 
+            // Add circles with expand indicator
             node.append("circle")
-                .attr("r", d => d.complexity ? Math.min(8 + d.complexity * 2, 25) : 12);
-
+                .attr("r", d => d.complexity ? Math.min(8 + d.complexity * 2, 25) : 12)
+                .attr("stroke", d => hasChildren(d) ? "#ffffff" : "none")
+                .attr("stroke-width", d => hasChildren(d) ? 2 : 0);
+
+            // Add expand/collapse indicator
+            node.filter(d => hasChildren(d))
+                .append("text")
+                .attr("class", "expand-indicator")
+                .attr("text-anchor", "middle")
+                .attr("dy", 5)
+                .style("font-size", "16px")
+                .style("font-weight", "bold")
+                .style("fill", "#ffffff")
+                .style("pointer-events", "none")
+                .text(d => collapsedNodes.has(d.id) ? "+" : "−");
+
+            // Add labels
             node.append("text")
                 .text(d => d.name)
                 .attr("dy", 30);
@@ -450,7 +515,61 @@ def _create_visualization_html(html_file: Path) -> None:
                 node.attr("transform", d => `translate(${d.x},${d.y})`);
             });
 
-            updateStats(data);
+            updateStats({nodes: visibleNodesList, links: visibleLinks, metadata: {total_files: allNodes.length}});
+        }
+
+        function hasChildren(node) {
+            return allLinks.some(l => (l.source.id || l.source) === node.id);
+        }
+
+        function toggleNode(event, d) {
+            event.stopPropagation();
+
+            if (!hasChildren(d)) return;
+
+            if (collapsedNodes.has(d.id)) {
+                // Expand: show children
+                expandNode(d);
+            } else {
+                // Collapse: hide children
+                collapseNode(d);
+            }
+
+            renderGraph();
+        }
+
+        function expandNode(node) {
+            collapsedNodes.delete(node.id);
+
+            // Find direct children
+            const children = allLinks
+                .filter(l => (l.source.id || l.source) === node.id)
+                .map(l => allNodes.find(n => n.id === (l.target.id || l.target)))
+                .filter(n => n);
+
+            children.forEach(child => {
+                visibleNodes.add(child.id);
+                collapsedNodes.add(child.id); // Children start collapsed
+            });
+        }
+
+        function collapseNode(node) {
+            collapsedNodes.add(node.id);
+
+            // Hide all descendants recursively
+            function hideDescendants(parentId) {
+                const children = allLinks
+                    .filter(l => (l.source.id || l.source) === parentId)
+                    .map(l => l.target.id || l.target);
+
+                children.forEach(childId => {
+                    visibleNodes.delete(childId);
+                    collapsedNodes.delete(childId);
+                    hideDescendants(childId);
+                });
+            }
+
+            hideDescendants(node.id);
         }
 
         function showTooltip(event, d) {
@@ -504,23 +623,29 @@ def _create_visualization_html(html_file: Path) -> None:
             `);
         }
 
-        document.getElementById("fileInput").addEventListener("change", (event) => {
-            const file = event.target.files[0];
-            if (file) {
-                const reader = new FileReader();
-                reader.onload = (e) => {
-                    const data = JSON.parse(e.target.result);
+        // Auto-load graph data on page load
+        window.addEventListener('DOMContentLoaded', () => {
+            const loadingEl = document.getElementById('loading');
+
+            fetch("chunk-graph.json")
+                .then(response => {
+                    if (!response.ok) {
+                        throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+                    }
+                    return response.json();
+                })
+                .then(data => {
+                    loadingEl.innerHTML = '<label style="color: #238636;">✓ Graph loaded successfully</label>';
+                    setTimeout(() => loadingEl.style.display = 'none', 2000);
                     visualizeGraph(data);
-                };
-                reader.readAsText(file);
-            }
+                })
+                .catch(err => {
+                    loadingEl.innerHTML = `<label style="color: #f85149;">✗ Failed to load graph data</label><br>` +
+                                         `<small style="color: #8b949e;">${err.message}</small><br>` +
+                                         `<small style="color: #8b949e;">Run: mcp-vector-search visualize export</small>`;
+                    console.error("Failed to load graph:", err);
+                });
         });
-
-        // Try to load default data
-        fetch("chunk-graph.json")
-            .then(response => response.json())
-            .then(data => visualizeGraph(data))
-            .catch(err => console.log("No default graph found. Please load a JSON file."));
     </script>
 </body>
 </html>'''

mcp_vector_search/core/models.py

@@ -53,7 +53,13 @@ class CodeChunk:
         # Generate chunk ID if not provided
         if self.chunk_id is None:
             import hashlib
-            id_string = f"{self.file_path}:{self.chunk_type}:{self.start_line}:{self.end_line}"
+            import uuid
+
+            # Include name to differentiate chunks at same location
+            name = self.function_name or self.class_name or ""
+            # Add UUID to guarantee uniqueness even for identical chunks
+            unique_suffix = str(uuid.uuid4())[:8]
+            id_string = f"{self.file_path}:{self.chunk_type}:{name}:{self.start_line}:{self.end_line}:{unique_suffix}"
             self.chunk_id = hashlib.sha256(id_string.encode()).hexdigest()[:16]
 
     @property

{mcp_vector_search-0.8.2.dist-info → mcp_vector_search-0.8.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-vector-search
-Version: 0.8.2
+Version: 0.8.4
 Summary: CLI-first semantic code search with MCP integration
 Project-URL: Homepage, https://github.com/bobmatnyc/mcp-vector-search
 Project-URL: Documentation, https://mcp-vector-search.readthedocs.io

{mcp_vector_search-0.8.2.dist-info → mcp_vector_search-0.8.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-mcp_vector_search/__init__.py,sha256=SJWzBn_EMpgwyAGK_RAudIBKV_lD9imaFJvi0dh62pc,299
+mcp_vector_search/__init__.py,sha256=M5nv5AoHPpQkDpjs3z7LVlatwiz3GoyAtx5eDyhiAx0,299
 mcp_vector_search/py.typed,sha256=lCKeV9Qcn9sGtbRsgg-LJO2ZwWRuknnnlmomq3bJFH0,43
 mcp_vector_search/cli/__init__.py,sha256=TNB7CaOASz8u3yHWLbNmo8-GtHF0qwUjVKWAuNphKgo,40
 mcp_vector_search/cli/didyoumean.py,sha256=F_ss-EX4F9RgnMsEhdTwLpyNCah9SqnBZc2tBtzASck,15918
@@ -19,7 +19,7 @@ mcp_vector_search/cli/commands/mcp.py,sha256=Mk4g43R9yRiJVMxsDFUsZldKqY0yi2coQmh
 mcp_vector_search/cli/commands/reset.py,sha256=bsIT6zjDf6gsvIkVaRaUClYzlTyNe--8t0NWkBY0ldU,13724
 mcp_vector_search/cli/commands/search.py,sha256=yyou7wO9qZ_w2oiKdyOrk2WUxvkFpc-Up8hpflxYlyw,24802
 mcp_vector_search/cli/commands/status.py,sha256=sa_0QHioCmPF5A7obqV2ls-9kmX_JYo7nq3XUe1dmrg,19630
-mcp_vector_search/cli/commands/visualize.py,sha256=QLefmVaeRUbdnabtEuCTkZXtVlL4DLVk4zeITFDGPww,16579
+mcp_vector_search/cli/commands/visualize.py,sha256=tipe_QLjkZboqEz8SfIx5mjYrAenqrKsQPnXkgG7GBg,21398
 mcp_vector_search/cli/commands/watch.py,sha256=2pyWRoo4fIppFnyQ4sW4IBLHmpb_IwnTjRnzHkVBPcQ,8927
 mcp_vector_search/config/__init__.py,sha256=r_qAQkU5gc0EQ2pv8EQARACe4klhrR_WRJqCb9lfGc0,54
 mcp_vector_search/config/constants.py,sha256=afXR6SvLLd8QYY4MG4s1vq-hCJiQsE5PhnE-XG9lvb4,1092
@@ -34,7 +34,7 @@ mcp_vector_search/core/exceptions.py,sha256=3bCjT8wmrLz_0e_Tayr90049zNTKYFWZa19k
 mcp_vector_search/core/factory.py,sha256=tM6Ft-V9buF7nn9xbRMU1ngji-BJOKt6BhtfQhFLmF4,10384
 mcp_vector_search/core/git_hooks.py,sha256=xOfPpzgKoNTwM-vbhAihUucgudBQk45bCAVR5zJOFlQ,10878
 mcp_vector_search/core/indexer.py,sha256=IpCzP50wLOttWuUI-NE0qwYq-LlbDo5lrGfVMFMzwAM,29089
-mcp_vector_search/core/models.py,sha256=h54wTwwXR5qU-YOVi8q8NEFmpnQfC4S0ER2mMJqn4bk,8512
+mcp_vector_search/core/models.py,sha256=ynhb6-YuY7m4FwQBHdDbts3hYbJdr8UeYmQkag3EEKA,8814
 mcp_vector_search/core/project.py,sha256=l81uc5B4CB8VXDbcHzF-_CagxIERDh23tH0iNqTePTs,10403
 mcp_vector_search/core/scheduler.py,sha256=PBSlu-ieDYCXOMGYY7QKv9UReFEDPHNmwnUv_xb4vxg,11761
 mcp_vector_search/core/search.py,sha256=9OC8-KwWdbw4y4QPQ-VXfz0encVHTJWYLtah3_chqG8,33682
@@ -57,8 +57,8 @@ mcp_vector_search/utils/__init__.py,sha256=Eq6lY-oPMfCt-GpPUbg9QbmTHuQVmTaVDBMU2
 mcp_vector_search/utils/gitignore.py,sha256=GiHQu9kv9PRLsWuNS8kbpXsTaBdhlsSHTu1NrZ8Ug5Y,8162
 mcp_vector_search/utils/timing.py,sha256=THC7mfbTYnUpnnDcblgQacYMzbEkfFoIShx6plmhCgg,11285
 mcp_vector_search/utils/version.py,sha256=d7fS-CLemxb8UzZ9j18zH0Y0Ud097ljKKYYOPulnGPE,1138
-mcp_vector_search-0.8.2.dist-info/METADATA,sha256=hyZnlaTWcDJ5SmHpQNVADg26RpQxy9pO-iZXgD-kWiw,19120
-mcp_vector_search-0.8.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_vector_search-0.8.2.dist-info/entry_points.txt,sha256=y3Ygtc_JiBchNEIL-tPABo7EbzBExGAxwGdkkeP5D2I,86
-mcp_vector_search-0.8.2.dist-info/licenses/LICENSE,sha256=FqZUgGJH_tZKZLQsMCpXaLawRyLmyFKRVfMwYyEcyTs,1072
-mcp_vector_search-0.8.2.dist-info/RECORD,,
+mcp_vector_search-0.8.4.dist-info/METADATA,sha256=MZyBEwQ8cB6X5VBBPWdoAqnj_VLrD5hTLuJchLIo0IA,19120
+mcp_vector_search-0.8.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_vector_search-0.8.4.dist-info/entry_points.txt,sha256=y3Ygtc_JiBchNEIL-tPABo7EbzBExGAxwGdkkeP5D2I,86
+mcp_vector_search-0.8.4.dist-info/licenses/LICENSE,sha256=FqZUgGJH_tZKZLQsMCpXaLawRyLmyFKRVfMwYyEcyTs,1072
+mcp_vector_search-0.8.4.dist-info/RECORD,,