refactor: update find_files method to return a list of file paths instead of a dictionary

Author: johnhuang316

src/code_index_mcp/server.py

@@ -13,7 +13,7 @@
 import logging
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
-from typing import AsyncIterator, Dict, Any, Optional
+from typing import AsyncIterator, Dict, Any, Optional, List
 
 # Third-party imports
 from mcp import types
@@ -185,8 +185,8 @@ def search_code_advanced(
     )
 
 @mcp.tool()
-@handle_mcp_tool_errors(return_type='dict')
-def find_files(pattern: str, ctx: Context) -> Dict[str, Any]:
+@handle_mcp_tool_errors(return_type='list')
+def find_files(pattern: str, ctx: Context) -> List[str]:
     """
     Find files matching a glob pattern using pre-built file index.
 
@@ -206,7 +206,7 @@ def find_files(pattern: str, ctx: Context) -> Dict[str, Any]:
         pattern: Glob pattern to match files (e.g., "*.py", "test_*.js", "README.md")
 
     Returns:
-        Dictionary with files list and status information
+        List of file paths matching the pattern
     """
     return FileDiscoveryService(ctx).find_files(pattern)
 

src/code_index_mcp/services/file_discovery_service.py

@@ -37,7 +37,7 @@ def __init__(self, ctx):
         super().__init__(ctx)
         self._matcher_tool = FileMatchingTool()
 
-    def find_files(self, pattern: str, max_results: Optional[int] = None) -> Dict[str, Any]:
+    def find_files(self, pattern: str, max_results: Optional[int] = None) -> List[str]:
         """
         Find files matching the given pattern using intelligent discovery.
 
@@ -50,7 +50,7 @@ def find_files(self, pattern: str, max_results: Optional[int] = None) -> Dict[st
             max_results: Maximum number of results to return (None for no limit)
 
         Returns:
-            Dictionary with discovery results and metadata
+            List of file paths matching the pattern
 
         Raises:
             ValueError: If pattern is invalid or project not set up
@@ -282,21 +282,14 @@ def _gather_discovery_metadata(self, all_files, matched_files, limited_files, pa
             'original_pattern': pattern
         }
 
-    def _format_discovery_result(self, discovery_result: FileDiscoveryResult) -> Dict[str, Any]:
+    def _format_discovery_result(self, discovery_result: FileDiscoveryResult) -> List[str]:
         """
         Format the discovery result according to business requirements.
 
         Args:
             discovery_result: Raw discovery result
 
         Returns:
-            Formatted result dictionary for MCP response
+            Simple list of file paths
         """
-        return {
-            'files': discovery_result.files,
-            'total': discovery_result.total_count,
-            'pattern': discovery_result.pattern_used,
-            'strategy': discovery_result.search_strategy,
-            'metadata': discovery_result.metadata,
-            'status': 'success'
-        }
+        return discovery_result.files