add all option to get_related chunks

gradio_mcp_space.py

@@ -670,14 +670,24 @@ def get_related_chunks(chunk_id: str, relation_type: str = "calls") -> str:
             return f"Error: Chunk '{chunk_id}' not found in knowledge graph"
 
         related = []
-        for _, target, attrs in knowledge_graph.graph.out_edges(chunk_id, data=True):
-            if attrs.get('relation') == relation_type:
+        if relation_type == "" or relation_type == "all":
+            # Get all outgoing edges regardless of relation type
+            for _, target, attrs in knowledge_graph.graph.out_edges(chunk_id, data=True):
                 target_node = knowledge_graph.graph.nodes[target]['data']
                 related.append({
                     "id": target,
                     "file_path": getattr(target_node, 'path', 'Unknown'),
                     "entity_name": attrs.get('entity_name')
                 })
+        else:     
+            for _, target, attrs in knowledge_graph.graph.out_edges(chunk_id, data=True):
+                if attrs.get('relation') == relation_type:
+                    target_node = knowledge_graph.graph.nodes[target]['data']
+                    related.append({
+                        "id": target,
+                        "file_path": getattr(target_node, 'path', 'Unknown'),
+                        "entity_name": attrs.get('entity_name')
+                    })
 
         if not related:
             return f"No chunks found with '{relation_type}' relationship from '{chunk_id}'"
@@ -1311,13 +1321,13 @@ def find_path(source_id: str, target_id: str, max_depth: int = 5) -> str:
 @observe(as_type="tool")
 def get_subgraph(node_id: str, depth: int = 2, edge_types: Optional[str] = None) -> str:
     """
-    Extract a subgraph around a node up to a specified depth.
+    Retrieve a subgraph around a node up to a specified depth.
 
     Optionally filters by edge types (comma-separated).
 
     Args:
         node_id: The ID of the central node
-        depth: The depth/radius of the subgraph to extract (default: 2)
+        depth: The depth/radius of the subgraph to Retrieve (default: 2)
         edge_types: Optional comma-separated list of edge types (e.g., 'calls,contains')
 
     Returns:
@@ -1997,7 +2007,7 @@ def create_gradio_app():
             with gr.Row():
                 with gr.Column():
                     related_chunk_id = gr.Textbox(label="Chunk ID", placeholder="Enter chunk ID...")
-                    relation_type = gr.Dropdown(choices=["calls", "contains", "declares", "uses"], label="Relation Type", value="calls")
+                    relation_type = gr.Dropdown(choices=["" , "calls", "contains", "declares", "uses"], label="Relation Type", value="calls")
                     related_btn = gr.Button("Get Related Chunks", variant="primary")
                 with gr.Column():
                     related_output = gr.Textbox(label="Related Chunks", lines=20, max_lines=30)
@@ -2057,7 +2067,7 @@ def create_gradio_app():
                     subgraph_node = gr.Textbox(label="Center Node ID", placeholder="Enter node ID...")
                     subgraph_depth = gr.Slider(1, 5, value=2, step=1, label="Depth")
                     subgraph_edge_types = gr.Textbox(label="Edge Types (comma-separated, optional)", placeholder="e.g., calls,contains")
-                    subgraph_btn = gr.Button("Extract Subgraph", variant="primary")
+                    subgraph_btn = gr.Button("Retrieve Subgraph", variant="primary")
                 with gr.Column():
                     subgraph_output = gr.Textbox(label="Subgraph", lines=20, max_lines=30)
             subgraph_btn.click(fn=get_subgraph, inputs=[subgraph_node, subgraph_depth, subgraph_edge_types], outputs=subgraph_output)