Skip to content

Commit 63b32ad

Browse files
brs96brs96
authored
Merge pull request #23 from neo4j-contrib/node-labels
algorithms can specify node labels if they want to now
2 parents 6b76e4f + e6bc14d commit 63b32ad

File tree

10 files changed

+423
-25
lines changed

10 files changed

+423
-25
lines changed

‎CHANGELOG.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
1. Fix a bug where disconnected nodes were not projected.
88
2. Fix a bug where multiple nodes containing the identifying name could be nondeterministically matched. The user must now specify the exact name.
99
3. Support weighted degree centrality.
10+
4. Support projecting specific node labels.
1011

1112
### Other Changes
1213

‎mcp_server/src/mcp_server_neo4j_gds/centrality_algorithm_specs.py‎

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,11 @@
6060
"Supported values are None, MinMax, Max, Mean, Log, and StdScore. "
6161
"To apply scaler-specific configuration, use the Map syntax: {scaler: 'name', ...}.",
6262
},
63+
"nodeLabels": {
64+
"type": "array",
65+
"items": {"type": "string"},
66+
"description": "The node labels used to project and run the article rank on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
67+
},
6368
},
6469
"required": [],
6570
},
@@ -74,6 +79,11 @@
7479
"type": "string",
7580
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
7681
},
82+
"nodeLabels": {
83+
"type": "array",
84+
"items": {"type": "string"},
85+
"description": "The node labels used to project and run Articulation points on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
86+
},
7787
},
7888
"required": [],
7989
},
@@ -89,6 +99,11 @@
8999
"items": {"type": "string"},
90100
"description": "List of node names to filter betweenness centrality results for.",
91101
},
102+
"nodeLabels": {
103+
"type": "array",
104+
"items": {"type": "string"},
105+
"description": "The node labels used to project and run Betweenness Centrality on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
106+
},
92107
"nodeIdentifierProperty": {
93108
"type": "string",
94109
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -111,6 +126,11 @@
111126
inputSchema={
112127
"type": "object",
113128
"properties": {
129+
"nodeLabels": {
130+
"type": "array",
131+
"items": {"type": "string"},
132+
"description": "The node labels used to project and run Bridges on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
133+
},
114134
"nodeIdentifierProperty": {
115135
"type": "string",
116136
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -126,6 +146,11 @@
126146
inputSchema={
127147
"type": "object",
128148
"properties": {
149+
"nodeLabels": {
150+
"type": "array",
151+
"items": {"type": "string"},
152+
"description": "The node labels used to project and run CELF on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
153+
},
129154
"seedSetSize": {
130155
"type": "integer",
131156
"description": "The number of nodes that maximize the expected spread in the network.",
@@ -159,6 +184,11 @@
159184
"items": {"type": "string"},
160185
"description": "List of node names to filter closeness centrality results for.",
161186
},
187+
"nodeLabels": {
188+
"type": "array",
189+
"items": {"type": "string"},
190+
"description": "The node labels used to project and run closeness centrality on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
191+
},
162192
"nodeIdentifierProperty": {
163193
"type": "string",
164194
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -186,6 +216,11 @@
186216
"type": "string",
187217
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
188218
},
219+
"nodeLabels": {
220+
"type": "array",
221+
"items": {"type": "string"},
222+
"description": "The node labels used to project and run Degree Centrality on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
223+
},
189224
"orientation": {
190225
"type": "string",
191226
"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) ",
@@ -217,6 +252,11 @@
217252
"items": {"type": "string"},
218253
"description": "List of node names to filter eigenvector centrality results for.",
219254
},
255+
"nodeLabels": {
256+
"type": "array",
257+
"items": {"type": "string"},
258+
"description": "The node labels used to project and run eigenvector centrality on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
259+
},
220260
"nodeIdentifierProperty": {
221261
"type": "string",
222262
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -274,6 +314,11 @@
274314
"items": {"type": "string"},
275315
"description": "List of node names to filter PageRank results for.",
276316
},
317+
"nodeLabels": {
318+
"type": "array",
319+
"items": {"type": "string"},
320+
"description": "The node labels used to project and run PageRank on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
321+
},
277322
"nodeIdentifierProperty": {
278323
"type": "string",
279324
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -327,6 +372,11 @@
327372
"items": {"type": "string"},
328373
"description": "List of node names to filter harmonic centrality results for.",
329374
},
375+
"nodeLabels": {
376+
"type": "array",
377+
"items": {"type": "string"},
378+
"description": "The node labels used to project and run harmonic centrality on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
379+
},
330380
"nodeIdentifierProperty": {
331381
"type": "string",
332382
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
@@ -352,6 +402,11 @@
352402
"type": "string",
353403
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
354404
},
405+
"nodeLabels": {
406+
"type": "array",
407+
"items": {"type": "string"},
408+
"description": "The node labels used to project and run hits on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
409+
},
355410
"hitsIterations": {
356411
"type": "integer",
357412
"description": "The number of hits iterations to run. The number of pregel iterations will be equal to hitsIterations * 4.",

‎mcp_server/src/mcp_server_neo4j_gds/community_algorithm_specs.py‎

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,11 @@
1717
"description": "The relationship property that holds the weight of the relationships. "
1818
"If not provided, all relationships are considered to have a weight of 1.",
1919
},
20+
"nodeLabels": {
21+
"type": "array",
22+
"items": {"type": "string"},
23+
"description": "The node labels used to project and run conductance on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
24+
},
2025
},
2126
"required": ["communityProperty"],
2227
},
@@ -54,6 +59,11 @@
5459
"type": "integer",
5560
"description": "The number of leaf nodes of the supporting tree data structure.",
5661
},
62+
"nodeLabels": {
63+
"type": "array",
64+
"items": {"type": "string"},
65+
"description": "The node labels used to project and run HDBSCAN on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
66+
},
5767
},
5868
"required": ["nodeProperty"],
5969
},
@@ -75,6 +85,11 @@
7585
"type": "string",
7686
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
7787
},
88+
"nodeLabels": {
89+
"type": "array",
90+
"items": {"type": "string"},
91+
"description": "The node labels used to project and run K Core decomposition on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
92+
},
7893
},
7994
"required": [],
8095
},
@@ -97,6 +112,11 @@
97112
"type": "string",
98113
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
99114
},
115+
"nodeLabels": {
116+
"type": "array",
117+
"items": {"type": "string"},
118+
"description": "The node labels used to project and run K-1 Coloring on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
119+
},
100120
"maxIterations": {
101121
"type": "integer",
102122
"description": "The maximum number of iterations to run the coloring algorithm.",
@@ -132,6 +152,11 @@
132152
"type": "string",
133153
"description": "Property name to use for identifying nodes (e.g., 'name', 'Name', 'title'). Use get_node_properties_keys to find available properties.",
134154
},
155+
"nodeLabels": {
156+
"type": "array",
157+
"items": {"type": "string"},
158+
"description": "The node labels used to project and run K-Means on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
159+
},
135160
"k": {
136161
"type": "integer",
137162
"description": "The number of clusters to create.",
@@ -202,6 +227,11 @@
202227
"type": "string",
203228
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
204229
},
230+
"nodeLabels": {
231+
"type": "array",
232+
"items": {"type": "string"},
233+
"description": "The node labels used to project and run Label Propagation on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
234+
},
205235
},
206236
},
207237
),
@@ -248,6 +278,11 @@
248278
"type": "string",
249279
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
250280
},
281+
"nodeLabels": {
282+
"type": "array",
283+
"items": {"type": "string"},
284+
"description": "The node labels used to project and run Leiden on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
285+
},
251286
},
252287
},
253288
),
@@ -277,6 +312,11 @@
277312
"items": {"type": "string"},
278313
"description": "Optional list of node names to filter results. Only nodes whose names (based on nodeIdentifierProperty) contain any of these values will be included in the results. Requires nodeIdentifierProperty to be specified.",
279314
},
315+
"nodeLabels": {
316+
"type": "array",
317+
"items": {"type": "string"},
318+
"description": "The node labels used to project and run Local Clustering Coefficient on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
319+
},
280320
},
281321
},
282322
),
@@ -325,6 +365,11 @@
325365
"type": "string",
326366
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
327367
},
368+
"nodeLabels": {
369+
"type": "array",
370+
"items": {"type": "string"},
371+
"description": "The node labels used to project and run Louvain on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
372+
},
328373
},
329374
},
330375
),
@@ -344,6 +389,11 @@
344389
"type": "string",
345390
"description": "Name of the relationship property to use as weights. If unspecified, the algorithm runs unweighted.",
346391
},
392+
"nodeLabels": {
393+
"type": "array",
394+
"items": {"type": "string"},
395+
"description": "The node labels used to project and run Modularity on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
396+
},
347397
},
348398
"required": ["communityProperty"],
349399
},
@@ -385,6 +435,11 @@
385435
"type": "string",
386436
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
387437
},
438+
"nodeLabels": {
439+
"type": "array",
440+
"items": {"type": "string"},
441+
"description": "The node labels used to project and run modularity optimization on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
442+
},
388443
},
389444
},
390445
),
@@ -404,6 +459,11 @@
404459
"type": "string",
405460
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
406461
},
462+
"nodeLabels": {
463+
"type": "array",
464+
"items": {"type": "string"},
465+
"description": "The node labels used to project and run SCC on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
466+
},
407467
},
408468
},
409469
),
@@ -432,6 +492,11 @@
432492
"items": {"type": "string"},
433493
"description": "Optional list of node names to filter results. Only nodes whose names (based on nodeIdentifierProperty) contain any of these values will be included in the results. Requires nodeIdentifierProperty to be specified.",
434494
},
495+
"nodeLabels": {
496+
"type": "array",
497+
"items": {"type": "string"},
498+
"description": "The node labels used to project and run Triangle Count on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
499+
},
435500
},
436501
},
437502
),
@@ -470,6 +535,11 @@
470535
"type": "string",
471536
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
472537
},
538+
"nodeLabels": {
539+
"type": "array",
540+
"items": {"type": "string"},
541+
"description": "The node labels used to project and run WCC on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
542+
},
473543
},
474544
},
475545
),
@@ -510,6 +580,11 @@
510580
"type": "string",
511581
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
512582
},
583+
"nodeLabels": {
584+
"type": "array",
585+
"items": {"type": "string"},
586+
"description": "The node labels used to project and run Max-K-Cut on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
587+
},
513588
},
514589
},
515590
),
@@ -539,6 +614,11 @@
539614
"type": "string",
540615
"description": "The name of a node property to use as node identifier in the result. If provided, the result will include a 'nodeName' column with values from this property.",
541616
},
617+
"nodeLabels": {
618+
"type": "array",
619+
"items": {"type": "string"},
620+
"description": "The node labels used to project and run SLLPA on. Nodes with different node labels will be ignored. Do not specify to run for all nodes",
621+
},
542622
},
543623
},
544624
),

0 commit comments

Comments
 (0)