mcp-code-indexer 4.2.4__py3-none-any.whl → 4.2.6__py3-none-any.whl

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/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)

{mcp_code_indexer-4.2.4.dist-info → mcp_code_indexer-4.2.6.dist-info}/METADATA

@@ -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.dist-info → mcp_code_indexer-4.2.6.dist-info}/RECORD

@@ -41,13 +41,13 @@ mcp_code_indexer/transport/__init__.py,sha256=OrdabRjO3EChnTZ06IIFxO6imRQ3PwtRtJ
 mcp_code_indexer/transport/base.py,sha256=Lb3IrL8wr1QvUQpx1GyBQW5bvDpJoUFyXWNAw0dbOK4,3258
 mcp_code_indexer/transport/http_transport.py,sha256=iDxW8CXEIPlpyOtSFkU1qw1FtbKbXgCvJXfVtlXbzIo,13291
 mcp_code_indexer/transport/stdio_transport.py,sha256=a-Pu3usx_NwkHsN2VU8Qe0EwcA7PGL54Gbu2Ee8e0lU,4792
-mcp_code_indexer/vector_mode/__init__.py,sha256=SwCTBPazMHshPPjSTHVGjyVG2mKPTuGP_G1ifVzt6ds,1086
+mcp_code_indexer/vector_mode/__init__.py,sha256=B6FwxtXOFR8CWZQRc40gHnLRrwmWRvpKcOQ8DU9qLrI,1137
 mcp_code_indexer/vector_mode/chunking/__init__.py,sha256=rjjFMbHsqWIBzL4IajYxXXJud_RvBrpFNjVcxnRIWCE,490
 mcp_code_indexer/vector_mode/chunking/ast_chunker.py,sha256=GTl_6U0nSgDRRzKS07tJ7RMX8AmJvvY_IsRn95hvVfA,14623
 mcp_code_indexer/vector_mode/chunking/chunk_optimizer.py,sha256=xD0zEibjt6FLBFaKHNc63-iKTtCgnOlLL_9Hc8mCrzE,19752
 mcp_code_indexer/vector_mode/chunking/language_handlers.py,sha256=YEpTVjzyJH445OjniGV05apexsfG5KVR4lwBEl4mGJc,18189
 mcp_code_indexer/vector_mode/config.py,sha256=g5p9Q4EAR20DfLv4RxaQnk3_UdysuvWS8rcsjs1vgwI,6680
-mcp_code_indexer/vector_mode/daemon.py,sha256=rIm7TvPFKUiPbMuTPyrafIEw5YA8sfMk3lRobKooyiI,12247
+mcp_code_indexer/vector_mode/daemon.py,sha256=fw1gJ6_R3vCt41-uv3M8UVoHv5qHmC3PwGaFBY41Uvk,13218
 mcp_code_indexer/vector_mode/monitoring/__init__.py,sha256=9rNWCvHxRMvYumdIrPjb5K9fpOwe1Aem24hdh8gXoDM,439
 mcp_code_indexer/vector_mode/monitoring/change_detector.py,sha256=X82e_sKbJJFPhqZFJubLQb8Rs-srRtS7sh0nUOsPCPw,10338
 mcp_code_indexer/vector_mode/monitoring/file_watcher.py,sha256=AQ6YHSKXPubtprLZngeLb0othJOCNQZ7wwXUvqwphT4,15299
@@ -58,8 +58,8 @@ mcp_code_indexer/vector_mode/providers/voyage_client.py,sha256=pfm9BOx5Temf0LM-V
 mcp_code_indexer/vector_mode/security/__init__.py,sha256=itfeuysSqV-m9xuo-CMkAoucxexVfPgeOU-ieTLvdls,336
 mcp_code_indexer/vector_mode/security/patterns.py,sha256=0xaiMnZm7YXswq3hVe_DJYePE9MhWuvizApLnmXus9M,11572
 mcp_code_indexer/vector_mode/security/redactor.py,sha256=tsFzhCJ99bp4EFqQVjZ-4f8Uf3ux9X4ODVR09oJG01U,13380
-mcp_code_indexer-4.2.4.dist-info/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
-mcp_code_indexer-4.2.4.dist-info/METADATA,sha256=NqTq19XV8AEAw-pV3fwqNQolFvjZkhcYauLABye5zHQ,27483
-mcp_code_indexer-4.2.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-mcp_code_indexer-4.2.4.dist-info/entry_points.txt,sha256=UABj7HZ0mC6rvF22gxaz2LLNLGQShTrFmp5u00iUtvo,67
-mcp_code_indexer-4.2.4.dist-info/RECORD,,
+mcp_code_indexer-4.2.6.dist-info/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
+mcp_code_indexer-4.2.6.dist-info/METADATA,sha256=ls_h9QUFCgYDJujnGSmscBSdOzzaXtPUYWtR2kglXfc,27483
+mcp_code_indexer-4.2.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+mcp_code_indexer-4.2.6.dist-info/entry_points.txt,sha256=UABj7HZ0mC6rvF22gxaz2LLNLGQShTrFmp5u00iUtvo,67
+mcp_code_indexer-4.2.6.dist-info/RECORD,,