hanzo-mcp 0.7.7__py3-none-any.whl → 0.8.1__py3-none-any.whl

hanzo_mcp/tools/vector/infinity_store.py

@@ -2,24 +2,27 @@
 
 import json
 import hashlib
+from typing import Any, Dict, List, Tuple, Optional
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple
 from dataclasses import dataclass
 
 try:
     import infinity_embedded
+
     INFINITY_AVAILABLE = True
 except ImportError:
     # Use mock implementation when infinity_embedded is not available
     from . import mock_infinity as infinity_embedded
+
     INFINITY_AVAILABLE = True  # Mock is always available
 
-from .ast_analyzer import ASTAnalyzer, FileAST, Symbol, create_symbol_embedding_text
+from .ast_analyzer import Symbol, FileAST, ASTAnalyzer, create_symbol_embedding_text
 
 
 @dataclass
 class Document:
     """Document representation for vector storage."""
+
     id: str
     content: str
     metadata: Dict[str, Any]
@@ -30,6 +33,7 @@ class Document:
 @dataclass
 class SearchResult:
     """Search result from vector database."""
+
     document: Document
     score: float
     distance: float
@@ -38,6 +42,7 @@ class SearchResult:
 @dataclass
 class SymbolSearchResult:
     """Search result for symbols."""
+
     symbol: Symbol
     score: float
     context_document: Optional[Document] = None
@@ -46,6 +51,7 @@ class SymbolSearchResult:
 @dataclass
 class UnifiedSearchResult:
     """Search result combining text, vector, and symbol search."""
+
     type: str  # 'document', 'symbol', 'reference'
     content: str
     file_path: str
@@ -58,7 +64,7 @@ class UnifiedSearchResult:
 
 class InfinityVectorStore:
     """Local vector database using Infinity."""
-    
+
     def __init__(
         self,
         data_path: Optional[str] = None,
@@ -66,43 +72,46 @@ class InfinityVectorStore:
         dimension: int = 1536,  # Default for OpenAI text-embedding-3-small
     ):
         """Initialize the Infinity vector store.
-        
+
         Args:
             data_path: Path to store vector database (default: ~/.config/hanzo/vector-store)
             embedding_model: Embedding model to use
             dimension: Vector dimension (must match embedding model)
         """
         if not INFINITY_AVAILABLE:
-            raise ImportError("infinity_embedded is required for vector store functionality")
-        
+            raise ImportError(
+                "infinity_embedded is required for vector store functionality"
+            )
+
         # Set up data path
         if data_path:
             self.data_path = Path(data_path)
         else:
             from hanzo_mcp.config.settings import get_config_dir
+
             self.data_path = get_config_dir() / "vector-store"
-        
+
         self.data_path.mkdir(parents=True, exist_ok=True)
-        
+
         self.embedding_model = embedding_model
         self.dimension = dimension
-        
+
         # Initialize AST analyzer
         self.ast_analyzer = ASTAnalyzer()
-        
+
         # Connect to Infinity
         self.infinity = infinity_embedded.connect(str(self.data_path))
         self.db = self.infinity.get_database("hanzo_mcp")
-        
+
         # Initialize tables
         self._initialize_tables()
-    
+
     def _initialize_tables(self):
         """Initialize database tables if they don't exist."""
         # Documents table
         try:
             self.documents_table = self.db.get_table("documents")
-        except:
+        except Exception:
             self.documents_table = self.db.create_table(
                 "documents",
                 {
@@ -112,13 +121,13 @@ class InfinityVectorStore:
                     "chunk_index": {"type": "integer"},
                     "metadata": {"type": "varchar"},  # JSON string
                     "embedding": {"type": f"vector,{self.dimension},float"},
-                }
+                },
             )
-        
+
         # Symbols table for code symbols
         try:
             self.symbols_table = self.db.get_table("symbols")
