vector-inspector 0.2.6__py3-none-any.whl → 0.3.1__py3-none-any.whl

vector_inspector/ui/views/search_view.py

@@ -12,6 +12,7 @@ from vector_inspector.core.connections.base_connection import VectorDBConnection
 from vector_inspector.ui.components.filter_builder import FilterBuilder
 from vector_inspector.ui.components.loading_dialog import LoadingDialog
 from vector_inspector.services.filter_service import apply_client_side_filters
+from vector_inspector.core.cache_manager import get_cache_manager, CacheEntry
 
 
 class SearchView(QWidget):
@@ -21,8 +22,10 @@ class SearchView(QWidget):
         super().__init__(parent)
         self.connection = connection
         self.current_collection: str = ""
+        self.current_database: str = ""
         self.search_results: Optional[Dict[str, Any]] = None
         self.loading_dialog = LoadingDialog("Searching...", self)
+        self.cache_manager = get_cache_manager()
         
         self._setup_ui()
         
@@ -112,12 +115,30 @@ class SearchView(QWidget):
         
         layout.addWidget(splitter)
         
-    def set_collection(self, collection_name: str):
+    def set_collection(self, collection_name: str, database_name: str = ""):
         """Set the current collection to search."""
         self.current_collection = collection_name
+        # Always update database_name if provided (even if empty string on first call)
+        if database_name:  # Only update if non-empty
+            self.current_database = database_name
+        
+        print(f"[SearchView] Setting collection: db='{self.current_database}', coll='{collection_name}'")
+        
+        # Check cache first
+        cached = self.cache_manager.get(self.current_database, self.current_collection)
+        if cached:
+            print(f"[SearchView] ✓ Cache HIT! Restoring search state.")
+            # Restore search query and results from cache
+            if cached.search_query:
+                self.query_input.setPlainText(cached.search_query)
+            if cached.search_results:
+                self.search_results = cached.search_results
+                self._display_results(cached.search_results)
+                return
+        
+        print(f"[SearchView] ✗ Cache MISS or no cached search.")
+        # Not in cache, clear form
         self.search_results = None
-        
-        # Clear search form inputs
         self.query_input.clear()
         self.results_table.setRowCount(0)
         self.results_status.setText(f"Collection: {collection_name}")
@@ -181,7 +202,7 @@ class SearchView(QWidget):
         QApplication.processEvents()
         
         try:
