mcp-code-indexer 4.0.1__tar.gz → 4.1.0__tar.gz

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mcp-code-indexer
-Version: 4.0.1
+Version: 4.1.0
 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
@@ -23,6 +23,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
+Provides-Extra: vector
 Requires-Dist: aiofiles (==23.2.0)
 Requires-Dist: aiohttp (>=3.8.0)
 Requires-Dist: aiosqlite (==0.19.0)
@@ -43,8 +44,8 @@ Description-Content-Type: text/markdown
 
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?44)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?44)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?46)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?46)](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.
@@ -183,20 +184,77 @@ The git hook integration provides intelligent automation:
 
 - **📊 Git Analysis**: Automatically analyzes git diffs after commits/merges
 - **🤖 AI Processing**: Uses OpenRouter API with Anthropic's Claude Sonnet 4
-- **⚡ Smart Updates**: Only processes files that actually changed  
+- **⚡ Smart Updates**: Only processes files that actually changed
 - **🔄 Overview Maintenance**: Updates project overview when structure changes
 - **🛡️ Error Isolation**: Git operations continue even if indexing fails
 - **⏱️ Rate Limiting**: Built-in retry logic with exponential backoff
 
 ### 🎯 Key Benefits
 
-💡 **Zero Manual Work**: Descriptions stay current without any effort  
-⚡ **Performance**: Only analyzes changed files, not entire codebase  
-🔒 **Reliability**: Robust error handling ensures git operations never fail  
-🎛️ **Configurable**: Support for custom models and timeout settings  
+💡 **Zero Manual Work**: Descriptions stay current without any effort
+⚡ **Performance**: Only analyzes changed files, not entire codebase
+🔒 **Reliability**: Robust error handling ensures git operations never fail
+🎛️ **Configurable**: Support for custom models and timeout settings
 
 **Learn More**: See [Git Hook Setup Guide](docs/git-hook-setup.md) for complete configuration options and troubleshooting.
 
+## 🧠 Vector Mode (BETA)
+
+🚀 **NEW Feature**: Semantic code search with vector embeddings! Experience AI-powered code discovery that understands context and meaning, not just keywords.
+
+### 🎯 What is Vector Mode?
+
+Vector Mode transforms how you search and understand codebases by using AI embeddings:
+
+- **🔍 Semantic Search**: Find code by meaning, not just text matching
+- **⚡ Real-time Indexing**: Automatic embedding generation as code changes  
+- **🛡️ Secure by Default**: Comprehensive secret redaction before API calls
+- **🌐 Multi-language**: Python, JavaScript, TypeScript with AST-based chunking
+- **📊 Smart Chunking**: Context-aware code segmentation for optimal embeddings
+
+### 🚀 Quick Start
+
+```bash
+# Install vector mode dependencies
+pip install mcp-code-indexer[vector]
+
+# Set required API keys
+export VOYAGE_API_KEY="pa-your-voyage-api-key"
+export TURBOPUFFER_API_KEY="your-turbopuffer-api-key" 
+
+# Start with vector mode enabled
+mcp-code-indexer --vector
+
+# The daemon automatically starts and begins indexing your projects
+```
+
+### 💡 Key Features
+
+- **🔐 Secret Redaction**: 20+ pattern types automatically detected and redacted
+- **🌳 Merkle Trees**: Efficient change detection without full directory scans
+- **🎛️ Circuit Breakers**: Resilient API integration with automatic retry logic
+- **📈 Production Ready**: Built for high-concurrency with comprehensive monitoring
+
+### 🔧 Advanced Configuration
+
+```bash
+# Custom configuration
+mcp-code-indexer --vector --vector-config /path/to/config.yaml
+
+# HTTP mode with vector search
+mcp-code-indexer --vector --http --port 8080
+```
+
+### 🛠️ Architecture
+
+Vector Mode adds powerful new MCP tools:
+- `vector_search` - Semantic code search across projects
+- `similarity_search` - Find similar code patterns  
+- `dependency_search` - Discover code relationships
+- `vector_status` - Monitor indexing progress
+
+**Status**: Currently in BETA - foundations implemented, full pipeline in development.
+
 ## 🔧 Development Setup
 
 ### 👨‍💻 For Contributors
@@ -369,14 +427,14 @@ Comprehensive documentation organized by user journey and expertise level.
 
 ### 📋 Quick References
 - **[Examples & Integrations](examples/)** - Ready-to-use configurations
-- **[Troubleshooting](#🚨-troubleshooting)** - Common issues & solutions  
+- **[Troubleshooting](#🚨-troubleshooting)** - Common issues & solutions
 - **[API Tools Summary](#🛠️-mcp-tools-available)** - All 11 tools at a glance
 
 **📚 Reading Paths:**
 - **New to MCP Code Indexer?** Quick Start → API Reference → HTTP API → Q&A Interface
 - **Web developers?** Quick Start → HTTP API Reference → Q&A Interface → Git Hooks
 - **AI/ML engineers?** Quick Start → Q&A Interface → API Reference → Git Hooks
-- **Setting up for a team?** CLI Reference → Configuration → Administrative Commands → Monitoring  
+- **Setting up for a team?** CLI Reference → Configuration → Administrative Commands → Monitoring
 - **Contributing to the project?** Architecture → Contributing → API Reference
 
 ## 🚦 System Requirements
@@ -390,7 +448,7 @@ Comprehensive documentation organized by user journey and expertise level.
 
 Tested with codebases up to **10,000 files**:
 - File description retrieval: **< 10ms**
-- Full-text search: **< 100ms** 
+- Full-text search: **< 100ms**
 - Codebase overview generation: **< 2s**
 - Merge conflict detection: **< 5s**
 
@@ -460,11 +518,11 @@ async def analyze_codebase(project_path):
         "projectName": "my-project",
         "folderPath": project_path
     })
-    
+
     if size_info["isLarge"]:
         # Use search for large codebases
         results = await mcp_client.call_tool("search_descriptions", {
-            "projectName": "my-project", 
+            "projectName": "my-project",
             "folderPath": project_path,
             "query": "authentication logic"
         })
@@ -484,18 +542,18 @@ async def analyze_codebase(project_path):
     python -c "
     import asyncio
     from mcp_client import MCPClient
-    
+
     async def update_descriptions():
         client = MCPClient('mcp-code-indexer')
-        
+
         # Find files without descriptions
         missing = await client.call_tool('find_missing_descriptions', {
             'projectName': '${{ github.repository }}',
             'folderPath': '.'
         })
-        
+
         # Process with AI and update...
-    
+
     asyncio.run(update_descriptions())
     "
 ```
@@ -606,7 +664,7 @@ mcp-code-indexer --map PROJECT_NAME
 ## 🛡️ Security Features
 
 - **Input validation** on all MCP tool parameters
-- **SQL injection protection** via parameterized queries  
+- **SQL injection protection** via parameterized queries
 - **File system sandboxing** with .gitignore respect
 - **Error sanitization** to prevent information leakage
 - **Async resource cleanup** to prevent memory leaks
@@ -638,7 +696,7 @@ Ready to supercharge your AI agents with intelligent codebase navigation?
 2. **[Master the API tools](docs/api-reference.md)** - Learn all 11 tools with examples
 3. **[Try HTTP API access](docs/http-api.md)** - REST API for web applications
 4. **[Explore AI-powered Q&A](docs/qa-interface.md)** - Ask questions about your code
-5. **[Set up git hooks](docs/git-hook-setup.md)** - Automate your workflow  
+5. **[Set up git hooks](docs/git-hook-setup.md)** - Automate your workflow
 
 **👥 Setting up for a team?**
 1. **[Learn all CLI commands](docs/cli-reference.md)** - Complete command reference
@@ -661,7 +719,7 @@ Ready to supercharge your AI agents with intelligent codebase navigation?
 
 We welcome contributions! See our **[Contributing Guide](docs/contributing.md)** for:
 - Development setup
-- Code style guidelines  
+- Code style guidelines
 - Testing requirements
 - Pull request process
 
@@ -672,7 +730,7 @@ MIT License - see **[LICENSE](LICENSE)** for details.
 ## 🙏 Built With
 
 - **[Model Context Protocol](https://github.com/modelcontextprotocol/python-sdk)** - The foundation for tool integration
-- **[tiktoken](https://pypi.org/project/tiktoken/)** - Fast BPE tokenization  
+- **[tiktoken](https://pypi.org/project/tiktoken/)** - Fast BPE tokenization
 - **[aiosqlite](https://pypi.org/project/aiosqlite/)** - Async SQLite operations
 - **[aiohttp](https://pypi.org/project/aiohttp/)** - Async HTTP client for OpenRouter API
 - **[tenacity](https://pypi.org/project/tenacity/)** - Robust retry logic and rate limiting
@@ -680,9 +738,9 @@ MIT License - see **[LICENSE](LICENSE)** for details.
 
 ---
 
-**Transform how your AI agents understand code!** 🚀  
+**Transform how your AI agents understand code!** 🚀
 
-🎯 **New User?** [Get started in 2 minutes](#-quick-start)  
-👨‍💻 **Developer?** [Explore the complete API](docs/api-reference.md)  
+🎯 **New User?** [Get started in 2 minutes](#-quick-start)
+👨‍💻 **Developer?** [Explore the complete API](docs/api-reference.md)
 🔧 **Production?** [Deploy with confidence](docs/configuration.md)
 

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/README.md

@@ -1,7 +1,7 @@
 # MCP Code Indexer 🚀
 
-[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?44)](https://badge.fury.io/py/mcp-code-indexer)
-[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?44)](https://pypi.org/project/mcp-code-indexer/)
+[![PyPI version](https://badge.fury.io/py/mcp-code-indexer.svg?46)](https://badge.fury.io/py/mcp-code-indexer)
+[![Python](https://img.shields.io/pypi/pyversions/mcp-code-indexer.svg?46)](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.
@@ -140,20 +140,77 @@ The git hook integration provides intelligent automation:
 
 - **📊 Git Analysis**: Automatically analyzes git diffs after commits/merges
 - **🤖 AI Processing**: Uses OpenRouter API with Anthropic's Claude Sonnet 4
-- **⚡ Smart Updates**: Only processes files that actually changed  
+- **⚡ Smart Updates**: Only processes files that actually changed
 - **🔄 Overview Maintenance**: Updates project overview when structure changes
 - **🛡️ Error Isolation**: Git operations continue even if indexing fails
 - **⏱️ Rate Limiting**: Built-in retry logic with exponential backoff
 
 ### 🎯 Key Benefits
 
-💡 **Zero Manual Work**: Descriptions stay current without any effort  
-⚡ **Performance**: Only analyzes changed files, not entire codebase  
-🔒 **Reliability**: Robust error handling ensures git operations never fail  
-🎛️ **Configurable**: Support for custom models and timeout settings  
+💡 **Zero Manual Work**: Descriptions stay current without any effort
+⚡ **Performance**: Only analyzes changed files, not entire codebase
+🔒 **Reliability**: Robust error handling ensures git operations never fail
+🎛️ **Configurable**: Support for custom models and timeout settings
 
 **Learn More**: See [Git Hook Setup Guide](docs/git-hook-setup.md) for complete configuration options and troubleshooting.
 
+## 🧠 Vector Mode (BETA)
+
+🚀 **NEW Feature**: Semantic code search with vector embeddings! Experience AI-powered code discovery that understands context and meaning, not just keywords.
+
+### 🎯 What is Vector Mode?
+
+Vector Mode transforms how you search and understand codebases by using AI embeddings:
+
+- **🔍 Semantic Search**: Find code by meaning, not just text matching
+- **⚡ Real-time Indexing**: Automatic embedding generation as code changes  
+- **🛡️ Secure by Default**: Comprehensive secret redaction before API calls
+- **🌐 Multi-language**: Python, JavaScript, TypeScript with AST-based chunking
+- **📊 Smart Chunking**: Context-aware code segmentation for optimal embeddings
+
+### 🚀 Quick Start
+
+```bash
+# Install vector mode dependencies
+pip install mcp-code-indexer[vector]
+
+# Set required API keys
+export VOYAGE_API_KEY="pa-your-voyage-api-key"
+export TURBOPUFFER_API_KEY="your-turbopuffer-api-key" 
+
+# Start with vector mode enabled
+mcp-code-indexer --vector
+
+# The daemon automatically starts and begins indexing your projects
+```
+
+### 💡 Key Features
+
+- **🔐 Secret Redaction**: 20+ pattern types automatically detected and redacted
+- **🌳 Merkle Trees**: Efficient change detection without full directory scans
+- **🎛️ Circuit Breakers**: Resilient API integration with automatic retry logic
+- **📈 Production Ready**: Built for high-concurrency with comprehensive monitoring
+
+### 🔧 Advanced Configuration
+
+```bash
+# Custom configuration
+mcp-code-indexer --vector --vector-config /path/to/config.yaml
+
+# HTTP mode with vector search
+mcp-code-indexer --vector --http --port 8080
+```
+
+### 🛠️ Architecture
+
+Vector Mode adds powerful new MCP tools:
+- `vector_search` - Semantic code search across projects
+- `similarity_search` - Find similar code patterns  
+- `dependency_search` - Discover code relationships
+- `vector_status` - Monitor indexing progress
+
+**Status**: Currently in BETA - foundations implemented, full pipeline in development.
+
 ## 🔧 Development Setup
 
 ### 👨‍💻 For Contributors
@@ -326,14 +383,14 @@ Comprehensive documentation organized by user journey and expertise level.
 
 ### 📋 Quick References
 - **[Examples & Integrations](examples/)** - Ready-to-use configurations
-- **[Troubleshooting](#🚨-troubleshooting)** - Common issues & solutions  
+- **[Troubleshooting](#🚨-troubleshooting)** - Common issues & solutions
 - **[API Tools Summary](#🛠️-mcp-tools-available)** - All 11 tools at a glance
 
 **📚 Reading Paths:**
 - **New to MCP Code Indexer?** Quick Start → API Reference → HTTP API → Q&A Interface
 - **Web developers?** Quick Start → HTTP API Reference → Q&A Interface → Git Hooks
 - **AI/ML engineers?** Quick Start → Q&A Interface → API Reference → Git Hooks
-- **Setting up for a team?** CLI Reference → Configuration → Administrative Commands → Monitoring  
+- **Setting up for a team?** CLI Reference → Configuration → Administrative Commands → Monitoring
 - **Contributing to the project?** Architecture → Contributing → API Reference
 
 ## 🚦 System Requirements
@@ -347,7 +404,7 @@ Comprehensive documentation organized by user journey and expertise level.
 
 Tested with codebases up to **10,000 files**:
 - File description retrieval: **< 10ms**
-- Full-text search: **< 100ms** 
+- Full-text search: **< 100ms**
 - Codebase overview generation: **< 2s**
 - Merge conflict detection: **< 5s**
 
@@ -417,11 +474,11 @@ async def analyze_codebase(project_path):
         "projectName": "my-project",
         "folderPath": project_path
     })
-    
+
     if size_info["isLarge"]:
         # Use search for large codebases
         results = await mcp_client.call_tool("search_descriptions", {
-            "projectName": "my-project", 
+            "projectName": "my-project",
             "folderPath": project_path,
             "query": "authentication logic"
         })
@@ -441,18 +498,18 @@ async def analyze_codebase(project_path):
     python -c "
     import asyncio
     from mcp_client import MCPClient
-    
+
     async def update_descriptions():
         client = MCPClient('mcp-code-indexer')
-        
+
         # Find files without descriptions
         missing = await client.call_tool('find_missing_descriptions', {
             'projectName': '${{ github.repository }}',
             'folderPath': '.'
         })
-        
+
         # Process with AI and update...
-    
+
     asyncio.run(update_descriptions())
     "
 ```
@@ -563,7 +620,7 @@ mcp-code-indexer --map PROJECT_NAME
 ## 🛡️ Security Features
 
 - **Input validation** on all MCP tool parameters
-- **SQL injection protection** via parameterized queries  
+- **SQL injection protection** via parameterized queries
 - **File system sandboxing** with .gitignore respect
 - **Error sanitization** to prevent information leakage
 - **Async resource cleanup** to prevent memory leaks
@@ -595,7 +652,7 @@ Ready to supercharge your AI agents with intelligent codebase navigation?
 2. **[Master the API tools](docs/api-reference.md)** - Learn all 11 tools with examples
 3. **[Try HTTP API access](docs/http-api.md)** - REST API for web applications
 4. **[Explore AI-powered Q&A](docs/qa-interface.md)** - Ask questions about your code
-5. **[Set up git hooks](docs/git-hook-setup.md)** - Automate your workflow  
+5. **[Set up git hooks](docs/git-hook-setup.md)** - Automate your workflow
 
 **👥 Setting up for a team?**
 1. **[Learn all CLI commands](docs/cli-reference.md)** - Complete command reference
@@ -618,7 +675,7 @@ Ready to supercharge your AI agents with intelligent codebase navigation?
 
 We welcome contributions! See our **[Contributing Guide](docs/contributing.md)** for:
 - Development setup
-- Code style guidelines  
+- Code style guidelines
 - Testing requirements
 - Pull request process
 
@@ -629,7 +686,7 @@ MIT License - see **[LICENSE](LICENSE)** for details.
 ## 🙏 Built With
 
 - **[Model Context Protocol](https://github.com/modelcontextprotocol/python-sdk)** - The foundation for tool integration
-- **[tiktoken](https://pypi.org/project/tiktoken/)** - Fast BPE tokenization  
+- **[tiktoken](https://pypi.org/project/tiktoken/)** - Fast BPE tokenization
 - **[aiosqlite](https://pypi.org/project/aiosqlite/)** - Async SQLite operations
 - **[aiohttp](https://pypi.org/project/aiohttp/)** - Async HTTP client for OpenRouter API
 - **[tenacity](https://pypi.org/project/tenacity/)** - Robust retry logic and rate limiting
@@ -637,8 +694,8 @@ MIT License - see **[LICENSE](LICENSE)** for details.
 
 ---
 
-**Transform how your AI agents understand code!** 🚀  
+**Transform how your AI agents understand code!** 🚀
 
-🎯 **New User?** [Get started in 2 minutes](#-quick-start)  
-👨‍💻 **Developer?** [Explore the complete API](docs/api-reference.md)  
+🎯 **New User?** [Get started in 2 minutes](#-quick-start)
+👨‍💻 **Developer?** [Explore the complete API](docs/api-reference.md)
 🔧 **Production?** [Deploy with confidence](docs/configuration.md)

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "mcp-code-indexer"
-version = "4.0.1"
+version = "4.1.0"
 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"]
@@ -14,10 +14,10 @@ homepage = "https://github.com/fluffypony/mcp-code-indexer"
 repository = "https://github.com/fluffypony/mcp-code-indexer"
 documentation = "https://github.com/fluffypony/mcp-code-indexer/blob/main/README.md"
 keywords = [
-    "mcp", 
-    "model-context-protocol", 
-    "code-indexer", 
-    "ai-tools", 
+    "mcp",
+    "model-context-protocol",
+    "code-indexer",
+    "ai-tools",
     "codebase-navigation",
     "file-descriptions",
     "llm-tools"
@@ -59,6 +59,15 @@ fastapi = ">=0.104.0"
 uvicorn = ">=0.24.0"
 python-multipart = ">=0.0.6"
 
+[tool.poetry.extras]
+vector = [
+    "voyageai",
+    "turbopuffer", 
+    "tree-sitter",
+    "watchdog",
+    "pyyaml"
+]
+
 [tool.poetry.group.dev.dependencies]
 pytest = ">=8.0.0"
 pytest-asyncio = ">=0.21.0"
@@ -111,7 +120,7 @@ known_first_party = ["src", "mcp_code_indexer"]
 
 # MyPy configuration
 [tool.mypy]
-python_version = "3.9"
+python_version = "3.10"
 warn_return_any = true
 warn_unused_configs = true
 disallow_untyped_defs = true
@@ -171,3 +180,15 @@ exclude_lines = [
     "class .*\\bProtocol\\):",
     "@(abc\\.)?abstractmethod"
 ]
+
+# Ruff configuration
+[tool.ruff]
+exclude = [
+    ".vulture_whitelist.py",
+    "venv",
+    ".git",
+    "__pycache__",
+    ".mypy_cache",
+    ".pytest_cache",
+    ".ruff_cache"
+]

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/src/mcp_code_indexer/__init__.py

@@ -6,10 +6,12 @@ intelligent codebase navigation through searchable file descriptions,
 token-aware overviews, and advanced merge capabilities.
 """
 
+
 # Delay import to avoid dependency issues during testing
-def get_server():
+def get_server() -> type:
     """Get MCPCodeIndexServer (lazy import)."""
     from .server.mcp_server import MCPCodeIndexServer
+
     return MCPCodeIndexServer
 
 
@@ -27,15 +29,15 @@ def _get_version() -> str:
         for pkg_name in ["mcp-code-indexer", "mcp_code_indexer"]:
             try:
                 return version(pkg_name)
-            except Exception:
+            except Exception:  # nosec B112
                 continue
-    except Exception:
+    except Exception:  # nosec B110
         pass
 
     # Fallback to reading from pyproject.toml (for development)
     try:
-        from pathlib import Path
         import sys
+        from pathlib import Path
 
         if sys.version_info >= (3, 11):
             import tomllib
@@ -48,7 +50,7 @@ def _get_version() -> str:
         pyproject_path = Path(__file__).parent.parent.parent / "pyproject.toml"
         with open(pyproject_path, "rb") as f:
             data = tomllib.load(f)
-        return data["project"]["version"]
+        return str(data["project"]["version"])
     except Exception:
         return "dev"
 

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/src/mcp_code_indexer/ask_handler.py

@@ -8,9 +8,9 @@ and sending them to Claude via OpenRouter API for direct responses.
 
 import logging
 from pathlib import Path
-from typing import Dict, Optional, Any
+from typing import Any, Dict, Optional
 
-from .claude_api_handler import ClaudeAPIHandler, ClaudeAPIError
+from .claude_api_handler import ClaudeAPIError, ClaudeAPIHandler
 from .database.database import DatabaseManager
 
 

{mcp_code_indexer-4.0.1 → mcp_code_indexer-4.1.0}/src/mcp_code_indexer/claude_api_handler.py

@@ -11,15 +11,15 @@ import json
 import logging
 import os
 from dataclasses import dataclass
-from typing import Dict, List, Optional, Any
 from pathlib import Path
+from typing import Any, Dict, List, Optional
 
 import aiohttp
 from tenacity import (
     retry,
-    wait_exponential,
-    stop_after_attempt,
     retry_if_exception_type,
+    stop_after_attempt,
+    wait_exponential,
 )
 
 from .database.database import DatabaseManager
@@ -194,7 +194,6 @@ class ClaudeAPIHandler:
                 async with session.post(
                     self.OPENROUTER_API_URL, headers=headers, json=payload
                 ) as response:
-
                     self.logger.info(f"Claude API response status: {response.status}")
 
                     if response.status == 429:
@@ -240,7 +239,7 @@ class ClaudeAPIHandler:
             raise ClaudeAPIError("Claude API request timed out")
 
     def validate_json_response(
-        self, response_text: str, required_keys: List[str] = None
+        self, response_text: str, required_keys: Optional[List[str]] = None
     ) -> Dict[str, Any]:
         """
         Validate and parse JSON response from Claude.
@@ -297,6 +296,12 @@ class ClaudeAPIHandler:
                     self.logger.debug(f"Extracted JSON from response: {extracted_json}")
                 data = json.loads(extracted_json)
 
+            # Ensure data is a dictionary
+            if not isinstance(data, dict):
+                raise ClaudeValidationError(
+                    f"Expected JSON object, got {type(data).__name__}"
+                )
+
             # Validate required keys if specified
             if required_keys:
                 missing_keys = [key for key in required_keys if key not in data]