-        except:
+        except Exception:
             self.symbols_table = self.db.create_table(
                 "symbols",
                 {
@@ -134,13 +143,13 @@ class InfinityVectorStore:
                     "docstring": {"type": "varchar"},
                     "metadata": {"type": "varchar"},  # JSON string
                     "embedding": {"type": f"vector,{self.dimension},float"},
-                }
+                },
             )
-        
+
         # AST table for storing complete file ASTs
         try:
             self.ast_table = self.db.get_table("ast_files")
-        except:
+        except Exception:
             self.ast_table = self.db.create_table(
                 "ast_files",
                 {
@@ -149,13 +158,13 @@ class InfinityVectorStore:
                     "language": {"type": "varchar"},
                     "ast_data": {"type": "varchar"},  # JSON string of complete AST
                     "last_updated": {"type": "varchar"},  # ISO timestamp
-                }
+                },
             )
-        
+
         # References table for cross-file references
         try:
             self.references_table = self.db.get_table("references")
-        except:
+        except Exception:
             self.references_table = self.db.create_table(
                 "references",
                 {
@@ -163,18 +172,22 @@ class InfinityVectorStore:
                     "source_file": {"type": "varchar"},
                     "target_file": {"type": "varchar"},
                     "symbol_name": {"type": "varchar"},
-                    "reference_type": {"type": "varchar"},  # import, call, inheritance, etc.
+                    "reference_type": {
+                        "type": "varchar"
+                    },  # import, call, inheritance, etc.
                     "line_number": {"type": "integer"},
                     "metadata": {"type": "varchar"},  # JSON string
-                }
+                },
             )
-    
-    def _generate_doc_id(self, content: str, file_path: str = "", chunk_index: int = 0) -> str:
+
+    def _generate_doc_id(
+        self, content: str, file_path: str = "", chunk_index: int = 0
+    ) -> str:
         """Generate a unique document ID."""
         content_hash = hashlib.sha256(content.encode()).hexdigest()[:16]
         path_hash = hashlib.sha256(file_path.encode()).hexdigest()[:8]
         return f"doc_{path_hash}_{chunk_index}_{content_hash}"
-    
+
     def add_document(
         self,
         content: str,
@@ -184,39 +197,43 @@ class InfinityVectorStore:
         embedding: Optional[List[float]] = None,
     ) -> str:
         """Add a document to the vector store.
-        
+
         Args:
             content: Document content
             metadata: Additional metadata
             file_path: Source file path
             chunk_index: Chunk index if document is part of larger file
             embedding: Pre-computed embedding (if None, will compute)
-            
+
         Returns:
             Document ID
         """
         doc_id = self._generate_doc_id(content, file_path or "", chunk_index)
-        
+
         # Generate embedding if not provided
         if embedding is None:
             embedding = self._generate_embedding(content)
-        
+
         # Prepare metadata
         metadata = metadata or {}
         metadata_json = json.dumps(metadata)
-        
+
         # Insert document
-        self.documents_table.insert([{
-            "id": doc_id,
-            "content": content,
-            "file_path": file_path or "",
-            "chunk_index": chunk_index,
-            "metadata": metadata_json,
-            "embedding": embedding,
-        }])
-        
+        self.documents_table.insert(
+            [
+                {
+                    "id": doc_id,
+                    "content": content,
+                    "file_path": file_path or "",
+                    "chunk_index": chunk_index,
+                    "metadata": metadata_json,
+                    "embedding": embedding,
+                }
+            ]
+        )
+
         return doc_id
-    
+
     def add_file(
         self,
         file_path: str,
@@ -225,45 +242,47 @@ class InfinityVectorStore:
         metadata: Dict[str, Any] = None,
     ) -> List[str]:
         """Add a file to the vector store by chunking it.
-        
+
         Args:
             file_path: Path to the file to add
             chunk_size: Maximum characters per chunk
             chunk_overlap: Characters to overlap between chunks
             metadata: Additional metadata for all chunks
-            
+
         Returns:
             List of document IDs for all chunks
         """
         path = Path(file_path)
         if not path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
-        
+
         # Read file content
         try:
