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

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

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-vault
-Version: 2.7.1
+Version: 2.7.2
 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.1 → vector_vault-2.7.2}/vectorvault/cloudmanager.py

@@ -16,9 +16,9 @@
 import tempfile
 import os
 import json
-from creds import CustomCredentials
-from vecreq import call_proj
-from itemize import cloud_name
+from .creds import CustomCredentials
+from .vecreq import call_proj
+from .itemize import cloud_name
 from google.cloud import storage
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
@@ -126,11 +126,6 @@ class CloudManager:
             print(f"Item at path {old_path} does not exist.")
             
     def delete_and_rename_all_items_after(self, item_id):
-        """
-        Deletes an item specified by item_id and renames all subsequent items 
-        by decrementing their item_ids by one.
-        """
-        
         # Delete the specified item
         self.delete_item(item_id)
         

{vector_vault-2.7.1 → vector_vault-2.7.2}/vectorvault/itemize.py

@@ -13,7 +13,7 @@
 # is strictly forbidden unless prior written permission is obtained
 # from Vector Vault. See license for consent.import datetime 
 
-from vecreq import call_name_vecs, call_buildpath
+from .vecreq import call_name_vecs, call_buildpath
 from annoy import AnnoyIndex
 import threading
 from copy import deepcopy

{vector_vault-2.7.1 → vector_vault-2.7.2}/vectorvault/tools_gpt.py

@@ -1,4 +1,4 @@
-from ai import AI
+from .ai import AI
 import time
 
 '''

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

@@ -18,15 +18,14 @@ import tempfile
 import os
 import time
 import re
-import openai
 import json
 import traceback
 from concurrent.futures import ThreadPoolExecutor
-from cloudmanager import CloudManager
-from ai import AI
-from itemize import itemize, name_vecs, get_item, get_vectors, build_return, cloud_name
-from vecreq import call_get_similar
-from tools_gpt import ToolsGPT
+from .cloudmanager import CloudManager
+from .ai import AI, openai
+from .itemize import itemize, name_vecs, get_item, get_vectors, build_return, cloud_name
+from .vecreq import call_get_similar
+from .tools_gpt import ToolsGPT
 
 
 class Vault:
@@ -96,7 +95,7 @@ class Vault:
         if self.saved_already == True:
             self.clear_cache()
             print("The last save was aborted before the build process finished. Clearing cache to start again...")
-        self.saved_already = True
+        self.saved_already = True # Make sure the if the save process is interrupted, data will not get corrupted
         start_time = time.time()
         self.vectors.build(trees)
 
@@ -143,29 +142,31 @@ class Vault:
         self.cloud_manager.delete()
         print('Vault deleted')
 
-    def delete_item(self, item_id : int):
+    def delete_items(self, item_ids : list[int]) -> None:
         '''
-            Deletes the item for the item id passed in
+            Deletes one or more items from item_id(s) passed in.
+            item_ids is a list of one or more integers
         '''
         if self.verbose == True:
             print('Deleting item and rebuilding database...')
 
-        num_existing_items = self.vectors.get_n_items()
-        self.cloud_manager.delete_and_rename_all_items_after(item_id)
-        new_index = get_vectors(self.dims)
-        plus_one = 0
-
-        for i in range((num_existing_items - 1)):
-            if i == item_id:
-                # equals 0 until the item being deleting has been removed
-                plus_one = 1 
-            else: 
-                vector = self.vectors.get_item_vector(i - plus_one) 
-                new_index.add_item(i - plus_one, vector)
-
-        self.x = i + 1
-        self.vectors = new_index
-        print('Item deleted')
+        for item_id in item_ids:
+            num_existing_items = self.vectors.get_n_items()
+            self.cloud_manager.delete_and_rename_all_items_after(item_id)
+            new_index = get_vectors(self.dims)
+            plus_one = 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: 
+                    vector = self.vectors.get_item_vector(i - plus_one) 
+                    new_index.add_item(i - plus_one, vector)
+
+            self.x = i + 1
+            self.vectors = new_index
+            print(f'Item {item_id} deleted')
 
     def check_index(self):
         '''