vector-inspector 0.3.1__tar.gz → 0.3.3__tar.gz

{vector_inspector-0.3.1 → vector_inspector-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-inspector
-Version: 0.3.1
+Version: 0.3.3
 Summary: A comprehensive desktop application for visualizing, querying, and managing vector database data
 Author-Email: Anthony Dawson <anthonypdawson+github@gmail.com>
 License: MIT
@@ -24,6 +24,8 @@ Requires-Dist: pyarrow>=14.0.0
 Requires-Dist: pinecone>=8.0.0
 Requires-Dist: keyring>=25.7.0
 Requires-Dist: hf-xet>=1.2.0
+Requires-Dist: pymilvus>=2.6.6
+Requires-Dist: lancedb>=0.27.0
 Description-Content-Type: text/markdown
 
 
@@ -31,7 +33,10 @@ Description-Content-Type: text/markdown
 
 > **Disclaimer:** This tool is currently under active development and is **not production ready**. Not all features have been thoroughly tested and code is released frequently. Use with caution in critical or production environments.
 
-![PyPI](https://img.shields.io/pypi/v/vector-inspector)
+[![CI](https://github.com/anthonypdawson/vector-inspector/actions/workflows/ci-tests.yml/badge.svg?branch=master)](https://github.com/anthonypdawson/vector-inspector/actions/workflows/ci-tests.yml)
+[![Publish](https://github.com/anthonypdawson/vector-inspector/actions/workflows/publish.yml/badge.svg?branch=master)](https://github.com/anthonypdawson/vector-inspector/actions/workflows/publish.yml)
+
+[![PyPI Version](https://img.shields.io/pypi/v/vector-inspector.svg?cacheSeconds=300)](https://pypi.org/project/vector-inspector/)
 [![PyPI Downloads](https://static.pepy.tech/personalized-badge/vector-inspector?period=total&units=INTERNATIONAL_SYSTEM&left_color=BLACK&right_color=GREEN&left_text=downloads)](https://pepy.tech/projects/vector-inspector)
 
 
@@ -62,8 +67,10 @@ Vector Inspector bridges the gap between vector databases and user-friendly data
 - Connect to vector databases:
   - ChromaDB (persistent local storage)
   - Qdrant (remote server or embedded local)
+  - Pinecone (cloud-hosted)
 - Unified interface regardless of backend provider
 - Automatically saves last connection configuration
+- Secure API key storage for cloud providers
 
 ### 2. **Data Visualization**
 - **Metadata Explorer**: Browse and filter vector entries by metadata fields

{vector_inspector-0.3.1 → vector_inspector-0.3.3}/README.md

@@ -3,7 +3,10 @@
 
 > **Disclaimer:** This tool is currently under active development and is **not production ready**. Not all features have been thoroughly tested and code is released frequently. Use with caution in critical or production environments.
 
-![PyPI](https://img.shields.io/pypi/v/vector-inspector)
+[![CI](https://github.com/anthonypdawson/vector-inspector/actions/workflows/ci-tests.yml/badge.svg?branch=master)](https://github.com/anthonypdawson/vector-inspector/actions/workflows/ci-tests.yml)
+[![Publish](https://github.com/anthonypdawson/vector-inspector/actions/workflows/publish.yml/badge.svg?branch=master)](https://github.com/anthonypdawson/vector-inspector/actions/workflows/publish.yml)
+
+[![PyPI Version](https://img.shields.io/pypi/v/vector-inspector.svg?cacheSeconds=300)](https://pypi.org/project/vector-inspector/)
 [![PyPI Downloads](https://static.pepy.tech/personalized-badge/vector-inspector?period=total&units=INTERNATIONAL_SYSTEM&left_color=BLACK&right_color=GREEN&left_text=downloads)](https://pepy.tech/projects/vector-inspector)
 
 
@@ -34,8 +37,10 @@ Vector Inspector bridges the gap between vector databases and user-friendly data
 - Connect to vector databases:
   - ChromaDB (persistent local storage)
   - Qdrant (remote server or embedded local)
+  - Pinecone (cloud-hosted)
 - Unified interface regardless of backend provider
 - Automatically saves last connection configuration
+- Secure API key storage for cloud providers
 
 ### 2. **Data Visualization**
 - **Metadata Explorer**: Browse and filter vector entries by metadata fields

{vector_inspector-0.3.1 → vector_inspector-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vector-inspector"
-version = "0.3.1"
+version = "0.3.3"
 description = "A comprehensive desktop application for visualizing, querying, and managing vector database data"
 authors = [
     { name = "Anthony Dawson", email = "anthonypdawson+github@gmail.com" },
@@ -21,6 +21,8 @@ dependencies = [
     "pinecone>=8.0.0",
     "keyring>=25.7.0",
     "hf-xet>=1.2.0",
+    "pymilvus>=2.6.6",
+    "lancedb>=0.27.0",
 ]
 requires-python = ">=3.10,<3.13"
 readme = "README.md"
@@ -37,6 +39,65 @@ Documentation = "https://github.com/anthonypdawson/vector-inspector#readme"
 [project.scripts]
 vector-inspector = "vector_inspector.main:main"
 
+[build-system]
+requires = [
+    "pdm-backend",
+]
+build-backend = "pdm.backend"
+
+[dependency-groups]
+dev = [
+    "pytest>=7.4.0",
+    "pytest-qt>=4.2.0",
+    "black>=23.12.0",
+    "ruff>=0.14.14",
+    "mypy>=1.19.1",
+    "ipython>=8.18.0",
+]
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+extend-exclude = [
+    "build",
+    "dist",
+    ".venv",
+]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+line-ending = "auto"
+skip-magic-trailing-comma = false
+
+[tool.ruff.lint]
+select = [
+    "E",
+    "F",
+    "W",
+    "I",
+    "UP",
+    "B",
+    "A",
+    "C4",
+    "DTZ",
+    "T20",
+    "RET",
+    "SIM",
+    "ARG",
+    "RUF",
+]
+ignore = [
+    "E501",
+    "I001",
+]
+
+[tool.ruff.lint.isort]
+combine-as-imports = true
+known-first-party = [
+    "vector_inspector",
+]
+
 [tool.pdm]
 distribution = true
 
@@ -48,19 +109,3 @@ cmd = "python -m vector_inspector.main"
 
 [tool.pdm.scripts.vector-inspector.env]
 PYTHONPATH = "src"
-
-[tool.pdm.dev-dependencies]
-dev = [
-    "pytest>=7.4.0",
-    "pytest-qt>=4.2.0",
-    "black>=23.12.0",
-    "ruff>=0.1.0",
-    "mypy>=1.7.0",
-    "ipython>=8.18.0",
-]
-
-[build-system]
-requires = [
-    "pdm-backend",
-]
-build-backend = "pdm.backend"

{vector_inspector-0.3.1 → vector_inspector-0.3.3}/src/vector_inspector/core/connection_manager.py

@@ -6,10 +6,12 @@ from enum import Enum
 from PySide6.QtCore import QObject, Signal
 
 from .connections.base_connection import VectorDBConnection
+from vector_inspector.core.logging import log_error
 
 
 class ConnectionState(Enum):
     """Possible connection states."""
+
     DISCONNECTED = "disconnected"
     CONNECTING = "connecting"
     CONNECTED = "connected"
@@ -18,18 +20,18 @@ class ConnectionState(Enum):
 
 class ConnectionInstance:
     """Represents a single active connection with its state and context."""
-    
+
     def __init__(
         self,
         connection_id: str,
         name: str,
         provider: str,
         connection: VectorDBConnection,
-        config: Dict[str, Any]
+        config: Dict[str, Any],
     ):
         """
         Initialize a connection instance.
-        
+
         Args:
             connection_id: Unique connection identifier
             name: User-friendly connection name
@@ -46,11 +48,11 @@ class ConnectionInstance:
         self.active_collection: Optional[str] = None
         self.collections: List[str] = []
         self.error_message: Optional[str] = None
-        
+
     def get_display_name(self) -> str:
         """Get a display-friendly connection name."""
         return f"{self.name} ({self.provider})"
-    
+
     def get_breadcrumb(self) -> str:
         """Get breadcrumb showing connection > collection."""
         if self.active_collection:
@@ -60,7 +62,7 @@ class ConnectionInstance:
 
 class ConnectionManager(QObject):
     """Manages multiple vector database connections and saved profiles.
-    
+
     Signals:
         connection_opened: Emitted when a new connection is opened (connection_id)
         connection_closed: Emitted when a connection is closed (connection_id)
@@ -69,7 +71,7 @@ class ConnectionManager(QObject):
         active_collection_changed: Emitted when active collection changes (connection_id, collection_name or None)
         collections_updated: Emitted when collections list is updated (connection_id, collections)
     """
-    
+
     # Signals
     connection_opened = Signal(str)  # connection_id
     connection_closed = Signal(str)  # connection_id
@@ -77,116 +79,119 @@ class ConnectionManager(QObject):
     active_connection_changed = Signal(object)  # connection_id or None
     active_collection_changed = Signal(str, object)  # connection_id, collection_name or None
     collections_updated = Signal(str, list)  # connection_id, collections
-    
+
     MAX_CONNECTIONS = 10  # Limit to prevent resource exhaustion
-    
+
     def __init__(self):
         """Initialize the connection manager."""
         super().__init__()
         self._connections: Dict[str, ConnectionInstance] = {}
         self._active_connection_id: Optional[str] = None
-        
+
     def create_connection(
         self,
         name: str,
         provider: str,
         connection: VectorDBConnection,
-        config: Dict[str, Any]
+        config: Dict[str, Any],
+        connection_id: str = None,
     ) -> str:
         """
         Create a new connection instance (not yet connected).
-        
+
         Args:
             name: User-friendly connection name
             provider: Provider type
             connection: The connection object
             config: Connection configuration
-            
+            connection_id: Optional. Use this ID instead of generating a new one (for profiles).
+
         Returns:
             The connection ID
-            
+
         Raises:
             RuntimeError: If maximum connections limit reached
         """
         if len(self._connections) >= self.MAX_CONNECTIONS:
             raise RuntimeError(f"Maximum number of connections ({self.MAX_CONNECTIONS}) reached")
-        
-        connection_id = str(uuid.uuid4())
+
+        if connection_id is None:
+            connection_id = str(uuid.uuid4())
         instance = ConnectionInstance(connection_id, name, provider, connection, config)
         self._connections[connection_id] = instance
-        
+
         # Set as active if it's the first connection
         if len(self._connections) == 1:
             self._active_connection_id = connection_id
             self.active_connection_changed.emit(connection_id)
-        
+
         # Don't emit connection_opened yet - wait until actually connected
         return connection_id
-    
+
     def mark_connection_opened(self, connection_id: str):
         """
         Mark a connection as opened (after successful connection).
-        
+
         Args:
             connection_id: ID of connection that opened
         """
         if connection_id in self._connections:
             self.connection_opened.emit(connection_id)
-    
+
     def get_connection(self, connection_id: str) -> Optional[ConnectionInstance]:
         """Get a connection instance by ID."""
         return self._connections.get(connection_id)
-    
+
     def get_active_connection(self) -> Optional[ConnectionInstance]:
         """Get the currently active connection instance."""
         if self._active_connection_id:
             return self._connections.get(self._active_connection_id)
         return None
-    
+
     def get_active_connection_id(self) -> Optional[str]:
         """Get the currently active connection ID."""
         return self._active_connection_id
-    
+
     def set_active_connection(self, connection_id: str) -> bool:
         """
         Set the active connection.
-        
+
         Args:
             connection_id: ID of connection to make active
-            
+
         Returns:
             True if successful, False if connection not found
         """
         if connection_id not in self._connections:
             return False
-        
+
         self._active_connection_id = connection_id
         self.active_connection_changed.emit(connection_id)
         return True
-    
+
     def close_connection(self, connection_id: str) -> bool:
         """
         Close and remove a connection.
-        
+
         Args:
             connection_id: ID of connection to close
-            
+
         Returns:
             True if successful, False if connection not found
         """
         instance = self._connections.get(connection_id)
         if not instance:
             return False
-        
+
         # Disconnect the connection
         try:
             instance.connection.disconnect()
         except Exception as e:
-            print(f"Error disconnecting: {e}")
-        
+            log_error("Error disconnecting: %s", e)
+
         # Remove from connections dict
         del self._connections[connection_id]
-        
+
         # If this was the active connection, set a new one or None
         if self._active_connection_id == connection_id:
             if self._connections:
@@ -196,14 +201,16 @@ class ConnectionManager(QObject):
             else:
                 self._active_connection_id = None
                 self.active_connection_changed.emit(None)
-        
+
         self.connection_closed.emit(connection_id)
         return True
-    
-    def update_connection_state(self, connection_id: str, state: ConnectionState, error: Optional[str] = None):
+
+    def update_connection_state(
+        self, connection_id: str, state: ConnectionState, error: Optional[str] = None
+    ):
         """
         Update the state of a connection.
-        
+
         Args:
             connection_id: ID of connection
             state: New connection state
@@ -217,11 +224,11 @@ class ConnectionManager(QObject):
             else:
                 instance.error_message = None
             self.connection_state_changed.emit(connection_id, state)
-    
+
     def update_collections(self, connection_id: str, collections: List[str]):
         """
         Update the collections list for a connection.
-        
+
         Args:
             connection_id: ID of connection
             collections: List of collection names
@@ -230,11 +237,11 @@ class ConnectionManager(QObject):
         if instance:
             instance.collections = collections
             self.collections_updated.emit(connection_id, collections)
-    
+
     def set_active_collection(self, connection_id: str, collection_name: Optional[str]):
         """
         Set the active collection for a connection.
-        
+
         Args:
             connection_id: ID of connection
             collection_name: Name of collection to make active, or None
@@ -243,29 +250,29 @@ class ConnectionManager(QObject):
         if instance:
             instance.active_collection = collection_name
             self.active_collection_changed.emit(connection_id, collection_name)
-    
+
     def get_all_connections(self) -> List[ConnectionInstance]:
         """Get list of all connection instances."""
         return list(self._connections.values())
-    
+
     def get_connection_count(self) -> int:
         """Get the number of active connections."""
         return len(self._connections)
-    
+
     def close_all_connections(self):
         """Close all connections. Typically called on application exit."""
         connection_ids = list(self._connections.keys())
         for conn_id in connection_ids:
             self.close_connection(conn_id)
-    
+
     def rename_connection(self, connection_id: str, new_name: str) -> bool:
         """
         Rename a connection.
-        
+
         Args:
             connection_id: ID of connection
             new_name: New name for the connection
-            
+
         Returns:
             True if successful, False if connection not found
         """
@@ -274,4 +281,3 @@ class ConnectionManager(QObject):
             instance.name = new_name
             return True
         return False
-