Merge pull request #25 from brs96/fix-name-match-and-cleanup

Fix name match and cleanup

Author: IoannisPanagiotas

CHANGELOG.md

@@ -5,6 +5,8 @@
 
 ### Bug Fixes
 1. Fix a bug where disconnected nodes were not projected.
+2. Fix a bug where multiple nodes containing the identifying name could be nondeterministically matched. The user must now specify the exact name.
+3. Support weighted degree centrality.
 
 ### Other Changes
 

mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_handlers.py

@@ -213,6 +213,7 @@ def execute(self, arguments: Dict[str, Any]) -> Any:
             nodes=arguments.get("nodes"),
             nodeIdentifierProperty=arguments.get("nodeIdentifierProperty"),
             orientation=arguments.get("orientation"),
+            relationshipWeightProperty=arguments.get("relationshipWeightProperty"),
         )
 
 

mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_specs.py

@@ -190,6 +190,10 @@
                     "type": "string",
                     "description": "The orientation used to compute node degrees. Supported orientations are NATURAL (for out-degree), REVERSE (for in-degree) and UNDIRECTED (for both in-degree and out-degree) ",
                 },
+                "relationshipWeightProperty": {
+                    "type": "string",
+                    "description": "Property of the relationship to use for weighting. If not specified, all relationships are treated equally.",
+                },
             },
             "required": [],
         },

mcp_server/src/mcp_server_neo4j_gds/node_translator.py

@@ -15,7 +15,7 @@ def translate_identifiers_to_ids(
             query = f"""
                     UNWIND $names AS name
                     MATCH (s)
-                    WHERE toLower(s.{node_identifier_property}) CONTAINS toLower(name)
+                    WHERE toLower(s.{node_identifier_property}) = toLower(name)
                     RETURN id(s) as node_id
                     """
             df = gds.run_cypher(
@@ -30,7 +30,7 @@ def translate_identifiers_to_ids(
             # Handle single  node name
             query = f"""
                     MATCH (s)
-                    WHERE toLower(s.{node_identifier_property}) CONTAINS toLower($name)
+                    WHERE toLower(s.{node_identifier_property}) = toLower($name)
                     RETURN id(s) as node_id
                     """
             df = gds.run_cypher(
@@ -77,7 +77,7 @@ def filter_identifiers(
     query = f"""
             UNWIND $names AS name
             MATCH (s)
-            WHERE toLower(s.{node_identifier_property}) CONTAINS toLower(name)
+            WHERE toLower(s.{node_identifier_property}) = toLower(name)
             RETURN id(s) as node_id
             """
     df = gds.run_cypher(

mcp_server/src/mcp_server_neo4j_gds/server.py

@@ -134,7 +134,7 @@ async def handle_call_tool(
                 write_stream,
                 InitializationOptions(
                     server_name="neo4j_gds",
-                    server_version="0.1.0",
+                    server_version="0.3.0",
                     capabilities=server.get_capabilities(
                         notification_options=NotificationOptions(),
                         experimental_capabilities={},

mcp_server/tests/test_centrality_algorithms.py

@@ -239,6 +239,8 @@ async def test_degree_centrality(mcp_client):
         {
             "nodes": ["King's Cross St. Pancras", "Oxford Circus"],
             "nodeIdentifierProperty": "name",
+            "orientation": "NATURAL",
+            "relationshipWeightProperty": "distance",
         },
     )