-            content = path.read_text(encoding='utf-8')
+            content = path.read_text(encoding="utf-8")
         except UnicodeDecodeError:
             # Try with different encoding
-            content = path.read_text(encoding='latin-1')
-        
+            content = path.read_text(encoding="latin-1")
+
         # Chunk the content
         chunks = self._chunk_text(content, chunk_size, chunk_overlap)
-        
+
         # Add metadata
         file_metadata = metadata or {}
-        file_metadata.update({
-            "file_name": path.name,
-            "file_extension": path.suffix,
-            "file_size": path.stat().st_size,
-        })
-        
+        file_metadata.update(
+            {
+                "file_name": path.name,
+                "file_extension": path.suffix,
+                "file_size": path.stat().st_size,
+            }
+        )
+
         # Add each chunk
         doc_ids = []
         for i, chunk in enumerate(chunks):
             chunk_metadata = file_metadata.copy()
             chunk_metadata["chunk_number"] = i
             chunk_metadata["total_chunks"] = len(chunks)
-            
+
             doc_id = self.add_document(
                 content=chunk,
                 metadata=chunk_metadata,
@@ -271,9 +290,9 @@ class InfinityVectorStore:
                 chunk_index=i,
             )
             doc_ids.append(doc_id)
-        
+
         return doc_ids
-    
+
     def add_file_with_ast(
         self,
         file_path: str,
@@ -282,138 +301,146 @@ class InfinityVectorStore:
         metadata: Dict[str, Any] = None,
     ) -> Tuple[List[str], Optional[FileAST]]:
         """Add a file with full AST analysis and symbol extraction.
-        
+
         Args:
             file_path: Path to the file to add
             chunk_size: Maximum characters per chunk for content
             chunk_overlap: Characters to overlap between chunks
             metadata: Additional metadata for all chunks
-            
+
         Returns:
             Tuple of (document IDs for content chunks, FileAST object)
         """
         path = Path(file_path)
         if not path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
-        
+
         # First add file content using existing method
         doc_ids = self.add_file(file_path, chunk_size, chunk_overlap, metadata)
-        
+
         # Analyze AST and symbols
         file_ast = self.ast_analyzer.analyze_file(file_path)
         if not file_ast:
             return doc_ids, None
-        
+
         # Store complete AST
         self._store_file_ast(file_ast)
-        
+
         # Store individual symbols with embeddings
         self._store_symbols(file_ast.symbols)
-        
+
         # Store cross-references
         self._store_references(file_ast)
-        
+
         return doc_ids, file_ast
-    
+
     def _store_file_ast(self, file_ast: FileAST):
         """Store complete file AST information."""
         from datetime import datetime
-        
+
         # Remove existing AST for this file
         try:
             self.ast_table.delete(f"file_path = '{file_ast.file_path}'")
-        except:
+        except Exception:
             pass
-        
+
         # Insert new AST
-        self.ast_table.insert([{
-            "file_path": file_ast.file_path,
-            "file_hash": file_ast.file_hash,
-            "language": file_ast.language,
-            "ast_data": json.dumps(file_ast.to_dict()),
-            "last_updated": datetime.now().isoformat(),
-        }])
-    
+        self.ast_table.insert(
+            [
+                {
+                    "file_path": file_ast.file_path,
+                    "file_hash": file_ast.file_hash,
+                    "language": file_ast.language,
+                    "ast_data": json.dumps(file_ast.to_dict()),
+                    "last_updated": datetime.now().isoformat(),
+                }
+            ]
+        )
+
     def _store_symbols(self, symbols: List[Symbol]):
         """Store symbols with vector embeddings."""
         if not symbols:
             return
-        
+
         # Remove existing symbols for these files
         file_paths = list(set(symbol.file_path for symbol in symbols))
         for file_path in file_paths:
             try:
                 self.symbols_table.delete(f"file_path = '{file_path}'")
-            except:
+            except Exception:
                 pass
-        
+
         # Insert new symbols
         symbol_records = []
         for symbol in symbols:
             # Create embedding text for symbol
             embedding_text = create_symbol_embedding_text(symbol)
             embedding = self._generate_embedding(embedding_text)
