vector-vault 2.7.2__tar.gz → 2.7.4__tar.gz

{vector_vault-2.7.2 → vector_vault-2.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector_vault
-Version: 2.7.2
+Version: 2.7.4
 Summary: Customize ChatGPT and unleash the full potential of generative AI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-2.7.2 → vector_vault-2.7.4}/setup.py

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

{vector_vault-2.7.2 → vector_vault-2.7.4}/vector_vault.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-vault
-Version: 2.7.2
+Version: 2.7.4
 Summary: Customize ChatGPT and unleash the full potential of generative AI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-2.7.2 → vector_vault-2.7.4}/vectorvault/vault.py

@@ -155,16 +155,18 @@ class Vault:
             self.cloud_manager.delete_and_rename_all_items_after(item_id)
             new_index = get_vectors(self.dims)
             plus_one = 0
+            count = 0
 
             for i in range((num_existing_items - 1)):
                 if i in item_ids:
                     # equals 0 until the item being deleting has been removed
                     plus_one += 1 
                 else: 
+                    count += 1
                     vector = self.vectors.get_item_vector(i - plus_one) 
                     new_index.add_item(i - plus_one, vector)
 
-            self.x = i + 1
+            self.x = count + 1
             self.vectors = new_index
             print(f'Item {item_id} deleted')
 
@@ -189,10 +191,12 @@ class Vault:
         '''
         num_existing_items = self.vectors.get_n_items()
         new_index = get_vectors(self.dims)
+        count = 0
         for i in range(num_existing_items):
+            count += 1
             vector = self.vectors.get_item_vector(i)
             new_index.add_item(i, vector)
-        self.x = i + 1
+        self.x = count + 1
         self.vectors = new_index
 
     def load_vectors(self):
@@ -264,7 +268,7 @@ class Vault:
         
         return segments
     
-    def get_items(self, ids: list = []) -> list:
+    def get_items(self, ids: list = [], include_total=False) -> list:
         '''
             Get one or more items from the database. 
             Input the item id(s) in a list. -> Returns the items 
@@ -296,8 +300,12 @@ class Vault:
             if self.verbose == True:
                 print(f"Retrieved {len(ids)} items --- %s seconds ---" % (time.time() - start_time))
 
-        return results
-    
+        if include_total:
+            return [results, self.get_total_items()]
+        else:
+            return results
+
+
     def get_items_by_vector(self, vector, n: int = 4, include_distances=False):
         '''
             Internal function that returns vector similar items. Requires input vector, returns similar items