vector-vault 7.2.7__tar.gz → 7.2.8__tar.gz

{vector_vault-7.2.7 → vector_vault-7.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector_vault
-Version: 7.2.7
+Version: 7.2.8
 Summary: Quickly create RAG apps, Agents, and Unleash the full power of AI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-7.2.7 → vector_vault-7.2.8}/setup.py

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

{vector_vault-7.2.7 → vector_vault-7.2.8}/vector_vault.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-vault
-Version: 7.2.7
+Version: 7.2.8
 Summary: Quickly create RAG apps, Agents, and Unleash the full power of AI with Vector Vault
 Home-page: https://github.com/John-Rood/VectorVault
 Author: VectorVault.io

{vector_vault-7.2.7 → vector_vault-7.2.8}/vectorvault/vault.py

@@ -1601,6 +1601,7 @@ class Vault:
             if self.verbose:
                 print("Performing image inference...")
 
+            full_response = ''
             for i in self._ai.platform.image_inference(
                 image_path=image_path,
                 image_url=image_url,
@@ -1609,8 +1610,10 @@ class Vault:
                 temperature=temperature,
                 timeout=timeout
             ):
+                full_response += i
                 yield i
             yield '!END'
+            return full_response
             
         else: 
             for segment in inputs:
@@ -1639,6 +1642,7 @@ class Vault:
                                 self.rate_limiter.on_success()
                             except Exception as e:
                                 raise e
+                            counter += 1
                             if counter == len(inputs):
                                 self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}') if self.cuid else None
                                 yield '!END'
@@ -1710,7 +1714,11 @@ class Vault:
                             print(f"API Failed too many times, exiting loop: {e}.")
                             break
             
-        print("get chat time --- %s seconds ---" % (time.time() - start_time)) if self.verbose else 0
+            print("get chat time --- %s seconds ---" % (time.time() - start_time)) if self.verbose else 0
+            if return_context:
+                return {'response': full_response, 'context': context}
+            else:
+                return full_response
             
 
     def print_stream(self, function, printing=True):