Support orientation in degree centrality

Author: IoannisPanagiotas

mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_handlers.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict
 
-
 from .algorithm_handler import AlgorithmHandler
 from .gds import projected_graph
 
@@ -17,7 +16,7 @@ def article_rank(self, **kwargs):
                 k: v
                 for k, v in kwargs.items()
                 if v is not None
-                and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
+                   and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
             }
             node_names = kwargs.get("nodes", None)
             node_identifier_property = kwargs.get("nodeIdentifierProperty")
@@ -292,7 +291,13 @@ def execute(self, arguments: Dict[str, Any]) -> Any:
 class DegreeCentralityHandler(AlgorithmHandler):
     def degree_centrality(self, **kwargs):
         with projected_graph(self.gds) as G:
-            centrality = self.gds.degree.stream(G)
+            params = {
+                k: v
+                for k, v in kwargs.items()
+                if v is not None and k not in ["nodes", "nodeIdentifierProperty"]
+            }
+            logger.info(f"Degree centrality parameters: {params}")
+            centrality = self.gds.degree.stream(G, **params)
 
         # Add node names to the results if nodeIdentifierProperty is provided
         node_identifier_property = kwargs.get("nodeIdentifierProperty")
@@ -332,6 +337,7 @@ def execute(self, arguments: Dict[str, Any]) -> Any:
         return self.degree_centrality(
             nodes=arguments.get("nodes"),
             nodeIdentifierProperty=arguments.get("nodeIdentifierProperty"),
+            orientation=arguments.get("orientation"),
         )
 
 
@@ -342,7 +348,7 @@ def eigenvector_centrality(self, **kwargs):
                 k: v
                 for k, v in kwargs.items()
                 if v is not None
-                and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
+                   and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
             }
             node_identifier_property = kwargs.get("nodeIdentifierProperty")
             source_nodes = kwargs.get("sourceNodes", None)
@@ -440,7 +446,7 @@ def pagerank(self, **kwargs):
                 k: v
                 for k, v in kwargs.items()
                 if v is not None
-                and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
+                   and k not in ["nodes", "nodeIdentifierProperty", "sourceNodes"]
             }
             node_identifier_property = kwargs.get("nodeIdentifierProperty")
             source_nodes = kwargs.get("sourceNodes", None)

mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_specs.py

@@ -186,6 +186,10 @@
                     "type": "string",
                     "description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
                 },
+                "orientation": {
+                    "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) ",
+                },
             },
             "required": [],
         },