mage-ai 0.9.47__py3-none-any.whl → 0.9.48__py3-none-any.whl

mage_ai/streaming/sources/nats_js.py

@@ -91,8 +91,19 @@ class NATSSource(BaseSource):
             # Default consumer_name to stream_name if not provided
             consumer_name = self.config.consumer_name or self.config.stream_name
 
-            await self.js.add_stream(name=self.config.stream_name, subjects=[self.config.subject])
+            # Check if the stream exists, and create it if it doesn't
+            try:
+                await self.js.stream_info(self.config.stream_name)
+            except Exception as e:
+                # Check the exception type or message to ensure it's about a missing stream
+                if 'stream not found' in str(e).lower():
+                    await self.js.add_stream(
+                        name=self.config.stream_name,
+                        subjects=[self.config.subject],
+                    )
+
             self.psub = await self.js.pull_subscribe(self.config.subject, consumer_name)
+
         except NoServersError as e:
             self._print(f'Caught NoServersError while connecting to NATS server: {e}')
         except Exception as e:

mage_ai/tests/data_preparation/executors/test_block_executor.py

@@ -99,6 +99,8 @@ class BlockExecutorTest(TestCase):
         self.block_executor._execute.assert_called_once_with(
             analyze_outputs=analyze_outputs,
             block_run_id=None,
+            block_run_outputs_cache=None,
+            cache_block_output_in_memory=False,
             callback_url=callback_url,
             global_vars=global_vars,
             input_from_output=input_from_output,

mage_ai/tests/data_preparation/models/test_pipeline.py

@@ -35,6 +35,7 @@ class PipelineTest(DBTestCase):
         pipeline = Pipeline('test_pipeline_2', self.repo_path)
 
         self.assertEqual(pipeline.to_dict(), dict(
+            cache_block_output_in_memory=False,
             concurrency_config=dict(),
             data_integration=None,
             description=None,
@@ -200,6 +201,7 @@ class PipelineTest(DBTestCase):
         pipeline.delete_block(block)
         pipeline = Pipeline('test_pipeline_3', self.repo_path)
         self.assertEqual(pipeline.to_dict(), dict(
+            cache_block_output_in_memory=False,
             concurrency_config=dict(),
             data_integration=None,
             description=None,
@@ -293,6 +295,7 @@ class PipelineTest(DBTestCase):
         pipeline.add_block(block4, upstream_block_uuids=['block2', 'block3'])
         pipeline.execute_sync()
         self.assertEqual(pipeline.to_dict(), dict(
+            cache_block_output_in_memory=False,
             concurrency_config=dict(),
             data_integration=None,
             description=None,
@@ -409,6 +412,7 @@ class PipelineTest(DBTestCase):
         pipeline.add_block(block7, upstream_block_uuids=['block2', 'block3', 'block6'])
         pipeline.execute_sync()
         self.assertEqual(pipeline.to_dict(), dict(
+            cache_block_output_in_memory=False,
             concurrency_config=dict(),
             data_integration=None,
             description=None,
@@ -661,6 +665,7 @@ class PipelineTest(DBTestCase):
             self.assertEqual(
                 config_json,
                 dict(
+                    cache_block_output_in_memory=False,
                     concurrency_config=dict(),
                     created_at='2023-08-01 08:08:24+00:00',
                     data_integration=None,

{mage_ai-0.9.47.dist-info → mage_ai-0.9.48.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mage-ai
-Version: 0.9.47
+Version: 0.9.48
 Summary: Mage is a tool for building and deploying data pipelines.
 Home-page: https://github.com/mage-ai/mage-ai
 Author: Mage
@@ -74,7 +74,7 @@ Requires-Dist: azure-mgmt-containerinstance ==10.1.0 ; extra == 'all'
 Requires-Dist: azure-storage-blob ==12.14.1 ; extra == 'all'
 Requires-Dist: boto3 ==1.26.60 ; extra == 'all'
 Requires-Dist: botocore ==1.29.60 ; extra == 'all'
-Requires-Dist: chroma >=0.4.17 ; extra == 'all'
+Requires-Dist: chromadb >=0.4.17 ; extra == 'all'
 Requires-Dist: clickhouse-connect ==0.5.20 ; extra == 'all'
 Requires-Dist: confluent-avro ; extra == 'all'
 Requires-Dist: db-dtypes ==1.0.5 ; extra == 'all'
@@ -136,7 +136,7 @@ Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery ~=3.0 ; extra == 'bigquery'
 Requires-Dist: db-dtypes ==1.0.5 ; extra == 'bigquery'
 Provides-Extra: chroma
-Requires-Dist: chroma >=0.4.17 ; extra == 'chroma'
+Requires-Dist: chromadb >=0.4.17 ; extra == 'chroma'
 Provides-Extra: clickhouse
 Requires-Dist: clickhouse-connect ==0.5.20 ; extra == 'clickhouse'
 Provides-Extra: dbt