add pagination to list_nodes_by_type

gradio_mcp_space.py

@@ -342,7 +342,7 @@ Node Types:
 
 
 @observe(as_type="tool")
-def list_nodes_by_type(node_type: str, limit: int = 20) -> str:
+def list_nodes_by_type(node_type: str, limit: int = 20, page: int = 1) -> str:
     """
     List nodes of a specific type in the knowledge graph.
 
@@ -357,13 +357,24 @@ def list_nodes_by_type(node_type: str, limit: int = 20) -> str:
         return "Error: Knowledge graph not initialized"
 
     try:
-        # Convert limit to int if it's a string (MCP may pass strings)
+        # Convert limit/page to int if they're strings (MCP/Gradio may pass 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)"
+
         g = knowledge_graph.graph
         matching_nodes = [
             {
@@ -372,18 +383,32 @@ def list_nodes_by_type(node_type: str, limit: int = 20) -> str:
             }
             for node_id, data in g.nodes(data=True)
             if getattr(data['data'], 'node_type', None) == node_type
-        ][:limit]
+        ]
 
-        if not matching_nodes:
+        total = len(matching_nodes)
+        if total == 0:
             return f"No nodes found of type '{node_type}'."
 
-        result = f"Nodes of type '{node_type}' ({len(matching_nodes)} results):\n"
+        # 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} nodes at {limit} per page)"
+
+        start_idx = (page - 1) * limit
+        end_idx = start_idx + limit
+        page_slice = matching_nodes[start_idx:end_idx]
+
+        result = f"Nodes of type '{node_type}' (Page {page}/{total_pages}, {total} total):\n"
         result += "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
 
-        for i, node in enumerate(matching_nodes, 1):
+        for i, node in enumerate(page_slice, start=start_idx + 1):
             result += f"{i}. {node['name']}\n"
             result += f"   ID: {node['id']}\n\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:
         return f"Error: {str(e)}"
@@ -440,7 +465,7 @@ def get_neighbors(node_id: str) -> str:
 @observe(as_type="tool")
 def go_to_definition(entity_name: str) -> str:
     """
-    Retrieve where an entity is declared or defined in the codebase.
+    Retrieve chunk node(s) where entity is declared or defined in the codebase.
 
     Locates and retrieves the declaration point for functions, classes, variables, etc.
 
@@ -1869,10 +1894,11 @@ def create_gradio_app():
                         label="Node Type"
                     )
                     type_limit = gr.Slider(1, 100, value=20, step=1, label="Max Results")
+                    type_page = gr.Slider(1, 100, value=1, step=1, label="Page")
                     type_btn = gr.Button("List Nodes", variant="primary")
                 with gr.Column():
                     type_output = gr.Textbox(label="Results", lines=20, max_lines=30)
-            type_btn.click(fn=list_nodes_by_type, inputs=[node_type_input, type_limit], outputs=type_output)
+            type_btn.click(fn=list_nodes_by_type, inputs=[node_type_input, type_limit, type_page], outputs=type_output)
             gr.Markdown(_tool_doc_md(list_nodes_by_type))
 
             gr.Markdown("---")