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

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

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vector-vault
-Version: 7.2.7
+Version: 7.2.9
 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.9}/vectorvault/vault.py

@@ -1668,11 +1668,9 @@ class Vault:
                                 raise e
 
                             if return_context:
-                                for item in context:
-                                    if not metatag:
-                                        for tag in item['metadata']:
-                                            yield str(item['metadata'][f'{tag}'])
-                                    else:
+                                # Only stream context items if metatag parameters are specified
+                                if metatag:
+                                    for item in context:
                                         if metatag_prefixes:
                                             if metatag_suffixes:
                                                 for i in range(len(metatag)):
@@ -1680,8 +1678,11 @@ class Vault:
                                             else:
                                                 for i in range(len(metatag)):
                                                     yield str(metatag_prefixes[i]) + str(item['metadata'][f'{metatag[i]}'])
-                                    yield item['data']
+                                        yield item['data']
+                                
                                 self.update_conversation_history(self.cuid, f'User: {text} \n\nAI: {full_response}') if self.cuid else None
+                                # Yield the context in the same format as get_chat
+                                yield json.dumps({'response': full_response, 'context': context})
                                 yield '!END'
                                 self.rate_limiter.on_success()
                             else: