vector-vault 5.2.4__tar.gz → 5.2.6__tar.gz

{vector_vault-5.2.4 → vector_vault-5.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector_vault
-Version: 5.2.4
+Version: 5.2.6
 Summary: Quickly create ChatGPT RAG apps and Unleash the full potential of GenAI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-5.2.4 → vector_vault-5.2.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="vector_vault",
-    version="5.2.4",
+    version="5.2.6",
     packages=find_packages(),
     author="VectorVault.io",
     author_email="john@johnrood.com",

{vector_vault-5.2.4 → vector_vault-5.2.6}/vector_vault.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-vault
-Version: 5.2.4
+Version: 5.2.6
 Summary: Quickly create ChatGPT RAG apps and Unleash the full potential of GenAI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-5.2.4 → vector_vault-5.2.6}/vectorvault/cloudmanager.py

@@ -116,22 +116,32 @@ class CloudManager:
     
     def build_update(self):
         _map = self.get_mapping()
-        entry_updated = False
-        for entry in _map:
-            if entry['vault'] == self.vault:
-                entry['last_use'] = time.time()
-                entry['total_use'] = entry.get('total_use', 0) + 1
-                entry_updated = True
-                break
-        
-        _map.append({'vault': self.vault, 'last_use': time.time(), 'total_use': 1}) if not entry_updated else 0
-        
+        for i in range(len(_map)):
+            if _map[i]['vault'] == self.vault:
+                _map[i]['last_use'] = time.time()
+                try:
+                    _map[i]['total_use'] += 1
+                except:
+                    _map[i]['total_use'] = 1
+
         with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
             _path = temp_file.name
             json.dump(_map, temp_file, indent=2)
             
         self.upload_temp_file(_path, f'{self.username}.json')
-        
+    
+    def build_data_update(self):
+        _map = self.get_mapping()
+        for i in range(len(_map)):
+            if _map[i]['vault'] == self.vault:
+                _map[i]['last_update'] = time.time()
+
+        with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
+            _path = temp_file.name
+            json.dump(_map, temp_file, indent=2)
+            
+        self.upload_temp_file(_path, f'{self.username}.json')
+    
     def delete_blob(self, blob):
         blob.delete()
 

{vector_vault-5.2.4 → vector_vault-5.2.6}/vectorvault/vault.py

@@ -254,7 +254,8 @@ class Vault:
                 total_saved_items += 1
 
         self.upload_vectors(vault)
-        self.cloud_manager.update()
+        self.update_vault_data(this_vault_only=True)
+        self.cloud_manager.build_data_update()
         print(f"upload time --- {(time.time() - start_time)} seconds --- {total_saved_items} items saved") if self.verbose else 0
             
 
@@ -278,6 +279,7 @@ class Vault:
         self.items.clear()
         self.cloud_manager.delete()
         self.x = 0
+        self.update_vault_data(remove_vault=True)
         print('Vault deleted')
     
 
@@ -287,35 +289,35 @@ class Vault:
         self.map.popitem()
 
 
-    def update_vault_data(self):
+    def update_vault_data(self, this_vault_only=False, remove_vault=False):
         nary = []
-        try:
+        try: # try because new users do not have this file on first call
             vault_data = self.cloud_manager.get_mapping()
-            all_vaults = self.get_vaults('')
-            _data = [vault for vault in vault_data if vault['vault'] in all_vaults]
-            for i in all_vaults:
-                total_items = self.get_total_items(i)
-                time.sleep(.01)
-                item = self.get_items([ total_items - 1 ], i)[0]['metadata']
-                try: 
-                    time_for_last_item = item['updated']
-                except:
-                    time_for_last_item = item['updated_at']
-
-                vault_dict = { 'vault': f'{i}', 'total_items': total_items, 'last_update': time_for_last_item }
-
-                for original_data in _data:  
-                    if original_data['vault'] == f'{i}':
-                        if 'last_use' in original_data:
-                            vault_dict['last_use'] = original_data['last_use']
-                        if 'total_use' in original_data:
-                            vault_dict['total_use'] = original_data['total_use']
-
-                nary.append(vault_dict)
-
-        except Exception as e:
+            all_vaults = [i['vault'] for i in vault_data]
+            if self.vault not in all_vaults:
+                nary.append({ 'vault': f'{self.vault}', 'total_items': self.get_total_items(), 'last_update': time.time(), 'last_use': time.time(), 'total_use': 1 })
+
+            for i in vault_data:
+                if not this_vault_only and not remove_vault: 
+                    total_items = self.get_total_items(i['vault']) 
+                else:
+                    total_items = i['total_items']
+
+                if this_vault_only and i['vault'] == self.vault:
+                    total_items = self.get_total_items()
+                
+                vault_dict = { 'vault': i['vault'], 'total_items': total_items, 'last_update': i['last_update'] }
+
+                if 'last_use' in i:
+                    vault_dict['last_use'] = i['last_use']
+                if 'total_use' in i:
+                    vault_dict['total_use'] = i['total_use']
+
+                nary.append(vault_dict) if not remove_vault else 0
+
+        except Exception as e: # so we make the file 
             print(f"An error occurred: {e}")
-            nary.append({ 'vault': 'demo', 'total_items': 0, 'last_update': time.time(), 'last_use': time.time(), 'total_use': 0 })
+            nary.append({ 'vault': 'demo', 'total_items': 0, 'last_update': time.time(), 'last_use': time.time(), 'total_use': 1 })
 
         with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
             nary_temp_file_path = temp_file.name
@@ -362,6 +364,7 @@ class Vault:
             self.remap(item_id)
             rebuild_vectors(item_id)
 
+        self.update_vault_data(this_vault_only=True)
         print(f'Item {item_id} deleted') if self.verbose else 0