mcp-code-indexer 4.2.4__tar.gz → 4.2.6__tar.gz

{mcp_code_indexer-4.2.4 → mcp_code_indexer-4.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mcp-code-indexer
-Version: 4.2.4
+Version: 4.2.6
 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?51)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?51)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?53)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?53)](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.4 → mcp_code_indexer-4.2.6}/README.md

@@ -1,7 +1,7 @@
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?51)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?51)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?53)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?53)](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.4 → mcp_code_indexer-4.2.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "mcp-code-indexer"
-version = "4.2.4"
+version = "4.2.6"
 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.4 → mcp_code_indexer-4.2.6}/src/mcp_code_indexer/vector_mode/__init__.py

@@ -18,9 +18,10 @@ def is_vector_mode_available() -> bool:
         import turbopuffer
         import tree_sitter
         import watchdog
+        # PyYAML package provides yaml module
         import yaml
         return True
-    except ImportError:
+    except ImportError as e:
         return False
 
 def get_vector_config_path() -> Path:

{mcp_code_indexer-4.2.4 → mcp_code_indexer-4.2.6}/src/mcp_code_indexer/vector_mode/daemon.py

@@ -39,11 +39,13 @@ class VectorDaemon:
         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()
         self.processing_queue: asyncio.Queue = asyncio.Queue(maxsize=config.max_queue_size)
         self.workers: list[asyncio.Task] = []
+        self.monitor_tasks: list[asyncio.Task] = []
         
         # Statistics
         self.stats = {
@@ -54,23 +56,24 @@ class VectorDaemon:
             "last_activity": time.time(),
         }
         
-        # Setup signal handlers
-        self._setup_signal_handlers()
+        # 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:
-            signal.signal(signal.SIGTERM, self._signal_handler)
-            signal.signal(signal.SIGINT, self._signal_handler)
+            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'):
-                signal.signal(signal.SIGHUP, self._signal_handler)
+                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, frame) -> None:
+    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()
     
     async def start(self) -> None:
         """Start the vector daemon."""
@@ -79,6 +82,10 @@ class VectorDaemon:
             return
         
         self.is_running = True
+        
+        # Setup signal handlers now that we have an event loop
+        self._setup_signal_handlers()
+        
         logger.info(
             "Starting vector daemon",
             extra={
@@ -101,6 +108,7 @@ class VectorDaemon:
             # Start monitoring tasks
             monitor_task = asyncio.create_task(self._monitor_projects())
             stats_task = asyncio.create_task(self._stats_reporter())
+            self.monitor_tasks.extend([monitor_task, stats_task])
             
             # Wait for shutdown signal
             await self._run_until_shutdown()
@@ -111,13 +119,13 @@ 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."""
-        while not self.shutdown_requested:
-            try:
-                await asyncio.sleep(1.0)
-            except asyncio.CancelledError:
-                break
+        await self.shutdown_event.wait()
     
     async def _monitor_projects(self) -> None:
         """Monitor projects for vector indexing requirements."""
@@ -141,6 +149,9 @@ class VectorDaemon:
                 
                 await asyncio.sleep(self.config.daemon_poll_interval)
                 
+            except asyncio.CancelledError:
+                logger.info("Project monitoring cancelled")
+                break
             except Exception as e:
                 logger.error(f"Error in project monitoring: {e}")
                 self.stats["errors_count"] += 1
@@ -180,6 +191,9 @@ class VectorDaemon:
                 await self._process_task(task, worker_id)
                 self.stats["last_activity"] = time.time()
                 
+            except asyncio.CancelledError:
+                logger.info(f"Worker {worker_id} cancelled")
+                break
             except Exception as e:
                 logger.error(f"Worker {worker_id} error: {e}")
                 self.stats["errors_count"] += 1
@@ -251,6 +265,9 @@ class VectorDaemon:
                 
                 await asyncio.sleep(60.0)  # Report every minute
                 
+            except asyncio.CancelledError:
+                logger.info("Stats reporting cancelled")
+                break
             except Exception as e:
                 logger.error(f"Error in stats reporting: {e}")
                 await asyncio.sleep(10.0)
@@ -264,9 +281,14 @@ class VectorDaemon:
         for worker in self.workers:
             worker.cancel()
         
-        # Wait for workers to finish
-        if self.workers:
-            await asyncio.gather(*self.workers, return_exceptions=True)
+        # Cancel monitor tasks
+        for task in self.monitor_tasks:
+            task.cancel()
+        
+        # Wait for all tasks to finish
+        all_tasks = self.workers + self.monitor_tasks
+        if all_tasks:
+            await asyncio.gather(*all_tasks, return_exceptions=True)
         
         logger.info("Vector daemon shutdown complete")
     
@@ -330,6 +352,7 @@ def main() -> None:
         asyncio.run(start_vector_daemon(args.config, args.db_path, args.cache_dir))
     except KeyboardInterrupt:
         logger.info("Daemon interrupted by user")
+        sys.exit(0)
     except Exception as e:
         logger.error(f"Daemon failed: {e}", exc_info=True)
         sys.exit(1)