vector-vault 4.3.2__tar.gz → 4.3.4__tar.gz

{vector_vault-4.3.2 → vector_vault-4.3.4}/PKG-INFO

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

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

{vector_vault-4.3.2 → vector_vault-4.3.4}/vector_vault.egg-info/PKG-INFO

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

@@ -99,7 +99,8 @@ class CloudManager:
         return _map
     
     def update(self):
-        T(target=call_update, args=(self.user, self.vault, self.api)).start()
+        t = T(target=call_update, args=(self.user, self.vault, self.api))
+        t.start()
     
     def build_update(self):
         _map = self.get_mapping()

{vector_vault-4.3.2 → vector_vault-4.3.4}/vectorvault/vault.py

@@ -82,7 +82,7 @@ class Vault:
             self.openai_key = openai_key
             openai.api_key = self.openai_key
         try:
-            self.cloud_manager = CloudManager(self.user, self.api, self.vault, self.openai_key)
+            self.cloud_manager = CloudManager(self.user, self.api, self.vault)
             print(f'Connected vault: {self.vault}') if self.verbose else 0 
         except Exception as e:
             print('API KEY NOT FOUND or no internet connection!', e)
@@ -1009,7 +1009,7 @@ class Vault:
                         print(f"API Failed too many times, exiting loop: {e}.")
                         break
 
-        self.update_conversation_history(self.cuid, response, user=False) if self.cuid else None
+        self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {response}', user=False) if self.cuid else None
         print("get chat time --- %s seconds ---" % (time.time() - start_time)) if self.verbose else 0
             
         return {'response': response, 'context': context} if return_context else response
@@ -1129,7 +1129,7 @@ class Vault:
                         except Exception as e:
                             raise e
                         if counter == len(inputs):
-                            self.update_conversation_history(self.cuid, full_response, user=False) if self.cuid else None
+                            self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}', user=False) if self.cuid else None
                             yield '!END'
                             self.rate_limiter.on_success()
                     
@@ -1167,11 +1167,11 @@ class Vault:
                                             for i in range(len(metatag)):
                                                 yield str(metatag_prefixes[i]) + str(item['metadata'][f'{metatag[i]}'])
                                 yield item['data']
-                            self.update_conversation_history(self.cuid, full_response, user=False) if self.cuid else None
+                            self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}', user=False) if self.cuid else None
                             yield '!END'
                             self.rate_limiter.on_success()
                         else:
-                            self.update_conversation_history(self.cuid, full_response, user=False) if self.cuid else None
+                            self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}', user=False) if self.cuid else None
                             yield '!END'
                             self.rate_limiter.on_success()
 
@@ -1183,7 +1183,7 @@ class Vault:
                             self.rate_limiter.on_success()
                         except Exception as e:
                             raise e
-                        self.update_conversation_history(self.cuid, full_response, user=False) if self.cuid else None
+                        self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}', user=False) if self.cuid else None
                         yield '!END'
 
                     self.rate_limiter.on_success()
@@ -1218,7 +1218,7 @@ class Vault:
                 return full_text
         
 
-    def print_vault_data(self, print_data: bool = True, return_data: bool = False):
+    def print_vault_data(self, return_data: bool = False):
         ''' 
             Function to print vault data 
         ''' 
@@ -1265,8 +1265,7 @@ class Vault:
             yield f"data: {json.dumps({'data': word})} \n\n"
 
 
-    def update_conversation_history(self, conversation_id, message, metadata_list = None, user = True):
-        message = f'User: {message}' if user else f'AI: {message}'
+    def update_conversation_history(self, conversation_id, message, metadata_list = None):
         try:
             metadata_list = metadata_list if metadata_list else json.loads(self.cloud_manager.download_text_from_cloud(f'user_history/{conversation_id}/metadata'))
         except:
@@ -1310,9 +1309,6 @@ class Vault:
 
         message_ids = golden_retriever(meta) if meta != None else []
 
-        update = T(target=self.update_conversation_history, args=(conversation_id, message, meta if meta else None))
-        update.start()
-
         print('message_ids:', message_ids) if self.verbose else None
 
         if message_ids != []:
@@ -1338,7 +1334,6 @@ class Vault:
 
             history = history + "Vector search conversation history:" + '\n'.join(vector_history)
 
-        update.join()
         return history