add pagination to get_neighbours

gradio_mcp_space.py

@@ -415,7 +415,7 @@ def list_nodes_by_type(node_type: str, limit: int = 20, page: int = 1) -> str:
 
 
 @observe(as_type="tool")
-def get_neighbors(node_id: str) -> str:
+def get_neighbors(node_id: str, limit: int = 20, page: int = 1) -> str:
     """
     Get all nodes directly connected to a given node.
 
@@ -434,14 +434,42 @@ def get_neighbors(node_id: str) -> str:
         if node_id not in knowledge_graph.graph:
             return f"Error: Node '{node_id}' not found in knowledge graph"
 
+        # Convert limit/page to int if they're strings
+        if isinstance(limit, str):
+            try:
+                limit = int(limit)
+            except ValueError:
+                return f"Error: 'limit' must be an integer, got '{limit}'"
+
+        if isinstance(page, str):
+            try:
+                page = int(page)
+            except ValueError:
+                return f"Error: 'page' must be an integer, got '{page}'"
+
+        if limit <= 0:
+            return "Error: limit must be a positive integer"
+        if page < 1:
+            return "Error: 'page' must be a positive integer (1 or greater)"
+
         neighbors = knowledge_graph.get_neighbors(node_id)
         if not neighbors:
             return f"No neighbors found for node '{node_id}'"
 
-        result = f"Neighbors of '{node_id}' ({len(neighbors)} total):\n"
+        total = len(neighbors)
+        # Pagination
+        total_pages = (total + limit - 1) // limit
+        if page > total_pages:
+            return f"Error: Page {page} does not exist. Total pages: {total_pages} (with {total} neighbors at {limit} per page)"
+
+        start_idx = (page - 1) * limit
+        end_idx = start_idx + limit
+        page_slice = neighbors[start_idx:end_idx]
+
+        result = f"Neighbors of '{node_id}' (Page {page}/{total_pages}, {total} total):\n"
         result += "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
 
-        for i, neighbor in enumerate(neighbors[:20], 1):
+        for i, neighbor in enumerate(page_slice, start=start_idx + 1):
             result += f"{i}. {neighbor.id}\n"
             result += f"   Name: {getattr(neighbor, 'name', 'Unknown')}\n"
             result += f"   Type: {neighbor.node_type}\n"
@@ -454,8 +482,9 @@ def get_neighbors(node_id: str) -> str:
                 result += f"   ← Relation: {edge_data.get('relation', 'Unknown')}\n"
             result += "\n"
 
-        if len(neighbors) > 20:
-            result += f"... and {len(neighbors) - 20} more neighbors\n"
+        # Pagination hint
+        if page < total_pages:
+            result += f"Use page={page + 1} to see the next page\n"
 
         return result
     except Exception as e:
@@ -1944,10 +1973,12 @@ def create_gradio_app():
             with gr.Row():
                 with gr.Column():
                     neighbor_node_id = gr.Textbox(label="Node ID", placeholder="Enter node ID...")
+                    neighbor_limit = gr.Slider(1, 100, value=20, step=1, label="Max Results")
+                    neighbor_page = gr.Slider(1, 100, value=1, step=1, label="Page")
                     neighbor_btn = gr.Button("Get Neighbors", variant="primary")
                 with gr.Column():
                     neighbor_output = gr.Textbox(label="Neighbors", lines=20, max_lines=30)
-            neighbor_btn.click(fn=get_neighbors, inputs=neighbor_node_id, outputs=neighbor_output)
+            neighbor_btn.click(fn=get_neighbors, inputs=[neighbor_node_id, neighbor_limit, neighbor_page], outputs=neighbor_output)
             gr.Markdown(_tool_doc_md(get_neighbors))
 
             gr.Markdown("---")