mcp-code-indexer 4.2.7__tar.gz → 4.2.9__tar.gz

{mcp_code_indexer-4.2.7 → mcp_code_indexer-4.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mcp-code-indexer
-Version: 4.2.7
+Version: 4.2.9
 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.
 License: MIT
 Keywords: mcp,model-context-protocol,code-indexer,ai-tools,codebase-navigation,file-descriptions,llm-tools
@@ -48,8 +48,8 @@ Description-Content-Type: text/markdown
 
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?54)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?54)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?56)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?56)](https://pypi.org/project/mcp-code-indexer/)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
 
 A production-ready **Model Context Protocol (MCP) server** that revolutionizes how AI agents navigate and understand codebases. Built for high-concurrency environments with advanced database resilience, the server provides instant access to intelligent descriptions, semantic search, and context-aware recommendations while maintaining 800+ writes/sec throughput.

{mcp_code_indexer-4.2.7 → mcp_code_indexer-4.2.9}/README.md

@@ -1,7 +1,7 @@
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?54)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?54)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?56)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?56)](https://pypi.org/project/mcp-code-indexer/)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
 
 A production-ready **Model Context Protocol (MCP) server** that revolutionizes how AI agents navigate and understand codebases. Built for high-concurrency environments with advanced database resilience, the server provides instant access to intelligent descriptions, semantic search, and context-aware recommendations while maintaining 800+ writes/sec throughput.

{mcp_code_indexer-4.2.7 → mcp_code_indexer-4.2.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "mcp-code-indexer"
-version = "4.2.7"
+version = "4.2.9"
 description = "MCP server that tracks file descriptions across codebases, enabling AI agents to efficiently navigate and understand code through searchable summaries and token-aware overviews."
 authors = ["MCP Code Indexer Contributors"]
 maintainers = ["MCP Code Indexer Contributors"]

{mcp_code_indexer-4.2.7 → mcp_code_indexer-4.2.9}/src/mcp_code_indexer/main.py

@@ -8,6 +8,7 @@ Entry point for the mcp-code-indexer package when installed via pip.
 import argparse
 import asyncio
 import json
+import signal
 import sys
 from pathlib import Path
 from typing import Any, Dict, List, Optional
@@ -1094,7 +1095,40 @@ async def main() -> None:
         if transport:
             transport.server = server
 
-        await server.run()
+        # Setup signal handling for graceful shutdown
+        shutdown_event = asyncio.Event()
+        
+        def signal_handler(signum):
+            logger.info(f"Received shutdown signal {signum}")
+            shutdown_event.set()
+        
+        # Register signal handlers
+        loop = asyncio.get_running_loop()
+        for sig in [signal.SIGTERM, signal.SIGINT]:
+            loop.add_signal_handler(sig, signal_handler, sig)
+        
+        # Run server and wait for shutdown signal
+        server_task = asyncio.create_task(server.run())
+        shutdown_task = asyncio.create_task(shutdown_event.wait())
+        
+        try:
+            # Wait for either server completion or shutdown signal
+            done, pending = await asyncio.wait(
+                [server_task, shutdown_task],
+                return_when=asyncio.FIRST_COMPLETED
+            )
+            
+            # Cancel remaining tasks
+            for task in pending:
+                task.cancel()
+                try:
+                    await task
+                except asyncio.CancelledError:
+                    pass
+                    
+        except Exception as e:
+            error_handler.log_error(e, context={"phase": "runtime"})
+            raise
 
     except Exception as e:
         error_handler.log_error(e, context={"phase": "startup"})
@@ -1104,10 +1138,13 @@ async def main() -> None:
         if vector_daemon_task and not vector_daemon_task.done():
             logger.info("Cancelling vector daemon")
             vector_daemon_task.cancel()
+        
+        # Wait for vector daemon to finish
+        if vector_daemon_task:
             try:
                 await vector_daemon_task
             except asyncio.CancelledError:
-                pass
+                logger.info("Vector daemon cancelled successfully")
 
 
 def cli_main() -> None:

{mcp_code_indexer-4.2.7 → mcp_code_indexer-4.2.9}/src/mcp_code_indexer/vector_mode/daemon.py

@@ -38,8 +38,6 @@ class VectorDaemon:
         self.db_manager = db_manager
         self.cache_dir = cache_dir
         self.is_running = False
-        self.shutdown_requested = False
-        self.shutdown_event = asyncio.Event()
         
         # Process tracking
         self.monitored_projects: Set[str] = set()
@@ -56,24 +54,7 @@ class VectorDaemon:
             "last_activity": time.time(),
         }
         
-        # Setup signal handlers will be done in start() method to access event loop
-        
-    def _setup_signal_handlers(self) -> None:
-        """Setup signal handlers for graceful shutdown."""
-        try:
-            loop = asyncio.get_running_loop()
-            for sig in [signal.SIGTERM, signal.SIGINT]:
-                loop.add_signal_handler(sig, self._signal_handler, sig)
-            if hasattr(signal, 'SIGHUP'):
-                loop.add_signal_handler(signal.SIGHUP, self._signal_handler, signal.SIGHUP)
-        except Exception as e:
-            logger.warning(f"Could not setup signal handlers: {e}")
-    
-    def _signal_handler(self, signum: int) -> None:
-        """Handle shutdown signals."""
-        logger.info(f"Received signal {signum}, initiating graceful shutdown")
-        self.shutdown_requested = True
-        self.shutdown_event.set()
+        # Signal handling is delegated to the parent process
     
     async def start(self) -> None:
         """Start the vector daemon."""
@@ -83,9 +64,6 @@ class VectorDaemon:
         
         self.is_running = True
         
-        # Setup signal handlers now that we have an event loop
-        self._setup_signal_handlers()
-        
         logger.info(
             "Starting vector daemon",
             extra={
@@ -119,19 +97,21 @@ class VectorDaemon:
         finally:
             await self._cleanup()
     
-    async def _notify_shutdown(self) -> None:
-        """Notify shutdown event (called from signal handler)."""
-        self.shutdown_event.set()
-    
     async def _run_until_shutdown(self) -> None:
         """Run daemon until shutdown is requested."""
-        await self.shutdown_event.wait()
+        # Wait indefinitely until task is cancelled by parent process
+        try:
+            while True:
+                await asyncio.sleep(1.0)
+        except asyncio.CancelledError:
+            logger.info("Vector daemon shutdown requested")
+            raise
     
     async def _monitor_projects(self) -> None:
         """Monitor projects for vector indexing requirements."""
         logger.info("Starting project monitoring")
         
-        while not self.shutdown_requested:
+        while self.is_running:
             try:
                 # Get all projects that need vector indexing
                 projects = await self.db_manager.get_all_projects()
@@ -176,7 +156,7 @@ class VectorDaemon:
         """Worker task to process queued items."""
         logger.info(f"Starting worker: {worker_id}")
         
-        while not self.shutdown_requested:
+        while self.is_running:
             try:
                 # Get task from queue with timeout
                 try:
@@ -245,7 +225,7 @@ class VectorDaemon:
     
     async def _stats_reporter(self) -> None:
         """Periodically report daemon statistics."""
-        while not self.shutdown_requested:
+        while self.is_running:
             try:
                 uptime = time.time() - self.stats["start_time"]