-            
+
             # Generate symbol ID
             symbol_id = self._generate_symbol_id(symbol)
-            
+
             # Prepare metadata
             symbol_metadata = {
                 "references": symbol.references,
                 "embedding_text": embedding_text,
             }
-            
-            symbol_records.append({
-                "id": symbol_id,
-                "name": symbol.name,
-                "type": symbol.type,
-                "file_path": symbol.file_path,
-                "line_start": symbol.line_start,
-                "line_end": symbol.line_end,
-                "scope": symbol.scope or "",
-                "parent": symbol.parent or "",
-                "signature": symbol.signature or "",
-                "docstring": symbol.docstring or "",
-                "metadata": json.dumps(symbol_metadata),
-                "embedding": embedding,
-            })
-        
+
+            symbol_records.append(
+                {
+                    "id": symbol_id,
+                    "name": symbol.name,
+                    "type": symbol.type,
+                    "file_path": symbol.file_path,
+                    "line_start": symbol.line_start,
+                    "line_end": symbol.line_end,
+                    "scope": symbol.scope or "",
+                    "parent": symbol.parent or "",
+                    "signature": symbol.signature or "",
+                    "docstring": symbol.docstring or "",
+                    "metadata": json.dumps(symbol_metadata),
+                    "embedding": embedding,
+                }
+            )
+
         if symbol_records:
             self.symbols_table.insert(symbol_records)
-    
+
     def _store_references(self, file_ast: FileAST):
         """Store cross-file references."""
         if not file_ast.dependencies:
             return
-        
+
         # Remove existing references for this file
         try:
             self.references_table.delete(f"source_file = '{file_ast.file_path}'")
-        except:
+        except Exception:
             pass
-        
+
         # Insert new references
         reference_records = []
         for i, dependency in enumerate(file_ast.dependencies):
             ref_id = f"{file_ast.file_path}_{dependency}_{i}"
-            reference_records.append({
-                "id": ref_id,
-                "source_file": file_ast.file_path,
-                "target_file": dependency,
-                "symbol_name": dependency,
-                "reference_type": "import",
-                "line_number": 0,  # Could be enhanced to track actual line numbers
-                "metadata": json.dumps({}),
-            })
-        
+            reference_records.append(
+                {
+                    "id": ref_id,
+                    "source_file": file_ast.file_path,
+                    "target_file": dependency,
+                    "symbol_name": dependency,
+                    "reference_type": "import",
+                    "line_number": 0,  # Could be enhanced to track actual line numbers
+                    "metadata": json.dumps({}),
+                }
+            )
+
         if reference_records:
             self.references_table.insert(reference_records)
-    
+
     def _generate_symbol_id(self, symbol: Symbol) -> str:
         """Generate unique symbol ID."""
         text = f"{symbol.file_path}_{symbol.type}_{symbol.name}_{symbol.line_start}"
         return hashlib.sha256(text.encode()).hexdigest()[:16]
-    
+
     def search_symbols(
         self,
         query: str,
@@ -423,37 +450,37 @@ class InfinityVectorStore:
         score_threshold: float = 0.0,
     ) -> List[SymbolSearchResult]:
         """Search for symbols using vector similarity.
-        
+
         Args:
             query: Search query
             symbol_type: Filter by symbol type (function, class, variable, etc.)
             file_path: Filter by file path
             limit: Maximum number of results
             score_threshold: Minimum similarity score
-            
+
         Returns:
             List of symbol search results
         """
         # Generate query embedding
         query_embedding = self._generate_embedding(query)
-        
+
         # Build search query
         search_query = self.symbols_table.output(["*"]).match_dense(
-            "embedding", 
-            query_embedding, 
-            "float", 
+            "embedding",
+            query_embedding,
+            "float",
             "ip",  # Inner product
-            limit * 2  # Get more results for filtering
+            limit * 2,  # Get more results for filtering
         )
