Support orientation in degree centrality

Author: IoannisPanagiotas

mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_handlers.py

@@ -292,7 +292,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 +338,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"),
         )
 
 

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": [],
         },