vector-inspector 0.2.1__tar.gz → 0.2.2__tar.gz

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/PKG-INFO

@@ -1,9 +1,13 @@
 Metadata-Version: 2.1
 Name: vector-inspector
-Version: 0.2.1
+Version: 0.2.2
 Summary: A comprehensive desktop application for visualizing, querying, and managing vector database data
 Author-Email: Anthony Dawson <anthonypdawson+github@gmail.com>
 License: MIT
+Project-URL: Homepage, https://github.com/anthony-dawson/vector-inspector
+Project-URL: Source, https://github.com/anthony-dawson/vector-inspector
+Project-URL: Issues, https://github.com/anthony-dawson/vector-inspector/issues
+Project-URL: Documentation, https://github.com/anthony-dawson/vector-inspector#readme
 Requires-Python: ==3.12.*
 Requires-Dist: chromadb>=0.4.22
 Requires-Dist: qdrant-client>=1.7.0
@@ -26,12 +30,13 @@ A comprehensive desktop application for visualizing, querying, and managing vect
 
 ## Overview
 
+Vector Inspector bridges the gap between vector databases and user-friendly data exploration tools. While vector databases are powerful for semantic search and AI applications, they often lack the intuitive inspection and management tools that traditional SQL databases have. This project aims to provide that missing layer.
+
 ## Table of Contents
 
 - [Overview](#overview)
 - [Key Features](#key-features)
 - [Architecture](#architecture)
-- [Application Structure](#application-structure)
 - [Use Cases](#use-cases)
 - [Feature Access](#feature-access)
 - [Roadmap](#roadmap)
@@ -42,8 +47,6 @@ A comprehensive desktop application for visualizing, querying, and managing vect
 - [License](#license)
 - [Acknowledgments](#acknowledgments)
 
-Vector Inspector bridges the gap between vector databases and user-friendly data exploration tools. While vector databases are powerful for semantic search and AI applications, they often lack the intuitive inspection and management tools that traditional SQL databases have. This project aims to provide that missing layer.
-
 ## Key Features
 
 ### 1. **Multi-Provider Support**
@@ -157,8 +160,8 @@ vector-inspector
 
 ```bash
 # Clone the repository
-git clone https://github.com/anthonypdawson/vector-viewer.git
-cd vector-viewer
+git clone https://github.com/anthonypdawson/vector-inspector.git
+cd vector-inspector
 
 # Install dependencies using PDM
 pdm install

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/README.md

@@ -1,225 +1,224 @@
-# Vector Inspector
-
-
-A comprehensive desktop application for visualizing, querying, and managing vector database data. Similar to SQL database viewers, Vector Inspector provides an intuitive GUI for exploring vector embeddings, metadata, and performing similarity searches across multiple vector database providers.
-
-## Overview
-
-## Table of Contents
-
-- [Overview](#overview)
-- [Key Features](#key-features)
-- [Architecture](#architecture)
-- [Application Structure](#application-structure)
-- [Use Cases](#use-cases)
-- [Feature Access](#feature-access)
-- [Roadmap](#roadmap)
-- [Installation](#installation)
-- [Configuration](#configuration)
-- [Development Setup](#development-setup)
-- [Contributing](#contributing)
-- [License](#license)
-- [Acknowledgments](#acknowledgments)
-
-Vector Inspector bridges the gap between vector databases and user-friendly data exploration tools. While vector databases are powerful for semantic search and AI applications, they often lack the intuitive inspection and management tools that traditional SQL databases have. This project aims to provide that missing layer.
-
-## Key Features
-
-### 1. **Multi-Provider Support**
-- Connect to vector databases:
-  - ChromaDB (persistent local storage)
-  - Qdrant (remote server or embedded local)
-- Unified interface regardless of backend provider
-- Automatically saves last connection configuration
-
-### 2. **Data Visualization**
-- **Metadata Explorer**: Browse and filter vector entries by metadata fields
-- **Vector Dimensionality Reduction**: Visualize high-dimensional vectors in 2D/3D using:
-  - t-SNE
-  - UMAP
-  - PCA
-- **Cluster Visualization**: Color-code vectors by metadata categories or clustering results
-- **Interactive Plots**: Zoom, pan, and select vectors for detailed inspection
-- **Data Distribution Charts**: Histograms and statistics for metadata fields
-
-### 3. **Search & Query Interface**
-- **Similarity Search**: 
-  - Text-to-vector search (with embedding model integration)
-  - Vector-to-vector search
-  - Find similar items to selected entries
-  - Adjustable top-k results and similarity thresholds
-- **Metadata Filtering**:
-  - SQL-like query builder for metadata
-  - Combine vector similarity with metadata filters
-  - Advanced filtering: ranges, IN clauses, pattern matching
-- **Hybrid Search**: Combine semantic search with keyword search
-- **Query History**: Save and reuse frequent queries
-
-### 4. **Data Management**
-- **Browse Collections/Indexes**: View all available collections with statistics
-- **CRUD Operations**:
-  - View individual vectors and their metadata
-  - Add new vectors (with auto-embedding options)
-  - Update metadata fields
-  - Delete vectors (single or batch)
-- **Bulk Import/Export**:
-  - Import from CSV, JSON, Parquet
-  - Export query results to various formats
-  - Backup and restore collections
-- **Schema Inspector**: View collection configuration, vector dimensions, metadata schema
-
-### 5. **SQL-Like Experience**
-- **Query Console**: Write queries in a familiar SQL-like syntax (where supported)
-- **Results Grid**: 
-  - Sortable, filterable table view
-  - Pagination for large result sets
-  - Column customization
-- **Data Inspector**: Click any row to see full details including raw vector
-- **Query Execution Plans**: Understand how queries are executed
-- **Auto-completion**: Intelligent suggestions for collection names, fields, and operations
-
-### 6. **Advanced Features**
-- **Embedding Model Integration**:
-  - Use OpenAI, Cohere, HuggingFace models for text-to-vector conversion
-  - Local model support (sentence-transformers)
-  - Custom model integration
-- **Vector Analysis**:
-  - Compute similarity matrices
-  - Identify outliers and anomalies
-  - Cluster analysis with k-means, DBSCAN
-- **Embedding Inspector**:
-  - For similar collections or items, automatically identify which vector dimensions (activations) most contribute to the similarity
-  - Map key activations to interpretable concepts (e.g., 'humor', 'sadness', 'anger') using metadata or labels
-  - Generate human-readable explanations for why items are similar
-- **Performance Monitoring**:
-  - Query latency tracking
-  - Index performance metrics
-  - Connection health monitoring
-
-## Architecture
-
-Vector Inspector is built with PySide6 (Qt for Python) for the GUI, providing a native desktop experience. The backend uses Python with support for multiple vector database providers through a unified interface.
-
-For detailed architecture information, see [docs/architecture.md](docs/architecture.md).
-
-## Use Cases
-
-1. **AI/ML Development**: Inspect embeddings generated during model development
-2. **RAG System Debugging**: Verify what documents are being retrieved
-3. **Data Quality Assurance**: Identify poorly embedded or outlier vectors
-4. **Production Monitoring**: Check vector database health and data consistency
-5. **Data Migration**: Transfer data between vector database providers
-6. **Education**: Learn and experiment with vector databases interactively
-
-## Feature Access
-
-Vector Inspector is available in both free (open source) and Pro versions. The free version includes all core features for ChromaDB and basic Qdrant support, while Pro adds advanced analytics and additional providers.
-
-See [FEATURES.md](FEATURES.md) for a complete feature comparison.
-
-## Roadmap
-
-**Current Status**: ✅ Phase 2 Complete
-
-See [ROADMAP.md](ROADMAP.md) for the complete development roadmap and planned features.
-
-## Installation
-
-### From PyPI (Recommended)
-
-```bash
-pip install vector-inspector
-vector-inspector
-```
-
-### From Source
-
-```bash
-# Clone the repository
-git clone https://github.com/anthonypdawson/vector-viewer.git
-cd vector-viewer
-
-# Install dependencies using PDM
-pdm install
-
-# Launch application
-./run.sh     # Linux/macOS
-./run.bat    # Windows
-```
-
-## Configuration
-
-Paths are resolved relative to the project root (where `pyproject.toml` is). For example, entering `./data/chroma_db` will use the absolute path resolved from the project root.
-
-The application automatically saves your last connection configuration to `~/.vector-viewer/settings.json`. The next time you launch the application, it will attempt to reconnect using the last saved settings.
-
-Example settings structure:
-```json
-{
-  "last_connection": {
-    "provider": "chromadb",
-    "connection_type": "persistent",
-    "path": "./data/chroma_db"
-  }
-}
-```
-
-## Development Setup
-
-```bash
-# Install PDM if you haven't already
-pip install pdm
-
-# Install dependencies with development tools (PDM will create venv automatically)
-pdm install -d
-
-# Run tests
-pdm run pytest
-
-# Run application in development mode
-./run.sh     # Linux/macOS
-./run.bat    # Windows
-
-# Or use Python module directly from src directory:
-cd src
-pdm run python -m vector_viewer
-```
-
-## Contributing
-
-Contributions are welcome! Areas where help is needed:
-- Additional vector database provider integrations
-- UI/UX improvements
-- Performance optimizations
-- Documentation
-- Test coverage
-
-Please see [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines.
-
-## License
-
-MIT License - See [LICENSE](LICENSE) file for details.
-
-## Acknowledgments
-
-This project draws inspiration from:
-- DBeaver (SQL database viewer)
-- MongoDB Compass (NoSQL database GUI)
-- Pinecone Console
-- Various vector database management tools
-
----
-
-**Status**: ✅ Phase 2 Complete - Advanced Features Implemented!
-
-**What's New in Phase 2:**
-- 🔍 Advanced metadata filtering with customizable filter rules (AND/OR logic)
-- ✏️ Double-click to edit items directly in the data browser
-- 📥 Import data from CSV, JSON, and Parquet files
-- 📤 Export filtered data to CSV, JSON, and Parquet formats
-- 💾 Comprehensive backup and restore system for collections
-- 🔄 Metadata filters integrated with search for powerful queries
-
-See [GETTING_STARTED.md](GETTING_STARTED.md) for usage instructions and [IMPLEMENTATION_SUMMARY.md](IMPLEMENTATION_SUMMARY.md) for technical details.
-
-**Contact**: Anthony Dawson
+# Vector Inspector
+
+
+A comprehensive desktop application for visualizing, querying, and managing vector database data. Similar to SQL database viewers, Vector Inspector provides an intuitive GUI for exploring vector embeddings, metadata, and performing similarity searches across multiple vector database providers.
+
+## Overview
+
+Vector Inspector bridges the gap between vector databases and user-friendly data exploration tools. While vector databases are powerful for semantic search and AI applications, they often lack the intuitive inspection and management tools that traditional SQL databases have. This project aims to provide that missing layer.
+
+## Table of Contents
+
+- [Overview](#overview)
+- [Key Features](#key-features)
+- [Architecture](#architecture)
+- [Use Cases](#use-cases)
+- [Feature Access](#feature-access)
+- [Roadmap](#roadmap)
+- [Installation](#installation)
+- [Configuration](#configuration)
+- [Development Setup](#development-setup)
+- [Contributing](#contributing)
+- [License](#license)
+- [Acknowledgments](#acknowledgments)
+
+## Key Features
+
+### 1. **Multi-Provider Support**
+- Connect to vector databases:
+  - ChromaDB (persistent local storage)
+  - Qdrant (remote server or embedded local)
+- Unified interface regardless of backend provider
+- Automatically saves last connection configuration
+
+### 2. **Data Visualization**
+- **Metadata Explorer**: Browse and filter vector entries by metadata fields
+- **Vector Dimensionality Reduction**: Visualize high-dimensional vectors in 2D/3D using:
+  - t-SNE
+  - UMAP
+  - PCA
+- **Cluster Visualization**: Color-code vectors by metadata categories or clustering results
+- **Interactive Plots**: Zoom, pan, and select vectors for detailed inspection
+- **Data Distribution Charts**: Histograms and statistics for metadata fields
+
+### 3. **Search & Query Interface**
+- **Similarity Search**: 
+  - Text-to-vector search (with embedding model integration)
+  - Vector-to-vector search
+  - Find similar items to selected entries
+  - Adjustable top-k results and similarity thresholds
+- **Metadata Filtering**:
+  - SQL-like query builder for metadata
+  - Combine vector similarity with metadata filters
+  - Advanced filtering: ranges, IN clauses, pattern matching
+- **Hybrid Search**: Combine semantic search with keyword search
+- **Query History**: Save and reuse frequent queries
+
+### 4. **Data Management**
+- **Browse Collections/Indexes**: View all available collections with statistics
+- **CRUD Operations**:
+  - View individual vectors and their metadata
+  - Add new vectors (with auto-embedding options)
+  - Update metadata fields
+  - Delete vectors (single or batch)
+- **Bulk Import/Export**:
+  - Import from CSV, JSON, Parquet
+  - Export query results to various formats
+  - Backup and restore collections
+- **Schema Inspector**: View collection configuration, vector dimensions, metadata schema
+
+### 5. **SQL-Like Experience**
+- **Query Console**: Write queries in a familiar SQL-like syntax (where supported)
+- **Results Grid**: 
+  - Sortable, filterable table view
+  - Pagination for large result sets
+  - Column customization
+- **Data Inspector**: Click any row to see full details including raw vector
+- **Query Execution Plans**: Understand how queries are executed
+- **Auto-completion**: Intelligent suggestions for collection names, fields, and operations
+
+### 6. **Advanced Features**
+- **Embedding Model Integration**:
+  - Use OpenAI, Cohere, HuggingFace models for text-to-vector conversion
+  - Local model support (sentence-transformers)
+  - Custom model integration
+- **Vector Analysis**:
+  - Compute similarity matrices
+  - Identify outliers and anomalies
+  - Cluster analysis with k-means, DBSCAN
+- **Embedding Inspector**:
+  - For similar collections or items, automatically identify which vector dimensions (activations) most contribute to the similarity
+  - Map key activations to interpretable concepts (e.g., 'humor', 'sadness', 'anger') using metadata or labels
+  - Generate human-readable explanations for why items are similar
+- **Performance Monitoring**:
+  - Query latency tracking
+  - Index performance metrics
+  - Connection health monitoring
+
+## Architecture
+
+Vector Inspector is built with PySide6 (Qt for Python) for the GUI, providing a native desktop experience. The backend uses Python with support for multiple vector database providers through a unified interface.
+
+For detailed architecture information, see [docs/architecture.md](docs/architecture.md).
+
+## Use Cases
+
+1. **AI/ML Development**: Inspect embeddings generated during model development
+2. **RAG System Debugging**: Verify what documents are being retrieved
+3. **Data Quality Assurance**: Identify poorly embedded or outlier vectors
+4. **Production Monitoring**: Check vector database health and data consistency
+5. **Data Migration**: Transfer data between vector database providers
+6. **Education**: Learn and experiment with vector databases interactively
+
+## Feature Access
+
+Vector Inspector is available in both free (open source) and Pro versions. The free version includes all core features for ChromaDB and basic Qdrant support, while Pro adds advanced analytics and additional providers.
+
+See [FEATURES.md](FEATURES.md) for a complete feature comparison.
+
+## Roadmap
+
+**Current Status**: ✅ Phase 2 Complete
+
+See [ROADMAP.md](ROADMAP.md) for the complete development roadmap and planned features.
+
+## Installation
+
+### From PyPI (Recommended)
+
+```bash
+pip install vector-inspector
+vector-inspector
+```
+
+### From Source
+
+```bash
+# Clone the repository
+git clone https://github.com/anthonypdawson/vector-inspector.git
+cd vector-inspector
+
+# Install dependencies using PDM
+pdm install
+
+# Launch application
+./run.sh     # Linux/macOS
+./run.bat    # Windows
+```
+
+## Configuration
+
+Paths are resolved relative to the project root (where `pyproject.toml` is). For example, entering `./data/chroma_db` will use the absolute path resolved from the project root.
+
+The application automatically saves your last connection configuration to `~/.vector-viewer/settings.json`. The next time you launch the application, it will attempt to reconnect using the last saved settings.
+
+Example settings structure:
+```json
+{
+  "last_connection": {
+    "provider": "chromadb",
+    "connection_type": "persistent",
+    "path": "./data/chroma_db"
+  }
+}
+```
+
+## Development Setup
+
+```bash
+# Install PDM if you haven't already
+pip install pdm
+
+# Install dependencies with development tools (PDM will create venv automatically)
+pdm install -d
+
+# Run tests
+pdm run pytest
+
+# Run application in development mode
+./run.sh     # Linux/macOS
+./run.bat    # Windows
+
+# Or use Python module directly from src directory:
+cd src
+pdm run python -m vector_viewer
+```
+
+## Contributing
+
+Contributions are welcome! Areas where help is needed:
+- Additional vector database provider integrations
+- UI/UX improvements
+- Performance optimizations
+- Documentation
+- Test coverage
+
+Please see [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines.
+
+## License
+
+MIT License - See [LICENSE](LICENSE) file for details.
+
+## Acknowledgments
+
+This project draws inspiration from:
+- DBeaver (SQL database viewer)
+- MongoDB Compass (NoSQL database GUI)
+- Pinecone Console
+- Various vector database management tools
+
+---
+
+**Status**: ✅ Phase 2 Complete - Advanced Features Implemented!
+
+**What's New in Phase 2:**
+- 🔍 Advanced metadata filtering with customizable filter rules (AND/OR logic)
+- ✏️ Double-click to edit items directly in the data browser
+- 📥 Import data from CSV, JSON, and Parquet files
+- 📤 Export filtered data to CSV, JSON, and Parquet formats
+- 💾 Comprehensive backup and restore system for collections
+- 🔄 Metadata filters integrated with search for powerful queries
+
+See [GETTING_STARTED.md](GETTING_STARTED.md) for usage instructions and [IMPLEMENTATION_SUMMARY.md](IMPLEMENTATION_SUMMARY.md) for technical details.
+
+**Contact**: Anthony Dawson

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vector-inspector"
-version = "0.2.1"
+version = "0.2.2"
 description = "A comprehensive desktop application for visualizing, querying, and managing vector database data"
 authors = [
     { name = "Anthony Dawson", email = "anthonypdawson+github@gmail.com" },
@@ -25,6 +25,12 @@ readme = "README.md"
 [project.license]
 text = "MIT"
 
+[project.urls]
+Homepage = "https://github.com/anthony-dawson/vector-inspector"
+Source = "https://github.com/anthony-dawson/vector-inspector"
+Issues = "https://github.com/anthony-dawson/vector-inspector/issues"
+Documentation = "https://github.com/anthony-dawson/vector-inspector#readme"
+
 [project.scripts]
 vector-inspector = "vector_inspector.main:main"
 

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/src/vector_inspector/ui/main_window.py

@@ -282,7 +282,7 @@ class MainWindow(QMainWindow):
             "<h2>Vector Inspector 0.1.0</h2>"
             "<p>A comprehensive desktop application for visualizing, "
             "querying, and managing vector database data.</p>"
-            '<p><a href="https://github.com/anthonypdawson/vector-viewer" style="color:#2980b9;">GitHub Project Page</a></p>'
+            '<p><a href="https://github.com/anthonypdawson/vector-inspector" style="color:#2980b9;">GitHub Project Page</a></p>'
             "<hr />"
             "<p>Built with PySide6 and ChromaDB</p>"
         )

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/tests/test_connections.py

@@ -1,60 +1,60 @@
-import pytest
-from vector_inspector.core.connections.chroma_connection import ChromaDBConnection
-from vector_inspector.core.connections.qdrant_connection import QdrantConnection
-import uuid
-
-@pytest.mark.parametrize("provider", ["chroma", "qdrant"])
-def test_provider_integration(provider, tmp_path):
-    """Test provider connection using standard add_items signature."""
-    collection_name = f"test_collection_{uuid.uuid4().hex[:8]}"
-    test_ids = ["id1", "id2"]
-    test_vectors = [[0.1, 0.2], [0.3, 0.4]]
-    test_docs = ["hello", "world"]
-    test_metadata = [{"type": "greeting"}, {"type": "noun"}]
-
-    if provider == "chroma":
-        conn = ChromaDBConnection()
-        assert conn.connect()
-        assert conn.create_collection(collection_name, vector_size=2)
-        # Use standard signature: collection_name, documents, metadatas, ids, embeddings
-        success = conn.add_items(
-            collection_name,
-            documents=test_docs,
-            metadatas=test_metadata,
-            ids=test_ids,
-            embeddings=test_vectors
-        )
-        assert success
-        assert collection_name in conn.list_collections()
-        # Verify items inserted
-        info = conn.get_collection_info(collection_name)
-        assert info["count"] == 2
-        res = conn.get_all_items(collection_name, limit=10)
-        assert len(res["documents"]) == 2
-        assert conn.delete_collection(collection_name)
-        assert collection_name not in conn.list_collections()
-
-    elif provider == "qdrant":
-        db_path = str(tmp_path / "qdrant_test")
-        conn = QdrantConnection(path=db_path)
-        assert conn.connect()
-        assert conn.create_collection(collection_name, vector_size=2, distance="Cosine")
-        # Use standard signature
-        success = conn.add_items(
-            collection_name,
-            documents=test_docs,
-            metadatas=test_metadata,
-            ids=test_ids,
-            embeddings=test_vectors
-        )
-        assert success
-        assert collection_name in conn.list_collections()
-        # Verify items inserted
-        info = conn.get_collection_info(collection_name)
-        if info["count"] == 0:
-            pytest.skip("Qdrant local upsert not supported in this environment")
-        assert info["count"] == 2
-        res = conn.get_all_items(collection_name, limit=10)
-        assert len(res["documents"]) == 2
-        assert conn.delete_collection(collection_name)
-        assert collection_name not in conn.list_collections()
+import pytest
+from vector_inspector.core.connections.chroma_connection import ChromaDBConnection
+from vector_inspector.core.connections.qdrant_connection import QdrantConnection
+import uuid
+
+@pytest.mark.parametrize("provider", ["chroma", "qdrant"])
+def test_provider_integration(provider, tmp_path):
+    """Test provider connection using standard add_items signature."""
+    collection_name = f"test_collection_{uuid.uuid4().hex[:8]}"
+    test_ids = ["id1", "id2"]
+    test_vectors = [[0.1, 0.2], [0.3, 0.4]]
+    test_docs = ["hello", "world"]
+    test_metadata = [{"type": "greeting"}, {"type": "noun"}]
+
+    if provider == "chroma":
+        conn = ChromaDBConnection()
+        assert conn.connect()
+        assert conn.create_collection(collection_name, vector_size=2)
+        # Use standard signature: collection_name, documents, metadatas, ids, embeddings
+        success = conn.add_items(
+            collection_name,
+            documents=test_docs,
+            metadatas=test_metadata,
+            ids=test_ids,
+            embeddings=test_vectors
+        )
+        assert success
+        assert collection_name in conn.list_collections()
+        # Verify items inserted
+        info = conn.get_collection_info(collection_name)
+        assert info["count"] == 2
+        res = conn.get_all_items(collection_name, limit=10)
+        assert len(res["documents"]) == 2
+        assert conn.delete_collection(collection_name)
+        assert collection_name not in conn.list_collections()
+
+    elif provider == "qdrant":
+        db_path = str(tmp_path / "qdrant_test")
+        conn = QdrantConnection(path=db_path)
+        assert conn.connect()
+        assert conn.create_collection(collection_name, vector_size=2, distance="Cosine")
+        # Use standard signature
+        success = conn.add_items(
+            collection_name,
+            documents=test_docs,
+            metadatas=test_metadata,
+            ids=test_ids,
+            embeddings=test_vectors
+        )
+        assert success
+        assert collection_name in conn.list_collections()
+        # Verify items inserted
+        info = conn.get_collection_info(collection_name)
+        if info["count"] == 0:
+            pytest.skip("Qdrant local upsert not supported in this environment")
+        assert info["count"] == 2
+        res = conn.get_all_items(collection_name, limit=10)
+        assert len(res["documents"]) == 2
+        assert conn.delete_collection(collection_name)
+        assert collection_name not in conn.list_collections()

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/tests/test_filter_service.py

@@ -1,101 +1,101 @@
-import pytest
-from vector_inspector.services.filter_service import apply_client_side_filters
-
-def sample_data():
-    return {
-        "ids": [1, 2, 3],
-        "documents": ["The quick brown fox", "Jumps over the lazy dog", "Hello world!"],
-        "metadatas": [
-            {"category": "animal", "author": "A"},
-            {"category": "animal", "author": "B"},
-            {"category": "greeting", "author": "C"},
-        ],
-        "embeddings": [[0.1], [0.2], [0.3]],
-    }
-
-def test_no_filters_returns_all():
-    data = sample_data()
-    result = apply_client_side_filters(data, [])
-    assert result == {
-        "ids": [1, 2, 3],
-        "documents": ["The quick brown fox", "Jumps over the lazy dog", "Hello world!"],
-        "metadatas": [
-            {"category": "animal", "author": "A"},
-            {"category": "animal", "author": "B"},
-            {"category": "greeting", "author": "C"},
-        ],
-        "embeddings": [[0.1], [0.2], [0.3]],
-    }
-
-def test_contains_document():
-    data = sample_data()
-    filters = [{"field": "document", "op": "contains", "value": "fox"}]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == [1]
-    assert result["documents"] == ["The quick brown fox"]
-    assert result["metadatas"] == [{"category": "animal", "author": "A"}]
-    assert result["embeddings"] == [[0.1]]
-
-def test_not_contains_metadata():
-    data = sample_data()
-    filters = [{"field": "category", "op": "not_contains", "value": "animal"}]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == [3]
-    assert result["documents"] == ["Hello world!"]
-    assert result["metadatas"] == [{"category": "greeting", "author": "C"}]
-    assert result["embeddings"] == [[0.3]]
-
-def test_multiple_filters():
-    data = sample_data()
-    filters = [
-        {"field": "category", "op": "contains", "value": "animal"},
-        {"field": "author", "op": "contains", "value": "B"},
-    ]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == [2]
-    assert result["documents"] == ["Jumps over the lazy dog"]
-    assert result["metadatas"] == [{"category": "animal", "author": "B"}]
-    assert result["embeddings"] == [[0.2]]
-
-def test_empty_data():
-    result = apply_client_side_filters({}, [])
-    assert result == {}
-
-def test_missing_fields():
-    data = {"ids": [1], "documents": ["foo"]}
-    filters = [{"field": "category", "op": "contains", "value": "animal"}]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == []
-    assert result["documents"] == []
-    assert result["metadatas"] == []
-
-def test_case_sensitivity():
-    data = {"ids": [1], "documents": ["Hello World"], "metadatas": [{"author": "Alice"}]}
-    filters = [{"field": "document", "op": "contains", "value": "hello world"}]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == [1]
-
-def test_non_string_metadata_value():
-    data = {"ids": [1], "documents": ["foo"], "metadatas": [{"num": 123}]}
-    filters = [{"field": "num", "op": "contains", "value": "123"}]
-    result = apply_client_side_filters(data, filters)
-    assert result["ids"] == [1]
-
-def test_unknown_operator():
-    data = {"ids": [1], "documents": ["foo"], "metadatas": [{"author": "Bob"}]}
-    filters = [{"field": "author", "op": "unknown", "value": "Bob"}]
-    result = apply_client_side_filters(data, filters)
-    # Unknown op: should not filter out
-    assert result["ids"] == [1]
-
-def test_large_input():
-    data = {
-        "ids": list(range(1000)),
-        "documents": ["doc" + str(i) for i in range(1000)],
-        "metadatas": [{"author": "A" if i % 2 == 0 else "B"} for i in range(1000)],
-        "embeddings": [[i] for i in range(1000)],
-    }
-    filters = [{"field": "author", "op": "contains", "value": "A"}]
-    result = apply_client_side_filters(data, filters)
-    assert all(m["author"] == "A" for m in result["metadatas"])
-    assert len(result["ids"]) == 500
+import pytest
+from vector_inspector.services.filter_service import apply_client_side_filters
+
+def sample_data():
+    return {
+        "ids": [1, 2, 3],
+        "documents": ["The quick brown fox", "Jumps over the lazy dog", "Hello world!"],
+        "metadatas": [
+            {"category": "animal", "author": "A"},
+            {"category": "animal", "author": "B"},
+            {"category": "greeting", "author": "C"},
+        ],
+        "embeddings": [[0.1], [0.2], [0.3]],
+    }
+
+def test_no_filters_returns_all():
+    data = sample_data()
+    result = apply_client_side_filters(data, [])
+    assert result == {
+        "ids": [1, 2, 3],
+        "documents": ["The quick brown fox", "Jumps over the lazy dog", "Hello world!"],
+        "metadatas": [
+            {"category": "animal", "author": "A"},
+            {"category": "animal", "author": "B"},
+            {"category": "greeting", "author": "C"},
+        ],
+        "embeddings": [[0.1], [0.2], [0.3]],
+    }
+
+def test_contains_document():
+    data = sample_data()
+    filters = [{"field": "document", "op": "contains", "value": "fox"}]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == [1]
+    assert result["documents"] == ["The quick brown fox"]
+    assert result["metadatas"] == [{"category": "animal", "author": "A"}]
+    assert result["embeddings"] == [[0.1]]
+
+def test_not_contains_metadata():
+    data = sample_data()
+    filters = [{"field": "category", "op": "not_contains", "value": "animal"}]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == [3]
+    assert result["documents"] == ["Hello world!"]
+    assert result["metadatas"] == [{"category": "greeting", "author": "C"}]
+    assert result["embeddings"] == [[0.3]]
+
+def test_multiple_filters():
+    data = sample_data()
+    filters = [
+        {"field": "category", "op": "contains", "value": "animal"},
+        {"field": "author", "op": "contains", "value": "B"},
+    ]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == [2]
+    assert result["documents"] == ["Jumps over the lazy dog"]
+    assert result["metadatas"] == [{"category": "animal", "author": "B"}]
+    assert result["embeddings"] == [[0.2]]
+
+def test_empty_data():
+    result = apply_client_side_filters({}, [])
+    assert result == {}
+
+def test_missing_fields():
+    data = {"ids": [1], "documents": ["foo"]}
+    filters = [{"field": "category", "op": "contains", "value": "animal"}]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == []
+    assert result["documents"] == []
+    assert result["metadatas"] == []
+
+def test_case_sensitivity():
+    data = {"ids": [1], "documents": ["Hello World"], "metadatas": [{"author": "Alice"}]}
+    filters = [{"field": "document", "op": "contains", "value": "hello world"}]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == [1]
+
+def test_non_string_metadata_value():
+    data = {"ids": [1], "documents": ["foo"], "metadatas": [{"num": 123}]}
+    filters = [{"field": "num", "op": "contains", "value": "123"}]
+    result = apply_client_side_filters(data, filters)
+    assert result["ids"] == [1]
+
+def test_unknown_operator():
+    data = {"ids": [1], "documents": ["foo"], "metadatas": [{"author": "Bob"}]}
+    filters = [{"field": "author", "op": "unknown", "value": "Bob"}]
+    result = apply_client_side_filters(data, filters)
+    # Unknown op: should not filter out
+    assert result["ids"] == [1]
+
+def test_large_input():
+    data = {
+        "ids": list(range(1000)),
+        "documents": ["doc" + str(i) for i in range(1000)],
+        "metadatas": [{"author": "A" if i % 2 == 0 else "B"} for i in range(1000)],
+        "embeddings": [[i] for i in range(1000)],
+    }
+    filters = [{"field": "author", "op": "contains", "value": "A"}]
+    result = apply_client_side_filters(data, filters)
+    assert all(m["author"] == "A" for m in result["metadatas"])
+    assert len(result["ids"]) == 500

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/tests/test_settings_service.py

@@ -1,101 +1,101 @@
-import json
-from pathlib import Path
-
-import pytest
-
-from vector_inspector.services.settings_service import SettingsService
-
-
-@pytest.fixture()
-def temp_home(tmp_path):
-    # Monkeypatch Path.home() to point to a temporary directory for isolation
-    original_home = Path.home
-    Path.home = lambda: tmp_path  # type: ignore
-    try:
-        yield tmp_path
-    finally:
-        Path.home = original_home  # restore
-
-
-def test_last_connection_roundtrip(temp_home):
-    svc = SettingsService()
-    assert svc.get_last_connection() is None
-
-    config = {
-        "provider": "chromadb",
-        "connection_type": "persistent",
-        "path": "./data/chroma_db",
-    }
-    svc.save_last_connection(config)
-
-    # Create a new service to ensure it reads from disk
-    svc2 = SettingsService()
-    assert svc2.get_last_connection() == config
-
-    # Validate file exists with expected content
-    settings_file = temp_home / ".vector-viewer" / "settings.json"
-    assert settings_file.exists()
-    data = json.loads(settings_file.read_text(encoding="utf-8"))
-    assert data["last_connection"] == config
-
-
-def test_set_get_and_clear(temp_home):
-    svc = SettingsService()
-
-    assert svc.get("theme", "light") == "light"
-    svc.set("theme", "dark")
-    assert svc.get("theme") == "dark"
-
-    # Ensure persisted
-    settings_file = temp_home / ".vector-viewer" / "settings.json"
-    data = json.loads(settings_file.read_text(encoding="utf-8"))
-    assert data["theme"] == "dark"
-
-    # Clear and verify
-    svc.clear()
-    assert svc.get("theme") is None
-
-    # File should reflect cleared settings
-    data2 = json.loads(settings_file.read_text(encoding="utf-8"))
-    assert data2 == {}
-
-
-def test_missing_settings_file(temp_home):
-    svc = SettingsService()
-    # Remove file if exists
-    settings_file = temp_home / ".vector-viewer" / "settings.json"
-    if settings_file.exists():
-        settings_file.unlink()
-    svc._load_settings()
-    assert svc.settings == {}
-
-
-def test_invalid_json_file(temp_home):
-    settings_file = temp_home / ".vector-viewer" / "settings.json"
-    settings_file.parent.mkdir(parents=True, exist_ok=True)
-    settings_file.write_text("{ invalid json }", encoding="utf-8")
-    svc = SettingsService()
-    # Should fallback to empty settings
-    assert svc.settings == {}
-
-
-def test_overwrite_key(temp_home):
-    svc = SettingsService()
-    svc.set("theme", "light")
-    svc.set("theme", "dark")
-    assert svc.get("theme") == "dark"
-
-
-def test_unicode_and_large_value(temp_home):
-    svc = SettingsService()
-    unicode_val = "你好, мир, hello!"
-    large_val = "x" * 10000
-    svc.set("greeting", unicode_val)
-    svc.set("blob", large_val)
-    assert svc.get("greeting") == unicode_val
-    assert svc.get("blob") == large_val
-    # Validate persistence
-    settings_file = temp_home / ".vector-viewer" / "settings.json"
-    data = json.loads(settings_file.read_text(encoding="utf-8"))
-    assert data["greeting"] == unicode_val
-    assert data["blob"] == large_val
+import json
+from pathlib import Path
+
+import pytest
+
+from vector_inspector.services.settings_service import SettingsService
+
+
+@pytest.fixture()
+def temp_home(tmp_path):
+    # Monkeypatch Path.home() to point to a temporary directory for isolation
+    original_home = Path.home
+    Path.home = lambda: tmp_path  # type: ignore
+    try:
+        yield tmp_path
+    finally:
+        Path.home = original_home  # restore
+
+
+def test_last_connection_roundtrip(temp_home):
+    svc = SettingsService()
+    assert svc.get_last_connection() is None
+
+    config = {
+        "provider": "chromadb",
+        "connection_type": "persistent",
+        "path": "./data/chroma_db",
+    }
+    svc.save_last_connection(config)
+
+    # Create a new service to ensure it reads from disk
+    svc2 = SettingsService()
+    assert svc2.get_last_connection() == config
+
+    # Validate file exists with expected content
+    settings_file = temp_home / ".vector-viewer" / "settings.json"
+    assert settings_file.exists()
+    data = json.loads(settings_file.read_text(encoding="utf-8"))
+    assert data["last_connection"] == config
+
+
+def test_set_get_and_clear(temp_home):
+    svc = SettingsService()
+
+    assert svc.get("theme", "light") == "light"
+    svc.set("theme", "dark")
+    assert svc.get("theme") == "dark"
+
+    # Ensure persisted
+    settings_file = temp_home / ".vector-viewer" / "settings.json"
+    data = json.loads(settings_file.read_text(encoding="utf-8"))
+    assert data["theme"] == "dark"
+
+    # Clear and verify
+    svc.clear()
+    assert svc.get("theme") is None
+
+    # File should reflect cleared settings
+    data2 = json.loads(settings_file.read_text(encoding="utf-8"))
+    assert data2 == {}
+
+
+def test_missing_settings_file(temp_home):
+    svc = SettingsService()
+    # Remove file if exists
+    settings_file = temp_home / ".vector-viewer" / "settings.json"
+    if settings_file.exists():
+        settings_file.unlink()
+    svc._load_settings()
+    assert svc.settings == {}
+
+
+def test_invalid_json_file(temp_home):
+    settings_file = temp_home / ".vector-viewer" / "settings.json"
+    settings_file.parent.mkdir(parents=True, exist_ok=True)
+    settings_file.write_text("{ invalid json }", encoding="utf-8")
+    svc = SettingsService()
+    # Should fallback to empty settings
+    assert svc.settings == {}
+
+
+def test_overwrite_key(temp_home):
+    svc = SettingsService()
+    svc.set("theme", "light")
+    svc.set("theme", "dark")
+    assert svc.get("theme") == "dark"
+
+
+def test_unicode_and_large_value(temp_home):
+    svc = SettingsService()
+    unicode_val = "你好, мир, hello!"
+    large_val = "x" * 10000
+    svc.set("greeting", unicode_val)
+    svc.set("blob", large_val)
+    assert svc.get("greeting") == unicode_val
+    assert svc.get("blob") == large_val
+    # Validate persistence
+    settings_file = temp_home / ".vector-viewer" / "settings.json"
+    data = json.loads(settings_file.read_text(encoding="utf-8"))
+    assert data["greeting"] == unicode_val
+    assert data["blob"] == large_val

{vector_inspector-0.2.1 → vector_inspector-0.2.2}/tests/vector_inspector.py

@@ -1,35 +1,35 @@
-import os
-import sys
-import tempfile
-from pathlib import Path
-
-import pytest
-
-
-def run_tests():
-    project_path = Path(__file__).parent.parent
-    os.chdir(project_path)
-
-    # Determine any args to pass to pytest. If there aren't any,
-    # default to running the whole test suite.
-    args = sys.argv[1:]
-    if len(args) == 0:
-        args = ["tests"]
-
-    returncode = pytest.main(
-        [
-            # Turn up verbosity
-            "-vv",
-            # Disable color
-            "--color=no",
-            # Overwrite the cache directory to somewhere writable
-            "-o",
-            f"cache_dir={tempfile.gettempdir()}/.pytest_cache",
-        ] + args
-    )
-
-    print(f">>>>>>>>>> EXIT {returncode} <<<<<<<<<<")
-
-
-if __name__ == "__main__":
-    run_tests()
+import os
+import sys
+import tempfile
+from pathlib import Path
+
+import pytest
+
+
+def run_tests():
+    project_path = Path(__file__).parent.parent
+    os.chdir(project_path)
+
+    # Determine any args to pass to pytest. If there aren't any,
+    # default to running the whole test suite.
+    args = sys.argv[1:]
+    if len(args) == 0:
+        args = ["tests"]
+
+    returncode = pytest.main(
+        [
+            # Turn up verbosity
+            "-vv",
+            # Disable color
+            "--color=no",
+            # Overwrite the cache directory to somewhere writable
+            "-o",
+            f"cache_dir={tempfile.gettempdir()}/.pytest_cache",
+        ] + args
+    )
+
+    print(f">>>>>>>>>> EXIT {returncode} <<<<<<<<<<")
+
+
+if __name__ == "__main__":
+    run_tests()