-        
+
         # Apply filters
         if symbol_type:
             search_query = search_query.filter(f"type = '{symbol_type}'")
         if file_path:
             search_query = search_query.filter(f"file_path = '{file_path}'")
-        
+
         search_results = search_query.to_pl()
-        
+
         # Convert to SymbolSearchResult objects
         results = []
         for row in search_results.iter_rows(named=True):
@@ -462,9 +489,9 @@ class InfinityVectorStore:
                 # Parse metadata
                 try:
                     metadata = json.loads(row["metadata"])
-                except:
+                except Exception:
                     metadata = {}
-                
+
                 # Create Symbol object
                 symbol = Symbol(
                     name=row["name"],
@@ -473,21 +500,23 @@ class InfinityVectorStore:
                     line_start=row["line_start"],
                     line_end=row["line_end"],
                     column_start=0,  # Not stored in table
-                    column_end=0,   # Not stored in table
+                    column_end=0,  # Not stored in table
                     scope=row["scope"],
                     parent=row["parent"] if row["parent"] else None,
                     docstring=row["docstring"] if row["docstring"] else None,
                     signature=row["signature"] if row["signature"] else None,
                     references=metadata.get("references", []),
                 )
-                
-                results.append(SymbolSearchResult(
-                    symbol=symbol,
-                    score=score,
-                ))
-        
+
+                results.append(
+                    SymbolSearchResult(
+                        symbol=symbol,
+                        score=score,
+                    )
+                )
+
         return results[:limit]
-    
+
     def search_ast_nodes(
         self,
         file_path: str,
@@ -495,24 +524,28 @@ class InfinityVectorStore:
         node_name: Optional[str] = None,
     ) -> Optional[FileAST]:
         """Search AST nodes within a specific file.
-        
+
         Args:
             file_path: File to search in
             node_type: Filter by AST node type
             node_name: Filter by node name
-            
+
         Returns:
             FileAST object if file found, None otherwise
         """
         try:
-            results = self.ast_table.output(["*"]).filter(f"file_path = '{file_path}'").to_pl()
-            
+            results = (
+                self.ast_table.output(["*"])
+                .filter(f"file_path = '{file_path}'")
+                .to_pl()
+            )
+
             if len(results) == 0:
                 return None
-            
+
             row = next(results.iter_rows(named=True))
             ast_data = json.loads(row["ast_data"])
-            
+
             # Reconstruct FileAST object
             file_ast = FileAST(
                 file_path=ast_data["file_path"],
@@ -524,44 +557,52 @@ class InfinityVectorStore:
                 exports=ast_data["exports"],
                 dependencies=ast_data["dependencies"],
             )
-            
+
             return file_ast
-            
+
         except Exception as e:
             import logging
+
             logger = logging.getLogger(__name__)
             logger.error(f"Error searching AST nodes: {e}")
             return None
-    
+
     def get_file_references(self, file_path: str) -> List[Dict[str, Any]]:
         """Get all files that reference the given file.
-        
+
         Args:
             file_path: File to find references for
-            
+
         Returns:
             List of reference information
         """
         try:
-            results = self.references_table.output(["*"]).filter(f"target_file = '{file_path}'").to_pl()
-            
+            results = (
+                self.references_table.output(["*"])
+                .filter(f"target_file = '{file_path}'")
+                .to_pl()
+            )
+
             references = []
             for row in results.iter_rows(named=True):
-                references.append({
-                    "source_file": row["source_file"],
-                    "symbol_name": row["symbol_name"],
-                    "reference_type": row["reference_type"],
-                    "line_number": row["line_number"],
-                })
-            
+                references.append(
+                    {
+                        "source_file": row["source_file"],
+                        "symbol_name": row["symbol_name"],
+                        "reference_type": row["reference_type"],
+                        "line_number": row["line_number"],
+                    }
+                )
+
             return references
-            
+
         except Exception as e:
             import logging
+
             logger = logging.getLogger(__name__)
             logger.error(f"Error getting file references: {e}")
             return []