-            # Perform query
+            # Always pass query_texts; provider handles embedding if needed
             results = self.connection.query_collection(
                 self.current_collection,
                 query_texts=[query_text],
@@ -196,6 +217,12 @@ class SearchView(QWidget):
             self.results_table.setRowCount(0)
             return
         
+        # Check if results have the expected structure
+        if not results.get("ids") or not isinstance(results["ids"], list) or len(results["ids"]) == 0:
+            self.results_status.setText("No results found or query failed")
+            self.results_table.setRowCount(0)
+            return
+        
         # Apply client-side filters if any
         if client_filters and results:
             # Restructure results for filtering
@@ -219,6 +246,19 @@ class SearchView(QWidget):
         self.search_results = results
         self._display_results(results)
         
+        # Save to cache
+        if self.current_database and self.current_collection:
+            self.cache_manager.update(
+                self.current_database,
+                self.current_collection,
+                search_query=query_text,
+                search_results=results,
+                user_inputs={
+                    'n_results': n_results,
+                    'filters': self.filter_builder.to_dict() if hasattr(self.filter_builder, 'to_dict') else {}
+                }
+            )
+        
     def _display_results(self, results: Dict[str, Any]):
         """Display search results in table."""
         ids = results.get("ids", [[]])[0]

vector_inspector/ui/views/visualization_view.py

@@ -13,6 +13,7 @@ import numpy as np
 
 from vector_inspector.core.connections.base_connection import VectorDBConnection
 from vector_inspector.services.visualization_service import VisualizationService
+from vector_inspector.ui.components.loading_dialog import LoadingDialog
 
 
 class VisualizationThread(QThread):
@@ -106,6 +107,9 @@ class VisualizationView(QWidget):
         self.status_label.setMaximumHeight(30)
         layout.addWidget(self.status_label)
         
+        # Loading dialog for data fetch and reduction
+        self.loading_dialog = LoadingDialog("Loading visualization...", self)
+        
     def set_collection(self, collection_name: str):
         """Set the current collection to visualize."""
         self.current_collection = collection_name
@@ -119,12 +123,17 @@ class VisualizationView(QWidget):
             QMessageBox.warning(self, "No Collection", "Please select a collection first.")
             return
             
-        # Load data with embeddings
+        # Load data with embeddings (show loading immediately)
+        self.loading_dialog.show_loading("Loading data for visualization...")
+        QApplication.processEvents()
         sample_size = self.sample_spin.value()
-        data = self.connection.get_all_items(
-            self.current_collection,
-            limit=sample_size
-        )
+        try:
+            data = self.connection.get_all_items(
+                self.current_collection,
+                limit=sample_size
+            )
+        finally:
+            self.loading_dialog.hide_loading()
         
         if data is None or not data or "embeddings" not in data or data["embeddings"] is None or len(data["embeddings"]) == 0:
             QMessageBox.warning(
@@ -152,10 +161,14 @@ class VisualizationView(QWidget):
         )
         self.visualization_thread.finished.connect(self._on_reduction_finished)
         self.visualization_thread.error.connect(self._on_reduction_error)
+        # Show loading during reduction
+        self.loading_dialog.show_loading("Reducing dimensions...")
+        QApplication.processEvents()
         self.visualization_thread.start()
         
     def _on_reduction_finished(self, reduced_data: Any):
         """Handle dimensionality reduction completion."""
+        self.loading_dialog.hide_loading()
         self.reduced_data = reduced_data
         self._create_plot()
         self.generate_button.setEnabled(True)
@@ -163,6 +176,7 @@ class VisualizationView(QWidget):
         
     def _on_reduction_error(self, error_msg: str):
         """Handle dimensionality reduction error."""
+        self.loading_dialog.hide_loading()
         print(f"Error: Visualization failed: {error_msg}")
         QMessageBox.warning(self, "Error", f"Visualization failed: {error_msg}")
         self.generate_button.setEnabled(True)

{vector_inspector-0.2.6.dist-info → vector_inspector-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-inspector
-Version: 0.2.6
+Version: 0.3.1
 Summary: A comprehensive desktop application for visualizing, querying, and managing vector database data
 Author-Email: Anthony Dawson <anthonypdawson+github@gmail.com>
 License: MIT
@@ -22,6 +22,8 @@ Requires-Dist: sentence-transformers>=2.2.0
 Requires-Dist: fastembed>=0.7.4
 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
 Description-Content-Type: text/markdown
 
 

vector_inspector-0.3.1.dist-info/RECORD

@@ -0,0 +1,55 @@
+vector_inspector-0.3.1.dist-info/METADATA,sha256=VXsx5HkbOhkHfsjoZV29MYBh6s3hsdal-3l4ZAppN88,9684
+vector_inspector-0.3.1.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
+vector_inspector-0.3.1.dist-info/entry_points.txt,sha256=u96envMI2NFImZUJDFutiiWl7ZoHrrev9joAgtyvTxo,80
+vector_inspector/__init__.py,sha256=Q8XbXn98o0eliQWPePhy-aGUz2KNnVg7bQq-sBPl7zQ,119
+vector_inspector/__main__.py,sha256=Vdhw8YA1K3wPMlbJQYL5WqvRzAKVeZ16mZQFO9VRmCo,62
+vector_inspector/config/__init__.py,sha256=vHkVsXSUdInsfzWSOLPZzaaELa3SGenAgfpY5EYbsYA,95
+vector_inspector/config/known_embedding_models.json,sha256=tnTWI3OvdL2z0YreL_iBzbceIXR69NDj-0tBcEV6NVM,15701
+vector_inspector/core/__init__.py,sha256=hjOqiJwF1P0rXjiOKhK4qDTvBY7G3m4kq8taH-gKrFM,57
+vector_inspector/core/cache_manager.py,sha256=cHdbIYR-eS9vLLTqvq4Xejyi5Z7Fm9DqMhb_PMZjnjY,5695
+vector_inspector/core/connection_manager.py,sha256=WwiVedHWTfqIuJKV7D52bYEupmpnnSPKcgG1odJCJjs,10003
+vector_inspector/core/connections/__init__.py,sha256=lDZ-Qv-CbBvVcSlT8K2824zojovEIKhykHVSLARHZWs,345
+vector_inspector/core/connections/base_connection.py,sha256=dQTNmYGCt3-rGjTkAj0vuTEAsFnp8i_FcPmlz6CYdV4,8909
+vector_inspector/core/connections/chroma_connection.py,sha256=wN1apwGLStjOd113Cd1jC--pxJVMHeX7CrtM1Ba0OL4,19753
+vector_inspector/core/connections/pinecone_connection.py,sha256=zdBK2yetVDwxLZV0PxJfnKbZHAWLvb3apsO8wBdVlbE,26942
+vector_inspector/core/connections/qdrant_connection.py,sha256=m0EFgSbWRVh3uv_JkbB0WEotu20G967TrHvwJrtedO4,31861
+vector_inspector/core/connections/template_connection.py,sha256=wsJiE4ma3cLUXk2eW5rnLMS5wG8JTekgEn46lHHQNoc,10642
+vector_inspector/core/embedding_providers/__init__.py,sha256=anorFsmLVrdK6bHP8YcPZF9DQTIJ9CEnrmJRnqOzWdc,440
+vector_inspector/core/embedding_providers/base_provider.py,sha256=SqQFoE6lmmxCdns6ce36_x5Ucivvo5jSp8HEbzs7kLw,3933
+vector_inspector/core/embedding_providers/clip_provider.py,sha256=vvFqoQZNB0PsFJxvDsoDf8-L_z61CrYROFT9IGt7nIE,9088
+vector_inspector/core/embedding_providers/provider_factory.py,sha256=aMoF9NEslTg_4f3QlIr-v0fXg0rb1zNfmavnPVczNy0,5973
+vector_inspector/core/embedding_providers/sentence_transformer_provider.py,sha256=3YGUt7-6VrfbWKMuo3vDzx9SJXi2CjbpdOqdtE5CymM,6913
+vector_inspector/core/embedding_utils.py,sha256=TcSZyVRZmmV9LlYCq8BJMZXTWfkGAYLFtcXXv-M15r8,5650
+vector_inspector/core/model_registry.py,sha256=8zr-bhMqxwuHZ3MeTrG6NItUHfmcFoypGuP5o1PYXvU,6194
+vector_inspector/main.py,sha256=puu1Fur298j6H8fG3_wF85RMhi4tjLZ0Are16kloMqM,479
+vector_inspector/services/__init__.py,sha256=QLgH7oybjHuEYDFNiBgmJxvSpgAzHEuBEPXa3SKJb_I,67
+vector_inspector/services/backup_restore_service.py,sha256=uqbVdCAb78HWZPDu6T0NONy4w7BFPhYsmy7omChdxkg,11582
+vector_inspector/services/credential_service.py,sha256=pdpglvLABnYQRvklgK38iPyG91IZ9Nm8WQFT9V0LofM,4354
+vector_inspector/services/filter_service.py,sha256=xDrMxNWsYzRcR1n0Fd-yp6Fo-4aLbVIDkhj2GKmrw5o,2370
+vector_inspector/services/import_export_service.py,sha256=OPCrBXBewCznu5o8wFGvduU0jGZAcBvp_Fpv15kdoJ4,10712
+vector_inspector/services/profile_service.py,sha256=Uhl9urxeSRI0p-mfaYgBrMI99byNIxJSodcXOgD_ybw,13408
+vector_inspector/services/settings_service.py,sha256=Yw5wvqpf7hjwVjvZRXxBgIKH8tj8sKfXAavoeOhtvsU,7648
+vector_inspector/services/visualization_service.py,sha256=mHI4qxT-V4R1kcOhE508vFTZ0HcmQICHvJ7dIoRracQ,4373
+vector_inspector/ui/__init__.py,sha256=262ZiXO6Luk8vZnhCIoYxOtGiny0bXK-BTKjxUNBx-w,43
+vector_inspector/ui/components/__init__.py,sha256=S-GWU1P820dJ6mHmeeBEy-CGF9fjpBeNf8vrbhRlFMk,30
+vector_inspector/ui/components/backup_restore_dialog.py,sha256=CrZ2u8vXzggv3aBkYR4FulpY74oZWMLW5BHU4dMiWug,13073
+vector_inspector/ui/components/connection_manager_panel.py,sha256=O0V7kE6immkWzO7bGYehCLqyz62va_421aL0EncuVQ8,13279
+vector_inspector/ui/components/filter_builder.py,sha256=NSR_hp-rzUZVAca6dIJhTxZA3igOKFM1g-YXiYPhFos,13360
+vector_inspector/ui/components/item_dialog.py,sha256=VMwehEjQ6xrdxWygR9J-hHsLfzOVb_E3ePUGYO_c7XA,3951
+vector_inspector/ui/components/loading_dialog.py,sha256=YEKYGU-R-Zz4CjXSArJtkNxgTy4O9hI5Bbt6qlIzD8U,1018
+vector_inspector/ui/components/profile_manager_panel.py,sha256=3CzJrY2pjzV8LAgJb0bLOKzK4Buy8v5P8rFqJfrXdRs,20966
+vector_inspector/ui/dialogs/__init__.py,sha256=xtT77L91PFfm3zHYRENHkWHJaKPm1htuUzRXAF53P8w,211
+vector_inspector/ui/dialogs/cross_db_migration.py,sha256=KGtlewzA83Xwkl0J1TUYUhiccYIXbRuNxHO59kNP_tI,15783
+vector_inspector/ui/dialogs/embedding_config_dialog.py,sha256=Pm6UBHmkcof3x1xXLiGLiMP6jLBcJvsPfY6df5jTtK4,13037
+vector_inspector/ui/dialogs/provider_type_dialog.py,sha256=W_FAJuvicwBUJJ7PyvKow9lc8_a5pnE3RIAsh-DVndQ,6809
+vector_inspector/ui/main_window.py,sha256=-l-S8GSAebIfu8YrGNUEetUlub5XB3jR-symaap8450,26207
+vector_inspector/ui/views/__init__.py,sha256=FeMtVzSbVFBMjdwLQSQqD0FRW4ieJ4ZKXtTBci2e_bw,30
+vector_inspector/ui/views/collection_browser.py,sha256=oG9_YGPoVuMs-f_zSd4EcITmEU9caxvwuubsFUrNf-c,3991
+vector_inspector/ui/views/connection_view.py,sha256=bfVRvcS2gLVC6ESux3RcNhYEwFhxp6NpByqJ2bfetvI,20204
+vector_inspector/ui/views/info_panel.py,sha256=fgxIjs5AlnyE0Ni5ZXWfrB3nb6O6w1rAOnMnW-akiBs,23848
+vector_inspector/ui/views/metadata_view.py,sha256=dIe7aH47xd6TBK2g-WogR2Ts0enHGeSAEUlu48sdTF8,30535
+vector_inspector/ui/views/search_view.py,sha256=Yrg7akY1vDgpWny-3Us6r1LQhd0QHB7fUvCZj1VpnA0,12522
+vector_inspector/ui/views/visualization_view.py,sha256=Z-oDwKSAxYx9yhln6pM00j9pWokjz5eYZMJ-Japg938,9841
+vector_inspector/utils/__init__.py,sha256=jhHBQC8C8bfhNlf6CAt07ejjStp_YAyleaYr2dm0Dk0,38
+vector_inspector/utils/lazy_imports.py,sha256=2XZ3ZnwTvZ5vvrh36nJ_TUjwwkgjoAED6i6P9yctvt0,1211
+vector_inspector-0.3.1.dist-info/RECORD,,

vector_inspector-0.2.6.dist-info/RECORD

@@ -1,35 +0,0 @@
-vector_inspector-0.2.6.dist-info/METADATA,sha256=TZTo9zaLOPYqQkGQcPNoPgTXflQYhVh7BNbsnBGrhhw,9624
-vector_inspector-0.2.6.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
-vector_inspector-0.2.6.dist-info/entry_points.txt,sha256=u96envMI2NFImZUJDFutiiWl7ZoHrrev9joAgtyvTxo,80
-vector_inspector/__init__.py,sha256=Q8XbXn98o0eliQWPePhy-aGUz2KNnVg7bQq-sBPl7zQ,119
-vector_inspector/__main__.py,sha256=Vdhw8YA1K3wPMlbJQYL5WqvRzAKVeZ16mZQFO9VRmCo,62
-vector_inspector/core/__init__.py,sha256=hjOqiJwF1P0rXjiOKhK4qDTvBY7G3m4kq8taH-gKrFM,57
-vector_inspector/core/connections/__init__.py,sha256=cCwDy69Jy8ajiFQICcWfPnGoMtfrq69brzXJ4sVYCQ0,271
-vector_inspector/core/connections/base_connection.py,sha256=PiwVlrk-eUKSka6gmE2GSyml0xe48PrIWAcKhBJMBv8,7131
-vector_inspector/core/connections/chroma_connection.py,sha256=8dmMtTeMylZRqIvjbBgH6BWqbdd4vv9bJ9uf-C7fLHw,13943
-vector_inspector/core/connections/qdrant_connection.py,sha256=jjqRR_zJ-d2Kq9jS2_v6zTE_PY3HI_oss8mJi6gDZvo,28634
-vector_inspector/core/connections/template_connection.py,sha256=wsJiE4ma3cLUXk2eW5rnLMS5wG8JTekgEn46lHHQNoc,10642
-vector_inspector/main.py,sha256=puu1Fur298j6H8fG3_wF85RMhi4tjLZ0Are16kloMqM,479
-vector_inspector/services/__init__.py,sha256=QLgH7oybjHuEYDFNiBgmJxvSpgAzHEuBEPXa3SKJb_I,67
-vector_inspector/services/backup_restore_service.py,sha256=2bbmLamGg7gaYUl3zy_-8wL8xbYvO00kZ2tbImZkzi0,11782
-vector_inspector/services/filter_service.py,sha256=r1y_lPi4Mo4ZqVq_hfbTzRxcCbdH7RlsXg7ytkBTKi8,2334
-vector_inspector/services/import_export_service.py,sha256=OPCrBXBewCznu5o8wFGvduU0jGZAcBvp_Fpv15kdoJ4,10712
-vector_inspector/services/settings_service.py,sha256=4QdUbd7k2YIPX2EWc8eHfx0L2uhsxpe-CgIVogVHVmk,2081
-vector_inspector/services/visualization_service.py,sha256=mHI4qxT-V4R1kcOhE508vFTZ0HcmQICHvJ7dIoRracQ,4373
-vector_inspector/ui/__init__.py,sha256=262ZiXO6Luk8vZnhCIoYxOtGiny0bXK-BTKjxUNBx-w,43
-vector_inspector/ui/components/__init__.py,sha256=S-GWU1P820dJ6mHmeeBEy-CGF9fjpBeNf8vrbhRlFMk,30
-vector_inspector/ui/components/backup_restore_dialog.py,sha256=CrZ2u8vXzggv3aBkYR4FulpY74oZWMLW5BHU4dMiWug,13073
-vector_inspector/ui/components/filter_builder.py,sha256=NSR_hp-rzUZVAca6dIJhTxZA3igOKFM1g-YXiYPhFos,13360
-vector_inspector/ui/components/item_dialog.py,sha256=VMwehEjQ6xrdxWygR9J-hHsLfzOVb_E3ePUGYO_c7XA,3951
-vector_inspector/ui/components/loading_dialog.py,sha256=YEKYGU-R-Zz4CjXSArJtkNxgTy4O9hI5Bbt6qlIzD8U,1018
-vector_inspector/ui/main_window.py,sha256=5icldgWWDEH3Ltp8yQze6NUWRe842KDiIB4_qOvP4WM,14193
-vector_inspector/ui/views/__init__.py,sha256=FeMtVzSbVFBMjdwLQSQqD0FRW4ieJ4ZKXtTBci2e_bw,30
-vector_inspector/ui/views/collection_browser.py,sha256=oG9_YGPoVuMs-f_zSd4EcITmEU9caxvwuubsFUrNf-c,3991
-vector_inspector/ui/views/connection_view.py,sha256=CpHPSDee3RhH3_wNu62RI3Fn5ffBIGt_Va5iC0p1IoY,18116
-vector_inspector/ui/views/info_panel.py,sha256=Rc45dCdrdPI2Ir0GBoKMZU9hPrWf4zFXlNYfLSMP_9U,13975
-vector_inspector/ui/views/metadata_view.py,sha256=NGd2lw7S0EhfV7HJGBcn4x1mg71W30YliEOTlW_dssM,26829
-vector_inspector/ui/views/search_view.py,sha256=p6vt2heSpEbiRge46VSedTZjg7i3-AWdS0f96JXzlEU,10527
-vector_inspector/ui/views/visualization_view.py,sha256=uPOUpKJ00eFqCCohYEfMD8D9lNv2cy2FymPHouXK0Go,9163
-vector_inspector/utils/__init__.py,sha256=jhHBQC8C8bfhNlf6CAt07ejjStp_YAyleaYr2dm0Dk0,38
-vector_inspector/utils/lazy_imports.py,sha256=2XZ3ZnwTvZ5vvrh36nJ_TUjwwkgjoAED6i6P9yctvt0,1211
-vector_inspector-0.2.6.dist-info/RECORD,,