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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector_vault
-Version: 2.7.3
+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.3 → vector_vault-2.7.4}/setup.py

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

{vector_vault-2.7.3 → 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.3
+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.3 → vector_vault-2.7.4}/vectorvault/vault.py

@@ -268,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 
@@ -300,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