add pagination to list nodes by type

gradio_mcp_space.py

@@ -982,7 +982,7 @@ def entity_relationships(node_id: str) -> str:
 
 
 @observe(as_type="tool")
-def search_by_type_and_name(node_type: str, name_query: str, limit: int = 10, fuzzy: bool = True) -> str:
+def search_by_type_and_name(node_type: str, name_query: str, limit: int = 10, page: int = 1, fuzzy: bool = True) -> str:
     """
     Search for nodes/entities by type and name substring with fuzzy matching support.
 
@@ -1005,19 +1005,27 @@ def search_by_type_and_name(node_type: str, name_query: str, limit: int = 10, fu
         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}'"
+
         # Convert fuzzy to bool if it's a string
         if isinstance(fuzzy, str):
             fuzzy = fuzzy.lower() in ('true', '1', 'yes')
-        
+
         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
         matches = []
@@ -1084,21 +1092,33 @@ def search_by_type_and_name(node_type: str, name_query: str, limit: int = 10, fu
                     "score": score
                 })
         
-        # Sort by match score (best matches first) and limit results
+        # Sort by match score (best matches first)
         matches.sort(key=lambda x: (x['score'], x['name'].lower()))
-        matches = matches[:limit]
 
-        if not matches:
+        total = len(matches)
+        if total == 0:
             return f"No matches for type '{node_type}' and name containing '{name_query}'."
 
-        result = f"Matches for type '{node_type}' and name '{name_query}' ({len(matches)} 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} results at {limit} per page)"
+
+        start_idx = (page - 1) * limit
+        end_idx = start_idx + limit
+        page_slice = matches[start_idx:end_idx]
+
+        result = f"Matches for type '{node_type}' and name '{name_query}' (Page {page}/{total_pages}, {total} total):\n"
         result += "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
 
-        for i, match in enumerate(matches, 1):
+        for i, match in enumerate(page_slice, start=start_idx + 1):
             result += f"{i}. {match['name']}\n"
             result += f"   ID: {match['id']}\n"
             result += f"   Type: {match['type']}\n\n"
 
+        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)}"
@@ -1910,10 +1930,13 @@ def create_gradio_app():
                         label="Node Type"
                     )
                     search_name = gr.Textbox(label="Name Contains", placeholder="Enter partial name...")
+                    search_limit = gr.Slider(1, 100, value=10, step=1, label="Max Results")
+                    search_page = gr.Slider(1, 100, value=1, step=1, label="Page")
+                    search_fuzzy = gr.Checkbox(label="Fuzzy Match", value=True)
                     search_type_btn = gr.Button("Search", variant="primary")
                 with gr.Column():
                     search_type_output = gr.Textbox(label="Results", lines=20, max_lines=30)
-            search_type_btn.click(fn=search_by_type_and_name, inputs=[search_type, search_name], outputs=search_type_output)
+            search_type_btn.click(fn=search_by_type_and_name, inputs=[search_type, search_name, search_limit, search_page, search_fuzzy], outputs=search_type_output)
             gr.Markdown(_tool_doc_md(search_by_type_and_name))
 
         with gr.Tab("🔗 Relationships"):