add filter used_in_repo in list_all_entities

gradio_mcp_space.py

@@ -247,9 +247,9 @@ Incoming Edges ({len(incoming)}):
 @observe(as_type="tool")
 def search_nodes(query: str, limit: int = 10) -> str:
     """
-    Search for nodes in the knowledge graph by query string.
+    Search for chunk nodes in the knowledge graph by query string.
 
-    Uses semantic and keyword search via the code index.
+    Uses keyword search via the code index.
 
     Args:
         query: The search string to match against code index
@@ -440,12 +440,12 @@ def get_neighbors(node_id: str) -> str:
 @observe(as_type="tool")
 def go_to_definition(entity_name: str) -> str:
     """
-    Find where an entity is declared or defined in the codebase.
+    Retrieve where an entity is declared or defined in the codebase.
 
-    Locates the declaration point for functions, classes, variables, etc.
+    Locates and retrieves the declaration point for functions, classes, variables, etc.
 
     Args:
-        entity_name: The name of the entity to find the definition for
+        entity_name: The name of the entity to retrieve the definition for
 
     Returns:
         str: A formatted string with definition locations
@@ -489,12 +489,12 @@ def go_to_definition(entity_name: str) -> str:
 @observe(as_type="tool")
 def find_usages(entity_name: str, limit: int = 20) -> str:
     """
-    Find all usages or calls of an entity in the codebase.
+    Retrieve all usages or calls of an entity in the codebase.
 
     Shows where functions, classes, variables, etc. are used.
 
     Args:
-        entity_name: The name of the entity to find usages for
+        entity_name: The name of the entity to retrieve usages for
         limit: Maximum number of usages to return (default: 20)
 
     Returns:
@@ -597,12 +597,12 @@ def get_file_structure(file_path: str) -> str:
 @observe(as_type="tool")
 def get_related_chunks(chunk_id: str, relation_type: str = "calls") -> str:
     """
-    Get chunks related to a given chunk by a specific relationship.
+    Retrieve chunks related to a given chunk by a specific relationship.
 
-    Find chunks connected via relationships like 'calls', 'contains', etc.
+    Retrieve chunks connected via relationships like 'calls', 'contains', etc.
 
     Args:
-        chunk_id: The ID of the chunk to find related chunks for
+        chunk_id: The ID of the chunk to retrieve related chunks for
         relation_type: The type of relationship to filter by (default: 'calls')
 
     Returns:
@@ -651,7 +651,8 @@ def list_all_entities(
     limit: int = 50,
     page: int = 1,
     entity_type: Optional[str] = None,
-    declared_in_repo: Optional[bool] = None
+    declared_in_repo: Optional[bool] = None,
+    called_in_repo: Optional[bool] = None
 ) -> str:
     """
     List all entities tracked in the knowledge graph with filtering and pagination options.
@@ -663,6 +664,7 @@ def list_all_entities(
         page: Page number for pagination, 1-indexed (default: 1)
         entity_type: Filter by entity type ('class', 'function', 'method', 'variable', 'parameter', 'function_call', 'method_call')
         declared_in_repo: If True, only return entities with declarations. If False, only entities without declarations. If None, return all.
+        called_in_repo: If True, only return entities that have usages/calls in the repo. If False, only entities without usages. If None, return all.
 
     Returns:
         str: A formatted string with all entities for the requested page
@@ -700,6 +702,15 @@ def list_all_entities(
                 declared_in_repo = False
             elif declared_in_repo.lower() in ("none", "null", "all", ""):
                 declared_in_repo = None
+
+        # Handle called_in_repo - convert string to bool if needed
+        if isinstance(called_in_repo, str):
+            if called_in_repo.lower() in ("true", "1", "yes"):
+                called_in_repo = True
+            elif called_in_repo.lower() in ("false", "0", "no"):
+                called_in_repo = False
+            elif called_in_repo.lower() in ("none", "null", "all", ""):
+                called_in_repo = None
         
         if not knowledge_graph.entities:
             return "No entities found in the knowledge graph."
@@ -721,6 +732,14 @@ def list_all_entities(
                 if not declared_in_repo and has_declaration:
                     continue
 
+            # Filter by called_in_repo (usages) if specified
+            if called_in_repo is not None:
+                has_calls = len(info.get('calling_chunk_ids', [])) > 0
+                if called_in_repo and not has_calls:
+                    continue
+                if not called_in_repo and has_calls:
+                    continue
+
             filtered_entities[entity_name] = info
 
         # Build the response with filtered entities
@@ -730,6 +749,8 @@ def list_all_entities(
                 filter_desc.append(f"type={entity_type}")
             if declared_in_repo is not None:
                 filter_desc.append(f"declared_in_repo={declared_in_repo}")
+            if called_in_repo is not None:
+                filter_desc.append(f"called_in_repo={called_in_repo}")
             filter_text = f" (filtered by {', '.join(filter_desc)})" if filter_desc else ""
             return f"No entities found{filter_text}."
 
@@ -769,6 +790,8 @@ def list_all_entities(
             result += f"\n(Filtered by type={entity_type})\n"
         if declared_in_repo is not None:
             result += f"(Filtered by declared_in_repo={declared_in_repo})\n"
+        if called_in_repo is not None:
+            result += f"(Filtered by called_in_repo={called_in_repo})\n"
 
         return result
     except Exception as e:
@@ -1758,8 +1781,8 @@ def create_gradio_app():
                     node_output = gr.Textbox(label="Node Information", lines=20, max_lines=30)
             node_info_btn.click(fn=get_node_info, inputs=node_id_input, outputs=node_output)
             node_edges_btn.click(fn=get_node_edges, inputs=node_id_input, outputs=node_output)
-            gr.Markdown(_tool_doc_md(get_node_info))
-            gr.Markdown(_tool_doc_md(get_node_edges))
+            gr.Markdown("#Get Node Info:" + _tool_doc_md(get_node_info))
+            gr.Markdown("#Get Node Edges:" + _tool_doc_md(get_node_edges))
 
         with gr.Tab("🏗️ Structure"):
             gr.Markdown("### Repository Tree")
@@ -1799,12 +1822,17 @@ def create_gradio_app():
                         label="Declared in Repo (optional)",
                         value=""
                     )
+                    called_in_repo = gr.Dropdown(
+                        choices=["", "true", "false"],
+                        label="Called in Repo (optional)",
+                        value=""
+                    )
                     list_entities_btn = gr.Button("List Entities", variant="primary")
                 with gr.Column():
                     list_entities_output = gr.Textbox(label="Entities", lines=20, max_lines=30)
             list_entities_btn.click(
                 fn=list_all_entities,
-                inputs=[entity_limit, entity_page, entity_type_filter, declared_in_repo],
+                inputs=[entity_limit, entity_page, entity_type_filter, declared_in_repo, called_in_repo],
                 outputs=list_entities_output,
             )
             gr.Markdown(_tool_doc_md(list_all_entities))