-    
+
     def search(
         self,
         query: str,
@@ -570,37 +611,41 @@ class InfinityVectorStore:
         filters: Dict[str, Any] = None,
     ) -> List[SearchResult]:
         """Search for similar documents.
-        
+
         Args:
             query: Search query
             limit: Maximum number of results
             score_threshold: Minimum similarity score
             filters: Metadata filters (not yet implemented)
-            
+
         Returns:
             List of search results
         """
         # Generate query embedding
         query_embedding = self._generate_embedding(query)
-        
+
         # Perform vector search
-        search_results = self.documents_table.output(["*"]).match_dense(
-            "embedding", 
-            query_embedding, 
-            "float", 
-            "ip",  # Inner product (cosine similarity)
-            limit
-        ).to_pl()
-        
+        search_results = (
+            self.documents_table.output(["*"])
+            .match_dense(
+                "embedding",
+                query_embedding,
+                "float",
+                "ip",  # Inner product (cosine similarity)
+                limit,
+            )
+            .to_pl()
+        )
+
         # Convert to SearchResult objects
         results = []
         for row in search_results.iter_rows(named=True):
             # Parse metadata
             try:
                 metadata = json.loads(row["metadata"])
-            except:
+            except Exception:
                 metadata = {}
-            
+
             # Create document
             document = Document(
                 id=row["id"],
@@ -609,64 +654,70 @@ class InfinityVectorStore:
                 file_path=row["file_path"] if row["file_path"] else None,
                 chunk_index=row["chunk_index"],
             )
-            
+
             # Score is the similarity (higher is better)
             score = row.get("score", 0.0)
             distance = 1.0 - score  # Convert similarity to distance
-            
+
             if score >= score_threshold:
-                results.append(SearchResult(
-                    document=document,
-                    score=score,
-                    distance=distance,
-                ))
-        
+                results.append(
+                    SearchResult(
+                        document=document,
+                        score=score,
+                        distance=distance,
+                    )
+                )
+
         return results
-    
+
     def delete_document(self, doc_id: str) -> bool:
         """Delete a document by ID.
-        
+
         Args:
             doc_id: Document ID to delete
-            
+
         Returns:
             True if document was deleted
         """
         try:
             self.documents_table.delete(f"id = '{doc_id}'")
             return True
-        except:
+        except Exception:
             return False
-    
+
     def delete_file(self, file_path: str) -> int:
         """Delete all documents from a specific file.
-        
+
         Args:
             file_path: File path to delete documents for
-            
+
         Returns:
             Number of documents deleted
         """
         try:
             # Get count first
-            results = self.documents_table.output(["id"]).filter(f"file_path = '{file_path}'").to_pl()
+            results = (
+                self.documents_table.output(["id"])
+                .filter(f"file_path = '{file_path}'")
+                .to_pl()
+            )
             count = len(results)
-            
+
             # Delete all documents for this file
             self.documents_table.delete(f"file_path = '{file_path}'")
             return count
-        except:
+        except Exception:
             return 0
-    
+
     def list_files(self) -> List[Dict[str, Any]]:
         """List all indexed files.
-        
+
         Returns:
             List of file information
         """
         try:
             results = self.documents_table.output(["file_path", "metadata"]).to_pl()
-            
+
             files = {}
             for row in results.iter_rows(named=True):
                 file_path = row["file_path"]
@@ -675,63 +726,66 @@ class InfinityVectorStore:
                         metadata = json.loads(row["metadata"])
                         files[file_path] = {
                             "file_path": file_path,
-                            "file_name": metadata.get("file_name", Path(file_path).name),
+                            "file_name": metadata.get(
+                                "file_name", Path(file_path).name
+                            ),
                             "file_size": metadata.get("file_size", 0),
                             "total_chunks": metadata.get("total_chunks", 1),
                         }
-                    except:
+                    except Exception:
                         files[file_path] = {
                             "file_path": file_path,
                             "file_name": Path(file_path).name,
                         }
-            
+
             return list(files.values())
-        except:
+        except Exception:
             return []
