mcp-vector-search 0.9.3__py3-none-any.whl → 0.12.1__py3-none-any.whl

mcp_vector_search/cli/didyoumean.py

@@ -57,6 +57,11 @@ class EnhancedDidYouMeanTyper(typer.Typer):
         # Get the underlying click group
         click_group = super().__call__(*args, **kwargs)
 
+        # If click_group is None (command already executed), return None
+        # This happens after command execution completes successfully
+        if click_group is None:
+            return None
+
         # Create enhanced DYM group with original group's properties
         enhanced_group = EnhancedDidYouMeanGroup(
             name=click_group.name,
@@ -161,6 +166,11 @@ def enhance_existing_typer(app: typer.Typer) -> typer.Typer:
         """Enhanced call that uses EnhancedDidYouMeanGroup."""
         click_group = original_call(*args, **kwargs)
 
+        # If click_group is None (command already executed), return None
+        # This happens after command execution completes successfully
+        if click_group is None:
+            return None
+
         # Create enhanced group
         enhanced_group = EnhancedDidYouMeanGroup(
             name=click_group.name,

mcp_vector_search/cli/main.py

@@ -33,13 +33,15 @@ unfamiliar codebases, finding similar patterns, and integrating with AI tools.
   3. Check status: [green]mcp-vector-search status[/green]
 
 [bold cyan]Main Commands:[/bold cyan]
-  init       🔧 Initialize project
+  install    📦 Install project and MCP integrations
+  uninstall  🗑️  Remove MCP integrations
+  init       🔧 Initialize project (simple)
   demo       🎬 Run interactive demo
   doctor     🩺 Check system health
   status     📊 Show project status
   search     🔍 Search code semantically
   index      📇 Index codebase
-  mcp        🤖 MCP integration for AI tools
+  mcp        🔌 MCP server operations
   config     ⚙️  Configure settings
   visualize  📊 Visualize code relationships
   help       ❓ Get help
@@ -56,48 +58,61 @@ from .commands.config import config_app  # noqa: E402
 from .commands.demo import demo_app  # noqa: E402
 from .commands.index import index_app  # noqa: E402
 from .commands.init import init_app  # noqa: E402
+from .commands.install import install_app  # noqa: E402
 from .commands.mcp import mcp_app  # noqa: E402
 from .commands.search import search_app, search_main  # noqa: E402, F401
 from .commands.status import main as status_main  # noqa: E402
+from .commands.uninstall import uninstall_app  # noqa: E402
 from .commands.visualize import app as visualize_app  # noqa: E402
 
 # ============================================================================
 # MAIN COMMANDS - Clean hierarchy
 # ============================================================================
 
-# 1. INIT - Initialize project
+# 1. INSTALL - Install project and MCP integrations (NEW!)
+app.add_typer(
+    install_app, name="install", help="📦 Install project and MCP integrations"
+)
+
+# 2. UNINSTALL - Remove MCP integrations (NEW!)
+app.add_typer(uninstall_app, name="uninstall", help="🗑️  Remove MCP integrations")
+app.add_typer(uninstall_app, name="remove", help="🗑️  Remove MCP integrations (alias)")
+
+# 3. INIT - Initialize project (simplified)
 # Use Typer group for init to support both direct call and subcommands
 app.add_typer(init_app, name="init", help="🔧 Initialize project for semantic search")
 
-# 2. DEMO - Interactive demo
+# 4. DEMO - Interactive demo
 app.add_typer(demo_app, name="demo", help="🎬 Run interactive demo with sample project")
 
-# 3. DOCTOR - System health check
+# 5. DOCTOR - System health check
 # (defined below inline)
 
-# 4. STATUS - Project status
+# 6. STATUS - Project status
 app.command("status", help="📊 Show project status and statistics")(status_main)
 
-# 5. SEARCH - Search code
+# 7. SEARCH - Search code
 # Register search as both a command and a typer group
 app.add_typer(search_app, name="search", help="🔍 Search code semantically")
 
-# 6. INDEX - Index codebase
+# 8. INDEX - Index codebase
 app.add_typer(index_app, name="index", help="📇 Index codebase for semantic search")
 
-# 7. MCP - MCP integration
-app.add_typer(mcp_app, name="mcp", help="🤖 Manage MCP integration for AI tools")
+# 9. MCP - MCP server operations (RESERVED for server ops only!)
+app.add_typer(mcp_app, name="mcp", help="🔌 MCP server operations")
 
-# 8. CONFIG - Configuration
+# 10. CONFIG - Configuration
 app.add_typer(config_app, name="config", help="⚙️  Manage project configuration")
 
-# 9. VISUALIZE - Code graph visualization
-app.add_typer(visualize_app, name="visualize", help="📊 Visualize code chunk relationships")
+# 11. VISUALIZE - Code graph visualization
+app.add_typer(
+    visualize_app, name="visualize", help="📊 Visualize code chunk relationships"
+)
 
-# 10. HELP - Enhanced help
+# 12. HELP - Enhanced help
 # (defined below inline)
 
-# 10. VERSION - Version info
+# 13. VERSION - Version info
 # (defined below inline)
 
 
@@ -120,11 +135,9 @@ def _deprecated_command(old_cmd: str, new_cmd: str):
     return wrapper
 
 
-# Deprecated: install -> init
-@app.command("install", hidden=True)
-def deprecated_install():
-    """[DEPRECATED] Use 'init' instead."""
-    _deprecated_command("install", "init")()
+# NOTE: 'install' command is now the primary command for project installation
+# Old 'install' was deprecated in favor of 'init' in v0.7.0
+# Now 'install' is back as the hierarchical installation command in v0.13.0
 
 
 # Deprecated: find -> search
@@ -432,7 +445,12 @@ def cli_with_suggestions():
     except Exception as e:
         # For other exceptions, show error and exit if verbose logging is enabled
         # Suppress internal framework errors in normal operation
-        if "--verbose" in sys.argv or "-v" in sys.argv:
+
+        # Suppress harmless didyoumean framework AttributeError (known issue)
+        # This occurs during Click/Typer cleanup after successful command completion
+        if isinstance(e, AttributeError) and "attribute" in str(e) and "name" in str(e):
+            pass  # Ignore - this is a harmless framework cleanup error
+        elif "--verbose" in sys.argv or "-v" in sys.argv:
             click.echo(f"Unexpected error: {e}", err=True)
             sys.exit(1)
         # Otherwise, just exit silently to avoid confusing error messages

mcp_vector_search/core/connection_pool.py

@@ -24,6 +24,16 @@ class PooledConnection:
     in_use: bool = False
     use_count: int = 0
 
+    @property
+    def age(self) -> float:
+        """Get the age of this connection in seconds."""
+        return time.time() - self.created_at
+
+    @property
+    def idle_time(self) -> float:
+        """Get the idle time of this connection in seconds."""
+        return time.time() - self.last_used
+
 
 class ChromaConnectionPool:
     """Connection pool for ChromaDB operations."""
@@ -209,18 +219,18 @@ class ChromaConnectionPool:
                 logger.debug(f"Created new connection (pool size: {len(self._pool)})")
                 return conn
 
-            # Pool is full, wait for a connection to become available
-            self._stats["pool_misses"] += 1
-            logger.warning(
-                "Connection pool exhausted, waiting for available connection"
-            )
+        # Pool is full, wait for a connection to become available (outside lock)
+        self._stats["pool_misses"] += 1
+        logger.warning("Connection pool exhausted, waiting for available connection")
 
-            # Wait for a connection (with timeout)
-            timeout = 30.0  # 30 seconds
-            start_time = time.time()
+        # Wait for a connection (with timeout) - release lock during wait
+        timeout = 30.0  # 30 seconds
+        start_time = time.time()
 
-            while time.time() - start_time < timeout:
-                await asyncio.sleep(0.1)
+        while time.time() - start_time < timeout:
+            await asyncio.sleep(0.1)
+            # Re-acquire lock to check for available connections
+            async with self._lock:
                 for conn in self._pool:
                     if not conn.in_use and self._is_connection_valid(conn):
                         conn.in_use = True
@@ -229,7 +239,7 @@ class ChromaConnectionPool:
                         self._stats["connections_reused"] += 1
                         return conn
 
-            raise DatabaseError("Connection pool timeout: no connections available")
+        raise DatabaseError("Connection pool timeout: no connections available")
 
     async def _release_connection(self, conn: PooledConnection) -> None:
         """Release a connection back to the pool."""
@@ -320,3 +330,31 @@ class ChromaConnectionPool:
         except Exception as e:
             logger.error(f"Connection pool health check failed: {e}")
             return False
+
+    # Backward compatibility aliases for old test API
+    async def cleanup(self) -> None:
+        """Alias for close() method (backward compatibility)."""
+        await self.close()
+
+    def _validate_connection(self, conn: PooledConnection) -> bool:
+        """Alias for _is_connection_valid() method (backward compatibility)."""
+        return self._is_connection_valid(conn)
+
+    async def _cleanup_idle_connections(self) -> None:
+        """Alias for _cleanup_expired_connections() method (backward compatibility)."""
+        await self._cleanup_expired_connections()
+
+    @property
+    def _connections(self) -> list[PooledConnection]:
+        """Alias for _pool attribute (backward compatibility)."""
+        return self._pool
+
+    @property
+    def _max_connections(self) -> int:
+        """Alias for max_connections attribute (backward compatibility)."""
+        return self.max_connections
+
+    @property
+    def _min_connections(self) -> int:
+        """Alias for min_connections attribute (backward compatibility)."""
+        return self.min_connections

mcp_vector_search/core/database.py

@@ -1,6 +1,7 @@
 """Database abstraction and ChromaDB implementation for MCP Vector Search."""
 
 import asyncio
+import json
 import shutil
 from abc import ABC, abstractmethod
 from pathlib import Path
@@ -273,16 +274,16 @@ class ChromaVectorDatabase(VectorDatabase):
                     "class_name": chunk.class_name or "",
                     "docstring": chunk.docstring or "",
                     "complexity_score": chunk.complexity_score,
-                    # Hierarchy fields
+                    # Hierarchy fields (convert lists to JSON strings for ChromaDB)
                     "chunk_id": chunk.chunk_id or "",
                     "parent_chunk_id": chunk.parent_chunk_id or "",
-                    "child_chunk_ids": chunk.child_chunk_ids or [],
+                    "child_chunk_ids": json.dumps(chunk.child_chunk_ids or []),
                     "chunk_depth": chunk.chunk_depth,
-                    # Additional metadata
-                    "decorators": chunk.decorators or [],
-                    "parameters": chunk.parameters or [],
+                    # Additional metadata (convert lists/dicts to JSON strings)
+                    "decorators": json.dumps(chunk.decorators or []),
+                    "parameters": json.dumps(chunk.parameters or []),
                     "return_type": chunk.return_type or "",
-                    "type_annotations": chunk.type_annotations or {},
+                    "type_annotations": json.dumps(chunk.type_annotations or {}),
                     # Monorepo support
                     "subproject_name": chunk.subproject_name or "",
                     "subproject_path": chunk.subproject_path or "",
@@ -500,16 +501,31 @@ class ChromaVectorDatabase(VectorDatabase):
 
         try:
             # Get all documents from collection
-            results = self._collection.get(
-                include=["metadatas", "documents"]
-            )
+            results = self._collection.get(include=["metadatas", "documents"])
 
             chunks = []
             if results and results.get("ids"):
-                for i, chunk_id in enumerate(results["ids"]):
+                for i, _chunk_id in enumerate(results["ids"]):
                     metadata = results["metadatas"][i]
                     content = results["documents"][i]
 
+                    # Parse JSON strings back to lists/dicts
+                    child_chunk_ids = metadata.get("child_chunk_ids", "[]")
+                    if isinstance(child_chunk_ids, str):
+                        child_chunk_ids = json.loads(child_chunk_ids)
+
+                    decorators = metadata.get("decorators", "[]")
+                    if isinstance(decorators, str):
+                        decorators = json.loads(decorators)
+
+                    parameters = metadata.get("parameters", "[]")
+                    if isinstance(parameters, str):
+                        parameters = json.loads(parameters)
+
+                    type_annotations = metadata.get("type_annotations", "{}")
+                    if isinstance(type_annotations, str):
+                        type_annotations = json.loads(type_annotations)
+
                     chunk = CodeChunk(
                         content=content,
                         file_path=Path(metadata["file_path"]),
@@ -524,12 +540,12 @@ class ChromaVectorDatabase(VectorDatabase):
                         complexity_score=metadata.get("complexity_score", 0.0),
                         chunk_id=metadata.get("chunk_id"),
                         parent_chunk_id=metadata.get("parent_chunk_id"),
-                        child_chunk_ids=metadata.get("child_chunk_ids", []),
+                        child_chunk_ids=child_chunk_ids,
                         chunk_depth=metadata.get("chunk_depth", 0),
-                        decorators=metadata.get("decorators", []),
-                        parameters=metadata.get("parameters", []),
+                        decorators=decorators,
+                        parameters=parameters,
                         return_type=metadata.get("return_type"),
-                        type_annotations=metadata.get("type_annotations", {}),
+                        type_annotations=type_annotations,
                         subproject_name=metadata.get("subproject_name"),
                         subproject_path=metadata.get("subproject_path"),
                     )
@@ -775,16 +791,18 @@ class PooledChromaVectorDatabase(VectorDatabase):
                             "class_name": chunk.class_name or "",
                             "docstring": chunk.docstring or "",
                             "complexity_score": chunk.complexity_score,
-                            # Hierarchy fields
+                            # Hierarchy fields (convert lists to JSON strings for ChromaDB)
                             "chunk_id": chunk.chunk_id or "",
                             "parent_chunk_id": chunk.parent_chunk_id or "",
-                            "child_chunk_ids": chunk.child_chunk_ids or [],
+                            "child_chunk_ids": json.dumps(chunk.child_chunk_ids or []),
                             "chunk_depth": chunk.chunk_depth,
-                            # Additional metadata
-                            "decorators": chunk.decorators or [],
-                            "parameters": chunk.parameters or [],
+                            # Additional metadata (convert lists/dicts to JSON strings)
+                            "decorators": json.dumps(chunk.decorators or []),
+                            "parameters": json.dumps(chunk.parameters or []),
                             "return_type": chunk.return_type or "",
-                            "type_annotations": chunk.type_annotations or {},
+                            "type_annotations": json.dumps(
+                                chunk.type_annotations or {}
+                            ),
                             # Monorepo support
                             "subproject_name": chunk.subproject_name or "",
                             "subproject_path": chunk.subproject_path or "",
@@ -1013,16 +1031,31 @@ class PooledChromaVectorDatabase(VectorDatabase):
         try:
             async with self._pool.get_connection() as conn:
                 # Get all documents from collection
-                results = conn.collection.get(
-                    include=["metadatas", "documents"]
-                )
+                results = conn.collection.get(include=["metadatas", "documents"])
 
                 chunks = []
                 if results and results.get("ids"):
-                    for i, chunk_id in enumerate(results["ids"]):
+                    for i, _chunk_id in enumerate(results["ids"]):
                         metadata = results["metadatas"][i]
                         content = results["documents"][i]
 
+                        # Parse JSON strings back to lists/dicts
+                        child_chunk_ids = metadata.get("child_chunk_ids", "[]")
+                        if isinstance(child_chunk_ids, str):
+                            child_chunk_ids = json.loads(child_chunk_ids)
+
+                        decorators = metadata.get("decorators", "[]")
+                        if isinstance(decorators, str):
+                            decorators = json.loads(decorators)
+
+                        parameters = metadata.get("parameters", "[]")
+                        if isinstance(parameters, str):
+                            parameters = json.loads(parameters)
+
+                        type_annotations = metadata.get("type_annotations", "{}")
+                        if isinstance(type_annotations, str):
+                            type_annotations = json.loads(type_annotations)
+
                         chunk = CodeChunk(
                             content=content,
                             file_path=Path(metadata["file_path"]),
@@ -1037,12 +1070,12 @@ class PooledChromaVectorDatabase(VectorDatabase):
                             complexity_score=metadata.get("complexity_score", 0.0),
                             chunk_id=metadata.get("chunk_id"),
                             parent_chunk_id=metadata.get("parent_chunk_id"),
-                            child_chunk_ids=metadata.get("child_chunk_ids", []),
+                            child_chunk_ids=child_chunk_ids,
                             chunk_depth=metadata.get("chunk_depth", 0),
-                            decorators=metadata.get("decorators", []),
-                            parameters=metadata.get("parameters", []),
+                            decorators=decorators,
+                            parameters=parameters,
                             return_type=metadata.get("return_type"),
-                            type_annotations=metadata.get("type_annotations", {}),
+                            type_annotations=type_annotations,
                             subproject_name=metadata.get("subproject_name"),
                             subproject_path=metadata.get("subproject_path"),
                         )