mcp-code-indexer 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

mcp_code_indexer/__init__.py

@@ -6,7 +6,7 @@ intelligent codebase navigation through searchable file descriptions,
 token-aware overviews, and advanced merge capabilities.
 """
 
-__version__ = "1.0.2"
+__version__ = "1.0.4"
 __author__ = "MCP Code Indexer Contributors"
 __email__ = ""
 __license__ = "MIT"

mcp_code_indexer/main.py

@@ -124,7 +124,9 @@ def cli_main():
         pass
     except Exception as e:
         # Log critical errors to stderr, not stdout
+        import traceback
         print(f"Server failed to start: {e}", file=sys.stderr)
+        print(f"Traceback: {traceback.format_exc()}", file=sys.stderr)
         sys.exit(1)
 
 

mcp_code_indexer/server/mcp_server.py

@@ -16,6 +16,7 @@ from typing import Any, Dict, List, Optional
 from mcp import types
 from mcp.server import Server
 from mcp.server.stdio import stdio_server
+from pydantic import ValidationError
 
 from mcp_code_indexer.database.database import DatabaseManager
 from mcp_code_indexer.file_scanner import FileScanner
@@ -75,6 +76,9 @@ class MCPCodeIndexServer:
         # Register handlers
         self._register_handlers()
         
+        # Add debug logging for server events
+        self.logger.debug("MCP server instance created and handlers registered")
+        
         self.logger.info(
             "MCP Code Index Server initialized", 
             extra={"structured_data": {"initialization": {"token_limit": token_limit}}}
@@ -648,21 +652,146 @@ class MCPCodeIndexServer:
                 **result
             }
     
-    async def run(self) -> None:
-        """Run the MCP server."""
-        await self.initialize()
-        
-        try:
-            async with stdio_server() as (read_stream, write_stream):
-                initialization_options = self.server.create_initialization_options()
+    async def _run_session_with_retry(self, read_stream, write_stream, initialization_options) -> None:
+        """Run a single MCP session with error handling and retry logic."""
+        max_retries = 3
+        base_delay = 1.0  # seconds
+        
+        for attempt in range(max_retries + 1):
+            try:
+                logger.info(f"Starting MCP server protocol session (attempt {attempt + 1})...")
                 await self.server.run(
                     read_stream,
                     write_stream, 
                     initialization_options
                 )
-        finally:
-            # Clean shutdown
-            await self.shutdown()
+                logger.info("MCP server session completed normally")
+                return  # Success, exit retry loop
+                
+            except ValidationError as e:
+                # Handle malformed requests gracefully
+                logger.warning(f"Received malformed request (attempt {attempt + 1}): {e}", extra={
+                    "structured_data": {
+                        "error_type": "ValidationError",
+                        "validation_errors": e.errors() if hasattr(e, 'errors') else str(e),
+                        "attempt": attempt + 1,
+                        "max_retries": max_retries
+                    }
+                })
+                
+                if attempt < max_retries:
+                    delay = base_delay * (2 ** attempt)  # Exponential backoff
+                    logger.info(f"Retrying in {delay} seconds...")
+                    await asyncio.sleep(delay)
+                else:
+                    logger.error("Max retries exceeded for validation errors. Server will continue but this session failed.")
+                    return
+                    
+            except (ConnectionError, BrokenPipeError, EOFError) as e:
+                # Handle client disconnection gracefully
+                logger.info(f"Client disconnected: {e}")
+                return
+                
+            except Exception as e:
+                # Handle other exceptions with full logging
+                import traceback
+                if "unhandled errors in a TaskGroup" in str(e) and "ValidationError" in str(e):
+                    # This is likely a ValidationError wrapped in a TaskGroup exception
+                    logger.warning(f"Detected wrapped validation error (attempt {attempt + 1}): {e}", extra={
+                        "structured_data": {
+                            "error_type": type(e).__name__, 
+                            "error_message": str(e),
+                            "attempt": attempt + 1,
+                            "max_retries": max_retries,
+                            "likely_validation_error": True
+                        }
+                    })
+                    
+                    if attempt < max_retries:
+                        delay = base_delay * (2 ** attempt)
+                        logger.info(f"Retrying in {delay} seconds...")
+                        await asyncio.sleep(delay)
+                    else:
+                        logger.error("Max retries exceeded for validation errors. Server will continue but this session failed.")
+                        return
+                else:
+                    # This is a genuine error, log and re-raise
+                    logger.error(f"MCP server session error: {e}", extra={
+                        "structured_data": {
+                            "error_type": type(e).__name__, 
+                            "error_message": str(e),
+                            "traceback": traceback.format_exc()
+                        }
+                    })
+                    raise
+
+    async def run(self) -> None:
+        """Run the MCP server with robust error handling."""
+        logger.info("Starting server initialization...")
+        await self.initialize()
+        logger.info("Server initialization completed, starting MCP protocol...")
+        
+        max_retries = 5
+        base_delay = 2.0  # seconds
+        
+        for attempt in range(max_retries + 1):
+            try:
+                async with stdio_server() as (read_stream, write_stream):
+                    logger.info(f"stdio_server context established (attempt {attempt + 1})")
+                    initialization_options = self.server.create_initialization_options()
+                    logger.debug(f"Initialization options: {initialization_options}")
+                    
+                    await self._run_session_with_retry(read_stream, write_stream, initialization_options)
+                    return  # Success, exit retry loop
+                        
+            except KeyboardInterrupt:
+                logger.info("Server stopped by user interrupt")
+                return
+                
+            except Exception as e:
+                import traceback
+                
+                # Check if this is a wrapped validation error
+                error_str = str(e)
+                is_validation_error = (
+                    "ValidationError" in error_str or 
+                    "Field required" in error_str or 
+                    "Input should be" in error_str or
+                    "pydantic_core._pydantic_core.ValidationError" in error_str
+                )
+                
+                if is_validation_error:
+                    logger.warning(f"Detected validation error in session (attempt {attempt + 1}): Malformed client request", extra={
+                        "structured_data": {
+                            "error_type": "ValidationError", 
+                            "error_message": "Client sent malformed request (likely missing clientInfo)",
+                            "attempt": attempt + 1,
+                            "max_retries": max_retries,
+                            "will_retry": attempt < max_retries
+                        }
+                    })
+                    
+                    if attempt < max_retries:
+                        delay = base_delay * (2 ** min(attempt, 3))  # Cap exponential growth
+                        logger.info(f"Retrying server in {delay} seconds...")
+                        await asyncio.sleep(delay)
+                        continue
+                    else:
+                        logger.warning("Max retries exceeded for validation errors. Server is robust against malformed requests.")
+                        return
+                else:
+                    # This is a genuine fatal error
+                    logger.error(f"Fatal server error: {e}", extra={
+                        "structured_data": {
+                            "error_type": type(e).__name__, 
+                            "error_message": str(e),
+                            "traceback": traceback.format_exc()
+                        }
+                    })
+                    raise
+        
+        # Clean shutdown
+        await self.shutdown()
     
     async def shutdown(self) -> None:
         """Clean shutdown of server resources."""

{mcp_code_indexer-1.0.2.dist-info → mcp_code_indexer-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-code-indexer
-Version: 1.0.2
+Version: 1.0.4
 Summary: MCP server that tracks file descriptions across codebases, enabling AI agents to efficiently navigate and understand code through searchable summaries and token-aware overviews.
 Author: MCP Code Indexer Contributors
 Maintainer: MCP Code Indexer Contributors
@@ -30,7 +30,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: tiktoken>=0.9.0
-Requires-Dist: mcp==1.0.0
+Requires-Dist: mcp>=1.9.0
 Requires-Dist: gitignore_parser==0.1.11
 Requires-Dist: pydantic>=2.8.0
 Requires-Dist: aiofiles==23.2.0

{mcp_code_indexer-1.0.2.dist-info → mcp_code_indexer-1.0.4.dist-info}/RECORD

@@ -1,8 +1,8 @@
-mcp_code_indexer/__init__.py,sha256=SqtCXmJGZ4zdPgDrCETFKwiyVSEH6W4EOApSFwdTAY0,473
+mcp_code_indexer/__init__.py,sha256=tFNpOTeNAG23YobUnL1oXAdr6CfSQXPtzvGswDz8uPg,473
 mcp_code_indexer/error_handler.py,sha256=cNSUFFrGBMLDv4qa78c7495L1wSl_dXCRbzCJOidx-Q,11590
 mcp_code_indexer/file_scanner.py,sha256=1Z6wq7H14V1OMAHIF4v9G7SY8hC1puDmU5IXsCKH4kU,11442
 mcp_code_indexer/logging_config.py,sha256=5L1cYIG8IAX91yCjc5pzkbO_KPt0bvm_ABHB53LBZjI,5184
-mcp_code_indexer/main.py,sha256=TJHyXc0mchaIyF5ABo24kcT8Sv1PsJibHVZoznGbrxo,3619
+mcp_code_indexer/main.py,sha256=Rou-mAN9-12PPP8jC7dIs2_UNambJuC2F8BF--j-0m8,3715
 mcp_code_indexer/merge_handler.py,sha256=lJR8eVq2qSrF6MW9mR3Fy8UzrNAaQ7RsI2FMNXne3vQ,14692
 mcp_code_indexer/token_counter.py,sha256=WrifOkbF99nWWHlRlhCHAB2KN7qr83GOHl7apE-hJcE,8460
 mcp_code_indexer/database/__init__.py,sha256=aPq_aaRp0aSwOBIq9GkuMNjmLxA411zg2vhdrAuHm-w,38
@@ -11,12 +11,12 @@ mcp_code_indexer/database/models.py,sha256=3wOxHKb6j3zKPWFSwB5g1TLpI507vLNZcqsxZ
 mcp_code_indexer/middleware/__init__.py,sha256=p-mP0pMsfiU2yajCPvokCUxUEkh_lu4XJP1LyyMW2ug,220
 mcp_code_indexer/middleware/error_middleware.py,sha256=v6jaHmPxf3qerYdb85X1tHIXLxgcbybpitKVakFLQTA,10109
 mcp_code_indexer/server/__init__.py,sha256=16xMcuriUOBlawRqWNBk6niwrvtv_JD5xvI36X1Vsmk,41
-mcp_code_indexer/server/mcp_server.py,sha256=6wm4XKfTLSpfHP9GpCqwWsekQFwwo9ngqM-ZXiJidx4,33424
+mcp_code_indexer/server/mcp_server.py,sha256=Ok4nrwkRNLGmHTEeW-Ij00gVMMRtS5aX-Hi-lUot5bg,39900
 mcp_code_indexer/tiktoken_cache/9b5ad71b2ce5302211f9c61530b329a4922fc6a4,sha256=Ijkht27pm96ZW3_3OFE-7xAPtR0YyTWXoRO8_-hlsqc,1681126
 mcp_code_indexer/tools/__init__.py,sha256=m01mxML2UdD7y5rih_XNhNSCMzQTz7WQ_T1TeOcYlnE,49
-mcp_code_indexer-1.0.2.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
-mcp_code_indexer-1.0.2.dist-info/METADATA,sha256=u-yHVXuTl_y5Qkrb4vWMBkQWmuS6OUB2RtrzwvsnimU,11930
-mcp_code_indexer-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mcp_code_indexer-1.0.2.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
-mcp_code_indexer-1.0.2.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
-mcp_code_indexer-1.0.2.dist-info/RECORD,,
+mcp_code_indexer-1.0.4.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
+mcp_code_indexer-1.0.4.dist-info/METADATA,sha256=ZOl-psh9Y4q9OtUFzuevW53lpm0SsczC7Nr3aSzIso4,11930
+mcp_code_indexer-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcp_code_indexer-1.0.4.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
+mcp_code_indexer-1.0.4.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
+mcp_code_indexer-1.0.4.dist-info/RECORD,,