simplify get rel types

Author: brs96

mcp_server/src/mcp_server_neo4j_gds/gds.py

@@ -204,9 +204,8 @@ def get_relationship_properties_keys(gds: GraphDataScience, relationshipTypes=No
     return df["properties_keys"].iloc[0]
 
 
-def get_relationship_types(gds: GraphDataScience, node_labels=None):
-    if node_labels is None:
-        node_labels = []
+def get_relationship_types(gds: GraphDataScience):
+    node_labels = []
     type_extractor = """
                      WITH type(r) AS type
                      WITH DISTINCT type

mcp_server/src/mcp_server_neo4j_gds/server.py

@@ -108,14 +108,6 @@ async def handle_list_tools() -> list[types.Tool]:
                         description="""Get relationship types in the database.""",
                         inputSchema={
                             "type": "object",
-                            "properties": {
-                                "nodeLabels": {
-                                    "type": "array",
-                                    "items": {"type": "string"},
-                                    "description": "Ignore relationships whose source and target node is not in the specified node labels. To use this parameter, use get_node_labels first.",
-                                },
-                            },
-                            "required": [],
                         },
                     ),
                 ]
@@ -151,10 +143,7 @@ async def handle_call_tool(
                 result = get_node_labels(gds)
                 return [types.TextContent(type="text", text=serialize_result(result))]
             elif name == "get_relationship_types":
-                if "nodeLabels" in arguments:
-                    result = get_relationship_types(gds, arguments["nodeLabels"])
-                else:
-                    result = get_relationship_types(gds)
+                result = get_relationship_types(gds)
                 return [types.TextContent(type="text", text=serialize_result(result))]
 
             else:

mcp_server/tests/test_basic_tools.py

@@ -125,3 +125,14 @@ async def test_get_node_labels(mcp_client):
     properties_keys = json.loads(result_text)
 
     assert properties_keys == ["UndergroundStation"]
+
+
+@pytest.mark.asyncio
+async def test_get_relationship_types(mcp_client):
+    result = await mcp_client.call_tool("get_relationship_types")
+
+    assert len(result) == 1
+    result_text = result[0]["text"]
+    properties_keys = json.loads(result_text)
+
+    assert properties_keys == ["LINK"]