-    
+
     def _chunk_text(self, text: str, chunk_size: int, overlap: int) -> List[str]:
         """Split text into overlapping chunks."""
         if len(text) <= chunk_size:
             return [text]
-        
+
         chunks = []
         start = 0
-        
+
         while start < len(text):
             end = start + chunk_size
-            
+
             # Try to break at word boundary
             if end < len(text):
                 # Look back for a good break point
                 break_point = end
                 for i in range(end - 100, start + 100, -1):
-                    if i > 0 and text[i] in '\n\r.!?':
+                    if i > 0 and text[i] in "\n\r.!?":
                         break_point = i + 1
                         break
                 end = break_point
-            
+
             chunk = text[start:end].strip()
             if chunk:
                 chunks.append(chunk)
-            
+
             start = max(start + chunk_size - overlap, end)
-        
+
         return chunks
-    
+
     def _generate_embedding(self, text: str) -> List[float]:
         """Generate embedding for text.
-        
+
         For now, this returns a dummy embedding. In a real implementation,
         you would call an embedding API (OpenAI, Cohere, etc.) or use a local model.
         """
         # This is a placeholder - you would implement actual embedding generation here
         # For now, return a random embedding of the correct dimension
         import random
+
         return [random.random() for _ in range(self.dimension)]
-    
+
     async def get_stats(self) -> Dict[str, Any]:
         """Get statistics about the vector store.
-        
+
         Returns:
             Dictionary with statistics
         """
@@ -739,30 +793,30 @@ class InfinityVectorStore:
             # Get document count
             doc_count_result = self.documents_table.output(["count(*)"]).to_pl()
             doc_count = doc_count_result.item(0, 0) if len(doc_count_result) > 0 else 0
-            
+
             # Get unique file count
             file_result = self.documents_table.output(["file_path"]).to_pl()
             unique_files = set()
             for row in file_result.iter_rows():
                 if row[0]:
                     unique_files.add(row[0])
-            
+
             # Get symbol count
             symbol_count = 0
             try:
                 symbol_result = self.symbols_table.output(["count(*)"]).to_pl()
                 symbol_count = symbol_result.item(0, 0) if len(symbol_result) > 0 else 0
-            except:
+            except Exception:
                 pass
-            
+
             # Get AST count
             ast_count = 0
             try:
                 ast_result = self.ast_table.output(["count(*)"]).to_pl()
                 ast_count = ast_result.item(0, 0) if len(ast_result) > 0 else 0
-            except:
+            except Exception:
                 pass
-            
+
             return {
                 "document_count": doc_count,
                 "vector_count": doc_count,  # Each document has a vector
@@ -779,57 +833,58 @@ class InfinityVectorStore:
                 "document_count": 0,
                 "vector_count": 0,
             }
-    
+
     async def clear(self) -> bool:
         """Clear all data from the vector store.
-        
+
         Returns:
             True if successful
         """
         try:
             # Delete all records from all tables
             self.documents_table.delete()
-            
+
             try:
                 self.symbols_table.delete()
-            except:
+            except Exception:
                 pass
-            
+
             try:
                 self.ast_table.delete()
-            except:
+            except Exception:
                 pass
-            
+
             try:
                 self.references_table.delete()
-            except:
+            except Exception:
                 pass
-            
+
             return True
         except Exception as e:
             import logging
+
             logger = logging.getLogger(__name__)
             logger.error(f"Error clearing vector store: {e}")
             return False
-    
+
     async def index_document(
         self,
         content: str,
         metadata: Dict[str, Any] = None,
     ) -> str:
         """Async version of add_document for consistency.
-        
+
         Args:
             content: Document content
             metadata: Additional metadata
-            
+
         Returns:
             Document ID
         """
         file_path = metadata.get("path") if metadata else None
         return self.add_document(content, metadata, file_path)
-    
+
     def close(self):
         """Close the database connection."""
-        if hasattr(self, 'infinity'):
-            self.infinity.disconnect()
+        if hasattr(self, "infinity"):
+            self.infinity.disconnect()