Merge pull request #64 from neo4j-contrib/data-modeling-remove-nvl-viz

* remove nvl viz
* shorten tool names

Author: a-s-g93

.gitignore

@@ -47,3 +47,4 @@ Thumbs.db
 __pycache__
 servers/mcp-neo4j-data-modeling/test.ipynb
 servers/mcp-neo4j-data-modeling/src/mcp_neo4j_data_modeling/temp.html
+mcp.json

servers/mcp-neo4j-data-modeling/CHANGELOG.md

@@ -1,12 +1,15 @@
 ## Next
 
 ### Fixed
+* Shorten tool names to comply with Cursor name length restrictions
 
 ### Changed
+* Removed NVL visualization due to compatibility issues
 
 ### Added
 * Code generation tools for ingestion queries
 * Resource that explains the recommended process of ingesting data into Neo4j
+* Mermaid visualization configuration generation
 
 ## v0.1.0
 

servers/mcp-neo4j-data-modeling/pyproject.toml

@@ -6,7 +6,6 @@ readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
     "mcp[cli]>=1.6.0",
-    "neo4j-viz>=0.3.1",
     "pydantic>=2.10.1",
 ]
 

servers/mcp-neo4j-data-modeling/src/mcp_neo4j_data_modeling/data_model.py

@@ -2,7 +2,6 @@
 from collections import Counter
 from typing import Any
 
-import neo4j_viz as nvl
 from pydantic import BaseModel, Field, ValidationInfo, field_validator
 
 NODE_COLOR_PALETTE = [
@@ -156,16 +155,6 @@ def all_properties_dict(self) -> dict[str, str]:
             props.update({self.key_property.name: f"{self.key_property.type} | KEY"})
         return props
 
-    def to_nvl(self) -> nvl.Node:
-        "Convert the node to a Neo4j Visualization Graph Node."
-        return nvl.Node(
-            id=self.label,
-            caption=self.label,
-            size=20,
-            caption_size=1,
-            properties=self.all_properties_dict,
-        )
-
     def get_mermaid_config_str(self) -> str:
         "Get the Mermaid configuration string for the node."
         props = [f"<br/>{self.key_property.name}: {self.key_property.type} | KEY"]
@@ -303,15 +292,6 @@ def all_properties_dict(self) -> dict[str, str]:
             props.update({self.key_property.name: f"{self.key_property.type} | KEY"})
         return props
 
-    def to_nvl(self) -> nvl.Relationship:
-        "Convert the relationship to a Neo4j Visualization Graph Relationship."
-        return nvl.Relationship(
-            source=self.start_node_label,
-            target=self.end_node_label,
-            caption=self.type,
-            properties=self.all_properties_dict,
-        )
-
     def get_mermaid_config_str(self) -> str:
         "Get the Mermaid configuration string for the relationship."
         props = (
@@ -511,13 +491,6 @@ def remove_relationship(
         except ValueError:
             pass
 
-    def to_nvl(self) -> nvl.VisualizationGraph:
-        "Convert the data model to a Neo4j Visualization Graph."
-        return nvl.VisualizationGraph(
-            nodes=[n.to_nvl() for n in self.nodes],
-            relationships=[r.to_nvl() for r in self.relationships],
-        )
-
     def _generate_mermaid_config_styling_str(self) -> str:
         "Generate the Mermaid configuration string for the data model."
         node_color_config = ""

servers/mcp-neo4j-data-modeling/src/mcp_neo4j_data_modeling/server.py

@@ -1,6 +1,4 @@
 import logging
-import os
-import webbrowser
 from typing import Any, Literal
 
 from mcp.server.fastmcp import FastMCP
@@ -20,9 +18,7 @@
 def create_mcp_server() -> FastMCP:
     """Create an MCP server instance for data modeling."""
 
-    mcp: FastMCP = FastMCP(
-        "mcp-neo4j-data-modeling", dependencies=["pydantic", "webbrowser"]
-    )
+    mcp: FastMCP = FastMCP("mcp-neo4j-data-modeling", dependencies=["pydantic"])
 
     @mcp.resource("resource://schema/node")
     def node_schema() -> dict[str, Any]:
@@ -107,35 +103,6 @@ def validate_data_model(
             logger.error(f"Validation error: {e}")
             raise ValueError(f"Validation error: {e}")
 
-    @mcp.tool()
-    def visualize_data_model_in_browser(data_model: DataModel) -> None:
-        "Open an interactive graph visualization in the default web browser. Validates the data model before opening the visualization. Warning: May not be useable in some environments such as Docker and Claude Desktop."
-        logger.info("Validating the data model.")
-        try:
-            dm_validated = DataModel.model_validate(data_model, strict=True)
-        except ValidationError as e:
-            logger.error(f"Validation error: {e}")
-            raise ValueError(f"Validation error: {e}")
-
-        def open_html_in_browser(html_content, filename="temp.html"):
-            """Opens an HTML string in the default web browser.
-
-            Args:
-                html_content: The HTML content as a string.
-                filename: The name of the temporary HTML file.
-            """
-            with open(filename, "w") as f:
-                f.write(html_content)
-
-            # Construct the file URL
-            file_url = "file://" + os.path.realpath(filename)
-            webbrowser.open_new_tab(file_url)
-
-        logger.info(
-            "Opening an interactive graph visualization in the default web browser."
-        )
-        open_html_in_browser(dm_validated.to_nvl().render().data)
-
     @mcp.tool()
     def load_from_arrows_json(arrows_data_model_dict: dict[str, Any]) -> DataModel:
         "Load a data model from the Arrows web application format. Returns a data model as a JSON string."
@@ -160,7 +127,7 @@ def get_mermaid_config_str(data_model: DataModel) -> str:
         return dm_validated.get_mermaid_config_str()
 
     @mcp.tool()
-    def get_node_cypher_ingest_query_for_many_records(
+    def get_node_cypher_ingest_query(
         node: Node = Field(description="The node to get the Cypher query for."),
     ) -> str:
         """
@@ -174,7 +141,7 @@ def get_node_cypher_ingest_query_for_many_records(
         return node.get_cypher_ingest_query_for_many_records()
 
     @mcp.tool()
-    def get_relationship_cypher_ingest_query_for_many_records(
+    def get_relationship_cypher_ingest_query(
         data_model: DataModel = Field(
             description="The data model snippet that contains the relationship, start node and end node."
         ),