wip

Author: IoannisPanagiotas

mcp_server/src/mcp_server_neo4j_gds/community_algorithm_handlers.py

@@ -58,12 +58,12 @@ class KCoreDecompositionHandler(AlgorithmHandler):
     def k_core_decomposition(self, **kwargs):
         with projected_graph(self.gds, undirected=True) as G:
             logger.info("Running K-Core Decomposition")
-            kcore_decomposition_result = self.gds.kcore.stream(G)
+            k_core_decomposition_result = self.gds.kcore.stream(G)
 
         # Add node names to the results if nodeIdentifierProperty is provided
         node_identifier_property = kwargs.get("nodeIdentifierProperty")
         translate_ids_to_identifiers(
-            self.gds, node_identifier_property, kcore_decomposition_result
+            self.gds, node_identifier_property, k_core_decomposition_result
         )
 
     def execute(self, arguments: Dict[str, Any]) -> Any:

mcp_server/src/mcp_server_neo4j_gds/node_translator.py

@@ -59,6 +59,7 @@ def translate_ids_to_identifiers(
             for node_id in results[id_name]
         ]
         results[node_identifier_output_name] = node_name_values
+        assert 1 == results[node_identifier_output_name].nunique()
 
 
 def filter_identifiers(