mage-ai 0.9.76__tar.gz → 0.9.77__tar.gz

{mage_ai-0.9.76/mage_ai.egg-info → mage_ai-0.9.77}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mage-ai
-Version: 0.9.76
+Version: 0.9.77
 Summary: Mage is a tool for building and deploying data pipelines.
 Home-page: https://github.com/mage-ai/mage-ai
 Author: Mage
@@ -41,7 +41,7 @@ Requires-Dist: memory_profiler
 Requires-Dist: newrelic==8.8.0
 Requires-Dist: numpy>=1.22.0
 Requires-Dist: pandas>=1.3.0
-Requires-Dist: polars<0.19.2,>=0.18.0
+Requires-Dist: polars==1.29.0
 Requires-Dist: protobuf>=4.25.0
 Requires-Dist: psutil==5.9.8
 Requires-Dist: pyairtable>=2.3.3
@@ -229,165 +229,128 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-<div align="left">
-  <h1>Mage AI</h1>
+# Mage OSS
 
-  <p>Give your data team magical powers.</p>
-
-<a href="https://mage.ai"><img alt="Mage AI GitHub repo stars" src="https://img.shields.io/github/stars/mage-ai/mage-ai?style=for-the-badge&logo=github&labelColor=000000&logoColor=FFFFFF&label=stars&color=0500ff" /></a>
-<a href="https://hub.docker.com/r/mageai/mageai"><img alt="Mage AI Docker downloads" src="https://img.shields.io/docker/pulls/mageai/mageai?style=for-the-badge&logo=docker&labelColor=000000&logoColor=FFFFFF&label=pulls&color=6A35FF" /></a>
-<a href="https://github.com/mage-ai/mage-ai/blob/master/LICENSE"><img alt="Mage AI license" src="https://img.shields.io/github/license/mage-ai/mage-ai?style=for-the-badge&logo=codeigniter&labelColor=000000&logoColor=FFFFFF&label=license&color=FFCC19" /></a>
-<a href="https://www.mage.ai/chat"><img alt="Join the Mage AI community" src="https://img.shields.io/badge/Join%20the%20community-black.svg?style=for-the-badge&logo=lightning&labelColor=000000&logoColor=FFFFFF&label=&color=DD55FF&logoWidth=20" /></a>
-
-</div>
+### Build modern data pipelines locally — fast, visual, and production-ready.
 
 <br />
 
-<a href="https://cloud.mage.ai/sign-up">
-  <picture>
-    <source media="(prefers-color-scheme: dark)" srcset="https://github.com/mage-ai/assets/blob/main/oss/hero.png?raw=true">
-    <img alt="Mage AI hero" src="https://github.com/mage-ai/assets/blob/main/oss/hero.png?raw=true" />
-  </picture>
-</a>
+Mage OSS is a self-hosted development environment designed to help teams create production-grade data pipelines with confidence.
 
-<br />
-<br />
+Ideal for automating ETL tasks, architecting data flow, or orchestrating transformations — all in a fast, notebook-style interface powered by modular code.
 
-Mage is a hybrid framework for transforming and integrating data. It combines the best of both worlds: the flexibility of notebooks with the rigor of modular code.
+When it’s time to scale, [Mage Pro](https://mage.ai) — our core platform — unlocks enterprise orchestration, collaboration, and AI-powered workflows.
 
 <br />
 
-- Extract and synchronize data from 3rd party sources.
-- Transform data with real-time and batch pipelines using Python, SQL, and R.
-- Load data into your data warehouse or data lake using our pre-built connectors.
-- Run, monitor, and orchestrate thousands of pipelines without losing sleep.
+[![GitHub Stars](https://img.shields.io/github/stars/mage-ai/mage-ai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://github.com/mage-ai/mage-ai/stargazers)
+[![Docker Pulls](https://img.shields.io/docker/pulls/mageai/mageai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://hub.docker.com/r/mageai/mageai)
+[![Slack Community](https://img.shields.io/badge/Join%20the%20community-black.svg?style=for-the-badge&logo=lightning&labelColor=000000&color=1C1C1C)](https://www.mage.ai/chat)
+[![License](https://img.shields.io/github/license/mage-ai/mage-ai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://github.com/mage-ai/mage-ai/blob/master/LICENSE)
 
 <br />
 
-Plus hundreds of enterprise-class features, infrastructure innovations, and magical surprises.
+## What you can do with Mage OSS
 
-<h1></h1>
+- Build pipelines locally with Python, SQL, or R in a modular notebook-style UI
 
-#### Available in two spellbinding versions
+- Run jobs manually or on a schedule (cron supported)
 
-<br />
+- Connect to databases, APIs, and cloud storage with prebuilt connectors
 
-<table style="width: 100%; border: none;" cellspacing="0" cellpadding="0" border="0">
-  <tr>
-    <td rowspan="2">
-      <a href="https://cloud.mage.ai/sign-up">
-        <img alt="Mage Pro" width="124" align="left" src="https://github.com/mage-ai/assets/blob/main/oss/card-pro.png?raw=true" />
-      </a>
-      <b>For teams.</b> Fully managed platform
-      for integrating and transforming data.
-    </td>
-    <td rowspan="2">
-      <a href="https://github.com/mage-ai/mage-ai?tab=readme-ov-file#its-magic">
-        <img alt="Mage OSS" width="124" align="left" src="https://github.com/mage-ai/assets/blob/main/oss/card-oss.png?raw=true" />
-      </a>
-      <b>Self-hosted.</b> System to build, run, and
-      manage data pipelines.
-    </td>
-  </tr>
-</table>
+- Debug visually with logs, live previews, and step-by-step execution
 
-<br />
+- Set up quickly with Docker, pip, or conda — no cloud account required
 
-<a href="https://cloud.mage.ai/sign-up">
-  <img alt="Try out Mage Pro" src="https://img.shields.io/badge/try%20it%20out%20now-black.svg?style=for-the-badge&logo=artifacthub&labelColor=6A35FF&logoColor=FFFFFF&label=mage%20pro&color=6A35FF">
-</a>
+- Your go-to workspace for local pipeline development — fully in your control.
+  
+<img width="100%" alt="mage" src="https://github.com/user-attachments/assets/75992872-20a6-4120-8bf0-9c22a3d66450" />
 
-<br />
-<br />
 
-<div align="left">
+<br /><br />
 
-# It’s magic.
+## Start local. Scale when you're ready.
 
-<p align="left">
-  For documentation on getting started, how to develop, and how to deploy to production check out the live
-  <br />
-  <a
-    href="https://docs.mage.ai?source=github"
-    target="_blank"
-  ><b>Developer documentation portal</b></a>.
-</p>
+Use Mage OSS to build and run pipelines on your machine. When you're ready for advanced tooling, performance, and AI-assisted productivity, Mage Pro is just one click away.
+
+[**Try Mage Pro free →**](https://mage.ai)
 
 <br />
 
-## 🏃‍♀️ Install
+### Quickstart
 
-The recommended way to install the latest version of Mage is through Docker with the following command:
+Install using Docker (recommended):
 
 ```bash
 docker pull mageai/mageai:latest
 ```
 
-You can also install Mage using pip or conda, though this may cause dependency issues without the proper environment.
+Or with pip:
 
 ```bash
 pip install mage-ai
 ```
+
+Or with conda:
+
 ```bash
 conda install -c conda-forge mage-ai
 ```
 
-Looking for help? The _fastest_ way to get started is by checking out our documentation [here](https://docs.mage.ai/getting-started/setup).
-
-Looking for quick examples? Open a [demo](https://demo.mage.ai/) project right in your browser or check out our [guides](https://docs.mage.ai/guides/overview).
+Full setup guide and docs: [docs.mage.ai](https://docs.mage.ai)
 
-## 🎮 Demo
+<br />
 
-### Live demo
+## Core Features
 
-Build and run a data pipeline with our <b>[demo app](https://demo.mage.ai/)</b>.
+| Feature | Description |
+| :- | :- |
+| Modular pipelines | Build pipelines block-by-block using Python, SQL, or R |
+| Notebook UI | Interactive editor for writing and documenting logic |
+| Data integrations | Prebuilt connectors to databases, APIs, and cloud storage |
+| Scheduling | Trigger pipelines manually or on a schedule |
+| Visual debugging | Step-by-step logs, data previews, and error handling |
+| dbt support | Build and run dbt models directly inside Mage |
 
-> WARNING
->
-> The live demo is public to everyone, please don’t save anything sensitive (e.g. passwords, secrets, etc).
-### Demo video (5 min)
+<br />
 
-[![Mage quick start demo](https://github.com/mage-ai/assets/blob/main/overview/overview-video.png?raw=True)](https://youtu.be/GswOdShLGmg)
+## Example Use Cases
 
-<sub><i>Click the image to play video</i></sub>
+- Move data from Google Sheets to Snowflake with a Python transform
+- Schedule a daily SQL pipeline to clean and aggregate product data
+- Develop dbt models in a visual notebook-style interface
+- Run simple ETL/ELT jobs locally with full transparency
 
 <br />
 
-## 🔮 [Features](https://docs.mage.ai/about/features)
+## Documentation
 
-|   |   |   |
-| --- | --- | --- |
-| 🎶 | <b>[Orchestration](https://docs.mage.ai/design/data-pipeline-management)</b> | Schedule and manage data pipelines with observability. |
-| 📓 | <b>[Notebook](https://docs.mage.ai/about/features#notebook-for-building-data-pipelines)</b> | Interactive Python, SQL, & R editor for coding data pipelines. |
-| 🏗️ | <b>[Data integrations](https://docs.mage.ai/data-integrations/overview)</b> | Synchronize data from 3rd party sources to your internal destinations. |
-| 🚰 | <b>[Streaming pipelines](https://docs.mage.ai/guides/streaming-pipeline)</b> | Ingest and transform real-time data. |
-| ❎ | <b>[dbt](https://docs.mage.ai/dbt/overview)</b> | Build, run, and manage your dbt models with Mage. |
+Looking for how-to guides, examples, or advanced configuration?
 
-<br />
+Explore our full documentation at [docs.mage.ai](https://docs.mage.ai).
 
-<b>A sample data pipeline defined across 3 files ➝</b>
 
 <br />
 
-1. Load data ➝
-    ```python
-    @data_loader
-    def load_csv_from_file() -> pl.DataFrame:
-        return pl.read_csv('default_repo/titanic.csv')
-    ```
-1. Transform data ➝
-    ```python
-    @transformer
-    def select_columns_from_df(df: pl.DataFrame, *args) -> pl.DataFrame:
-        return df[['Age', 'Fare', 'Survived']]
-    ```
-1. Export data ➝
-    ```python
-    @data_exporter
-    def export_titanic_data_to_disk(df: pl.DataFrame) -> None:
-        df.to_csv('default_repo/titanic_transformed.csv')
-    ```
+## Contributing
+
+We welcome contributions of all kinds — bug fixes, docs, new features, or community examples.
+
+Start with our [contributing guide](https://docs.mage.ai/contributing/overview), check out open issues, or suggest improvements.
 
 <br />
 
-[<img alt="Water mage casting spell" height="300" src="https://github.com/mage-ai/assets/blob/main/mage-water-charging-up.svg?raw=True" />](https://www.mage.ai/)
+## Ready to scale? Mage Pro has you covered.
+
+Mage Pro is a powered-up platform built for teams.
+It adds everything you need for production pipelines, at scale.
+
+- Magical AI-assisted development and debugging
+- Multi-environment orchestration
+- Role-based access control
+- Real-time monitoring & alerts
+- Powerful CI/CD & version control
+- Powerful enterprise features
+- Available fully managed, hybrid, or on-premises
+
+[**Try Mage Pro free →**](https://mage.ai)

mage_ai-0.9.77/README.md

@@ -0,0 +1,125 @@
+# Mage OSS
+
+### Build modern data pipelines locally — fast, visual, and production-ready.
+
+<br />
+
+Mage OSS is a self-hosted development environment designed to help teams create production-grade data pipelines with confidence.
+
+Ideal for automating ETL tasks, architecting data flow, or orchestrating transformations — all in a fast, notebook-style interface powered by modular code.
+
+When it’s time to scale, [Mage Pro](https://mage.ai) — our core platform — unlocks enterprise orchestration, collaboration, and AI-powered workflows.
+
+<br />
+
+[![GitHub Stars](https://img.shields.io/github/stars/mage-ai/mage-ai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://github.com/mage-ai/mage-ai/stargazers)
+[![Docker Pulls](https://img.shields.io/docker/pulls/mageai/mageai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://hub.docker.com/r/mageai/mageai)
+[![Slack Community](https://img.shields.io/badge/Join%20the%20community-black.svg?style=for-the-badge&logo=lightning&labelColor=000000&color=1C1C1C)](https://www.mage.ai/chat)
+[![License](https://img.shields.io/github/license/mage-ai/mage-ai?style=for-the-badge&labelColor=000000&color=1C1C1C)](https://github.com/mage-ai/mage-ai/blob/master/LICENSE)
+
+<br />
+
+## What you can do with Mage OSS
+
+- Build pipelines locally with Python, SQL, or R in a modular notebook-style UI
+
+- Run jobs manually or on a schedule (cron supported)
+
+- Connect to databases, APIs, and cloud storage with prebuilt connectors
+
+- Debug visually with logs, live previews, and step-by-step execution
+
+- Set up quickly with Docker, pip, or conda — no cloud account required
+
+- Your go-to workspace for local pipeline development — fully in your control.
+  
+<img width="100%" alt="mage" src="https://github.com/user-attachments/assets/75992872-20a6-4120-8bf0-9c22a3d66450" />
+
+
+<br /><br />
+
+## Start local. Scale when you're ready.
+
+Use Mage OSS to build and run pipelines on your machine. When you're ready for advanced tooling, performance, and AI-assisted productivity, Mage Pro is just one click away.
+
+[**Try Mage Pro free →**](https://mage.ai)
+
+<br />
+
+### Quickstart
+
+Install using Docker (recommended):
+
+```bash
+docker pull mageai/mageai:latest
+```
+
+Or with pip:
+
+```bash
+pip install mage-ai
+```
+
+Or with conda:
+
+```bash
+conda install -c conda-forge mage-ai
+```
+
+Full setup guide and docs: [docs.mage.ai](https://docs.mage.ai)
+
+<br />
+
+## Core Features
+
+| Feature | Description |
+| :- | :- |
+| Modular pipelines | Build pipelines block-by-block using Python, SQL, or R |
+| Notebook UI | Interactive editor for writing and documenting logic |
+| Data integrations | Prebuilt connectors to databases, APIs, and cloud storage |
+| Scheduling | Trigger pipelines manually or on a schedule |
+| Visual debugging | Step-by-step logs, data previews, and error handling |
+| dbt support | Build and run dbt models directly inside Mage |
+
+<br />
+
+## Example Use Cases
+
+- Move data from Google Sheets to Snowflake with a Python transform
+- Schedule a daily SQL pipeline to clean and aggregate product data
+- Develop dbt models in a visual notebook-style interface
+- Run simple ETL/ELT jobs locally with full transparency
+
+<br />
+
+## Documentation
+
+Looking for how-to guides, examples, or advanced configuration?
+
+Explore our full documentation at [docs.mage.ai](https://docs.mage.ai).
+
+
+<br />
+
+## Contributing
+
+We welcome contributions of all kinds — bug fixes, docs, new features, or community examples.
+
+Start with our [contributing guide](https://docs.mage.ai/contributing/overview), check out open issues, or suggest improvements.
+
+<br />
+
+## Ready to scale? Mage Pro has you covered.
+
+Mage Pro is a powered-up platform built for teams.
+It adds everything you need for production pipelines, at scale.
+
+- Magical AI-assisted development and debugging
+- Multi-environment orchestration
+- Role-based access control
+- Real-time monitoring & alerts
+- Powerful CI/CD & version control
+- Powerful enterprise features
+- Available fully managed, hybrid, or on-premises
+
+[**Try Mage Pro free →**](https://mage.ai)

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/api/resources/GitFileResource.py

@@ -32,6 +32,14 @@ class GitFileResource(GenericResource):
                 pass
 
         file_path_absolute = os.path.join(git_manager.repo_path, file_path)
+
+        # Prevent path traversal by resolving the absolute path location
+        # and checking if it's within the repo
+        if not os.path.abspath(file_path_absolute).startswith(
+            os.path.abspath(git_manager.repo_path)
+        ):
+            raise Exception("Access denied: Attempted path traversal")
+
         file = File.from_path(file_path_absolute)
         if not file.exists():
             file = File.from_path(file_path_absolute, '')

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/cli/main.py

@@ -194,7 +194,11 @@ def run(
     from mage_ai.orchestration.db.models.schedules import PipelineRun
     from mage_ai.orchestration.utils.git import log_git_sync, run_git_sync
     from mage_ai.server.logger import Logger
-    from mage_ai.settings import SENTRY_DSN, SENTRY_TRACES_SAMPLE_RATE
+    from mage_ai.settings import (
+        SENTRY_DSN,
+        SENTRY_SERVER_NAME,
+        SENTRY_TRACES_SAMPLE_RATE,
+    )
     from mage_ai.shared.hash import merge_dict
 
     logger = Logger().new_server_logger(__name__)
@@ -204,6 +208,7 @@ def run(
         sentry_sdk.init(
             sentry_dsn,
             traces_sample_rate=SENTRY_TRACES_SAMPLE_RATE,
+            server_name=SENTRY_SERVER_NAME,
         )
     (enable_new_relic, application) = initialize_new_relic()
 

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/executors/block_executor.py

@@ -693,7 +693,7 @@ class BlockExecutor:
                             ),
                             tags=tags,
                         )
-                    self._execute_callback(
+                    self.execute_callback(
                         'on_failure',
                         block_run_id=block_run_id,
                         callback_kwargs=dict(__error=error, retry=self.retry_metadata),
@@ -748,7 +748,7 @@ class BlockExecutor:
             # success callback because this isn’t the last data integration block that needs
             # to run.
             if not data_integration_metadata or is_original_block:
-                self._execute_callback(
+                self.execute_callback(
                     'on_success',
                     block_run_id=block_run_id,
                     callback_kwargs=dict(retry=self.retry_metadata),
@@ -1253,7 +1253,7 @@ class BlockExecutor:
 
         return result
 
-    def _execute_callback(
+    def execute_callback(
         self,
         callback: str,
         global_vars: Dict,
@@ -1275,6 +1275,11 @@ class BlockExecutor:
             dynamic_block_index: Index of the dynamic block.
             dynamic_upstream_block_uuids: List of UUIDs of the dynamic upstream blocks.
         """
+        if logging_tags is None:
+            logging_tags = self.build_tags(
+                block_run_id=block_run_id,
+                pipeline_run_id=pipeline_run.id if pipeline_run is not None else None,
+            )
         upstream_block_uuids_override = []
         if is_dynamic_block_child(self.block):
             if not self.block_run and block_run_id:

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/executors/pipeline_executor.py

@@ -10,10 +10,13 @@ from mage_ai.data_preparation.logging.logger import DictLogger
 from mage_ai.data_preparation.logging.logger_manager_factory import LoggerManagerFactory
 from mage_ai.data_preparation.models.pipeline import Pipeline
 from mage_ai.orchestration.db.models.schedules import BlockRun, PipelineRun
+from mage_ai.server.logger import Logger
 from mage_ai.shared.hash import merge_dict
 from mage_ai.usage_statistics.constants import EventNameType, EventObjectType
 from mage_ai.usage_statistics.logger import UsageStatisticLogger
 
+logger = Logger().new_server_logger(__name__)
+
 
 class PipelineExecutor:
     def __init__(self, pipeline: Pipeline, execution_partition: str = None):
@@ -53,25 +56,38 @@ class PipelineExecutor:
             update_status (bool): Whether to update the execution status.
             **kwargs: Additional keyword arguments.
         """
-        if pipeline_run_id is None:
-            # Execute the pipeline without block runs
-            asyncio.run(self.pipeline.execute(
-                analyze_outputs=analyze_outputs,
-                global_vars=global_vars,
-                run_sensors=run_sensors,
-                run_tests=run_tests,
-                update_status=update_status,
-            ))
-        else:
-            # Supported pipeline types: Standard batch pipeline
-            pipeline_run = PipelineRun.query.get(pipeline_run_id)
-            if pipeline_run.status != PipelineRun.PipelineRunStatus.RUNNING:
-                return
-            asyncio.run(self.__run_blocks(
-                pipeline_run,
-                allow_blocks_to_fail=allow_blocks_to_fail,
-                global_vars=global_vars,
-            ))
+        # Create the async task to execute
+        async def _execute_task():
+            if pipeline_run_id is None:
+                # Execute the pipeline without block runs
+                await self.pipeline.execute(
+                    analyze_outputs=analyze_outputs,
+                    global_vars=global_vars,
+                    run_sensors=run_sensors,
+                    run_tests=run_tests,
+                    update_status=update_status,
+                )
+            else:
+                # Supported pipeline types: Standard batch pipeline
+                pipeline_run = PipelineRun.query.get(pipeline_run_id)
+                if pipeline_run.status != PipelineRun.PipelineRunStatus.RUNNING:
+                    return
+                await self.__run_blocks(
+                    pipeline_run,
+                    allow_blocks_to_fail=allow_blocks_to_fail,
+                    global_vars=global_vars,
+                )
+        # Execute the task based on current context
+        try:
+            loop = asyncio.get_running_loop()
+            logger.info(f'[PipelineExecutor] Found running loop {loop}')
+            # We're in an async context, use create_task
+            task = asyncio.create_task(_execute_task())
+            loop.run_until_complete(task)
+        except RuntimeError:
+            # No running loop, safe to use asyncio.run
+            logger.info('[PipelineExecutor] No running loop, using asyncio.run')
+            asyncio.run(_execute_task())
 
         self.logger_manager.output_logs_to_destination()
 

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/models/block/__init__.py

@@ -4464,29 +4464,36 @@ class CallbackBlock(AddonBlock):
             elif 'on_success' == callback:
                 callback_functions_legacy = success_functions
                 callback_status = CallbackStatus.SUCCESS
+            elif 'on_cancelled' == callback:
+                callback_functions_legacy = []
+                callback_status = CallbackStatus.CANCELLED
 
-            # Fetch input variables
-            input_vars, kwargs_vars, upstream_block_uuids = self.fetch_input_variables(
-                None,
-                dynamic_block_index=dynamic_block_index,
-                dynamic_block_indexes=dynamic_block_indexes,
-                dynamic_upstream_block_uuids=dynamic_upstream_block_uuids,
-                execution_partition=execution_partition,
-                from_notebook=from_notebook,
-                global_vars=global_vars,
-                metadata=metadata,
-                upstream_block_uuids=[parent_block.uuid] if parent_block else None,
-                upstream_block_uuids_override=upstream_block_uuids_override,
-            )
-
-            # Copied logic from the method self.execute_block
-            outputs_from_input_vars = {}
-            upstream_block_uuids_length = len(upstream_block_uuids)
-            for idx, input_var in enumerate(input_vars):
-                if idx < upstream_block_uuids_length:
-                    upstream_block_uuid = upstream_block_uuids[idx]
-                    outputs_from_input_vars[upstream_block_uuid] = input_var
-                    outputs_from_input_vars[f'df_{idx + 1}'] = input_var
+            if callback_status in [CallbackStatus.FAILURE, CallbackStatus.SUCCESS]:
+                # Fetch input variables
+                input_vars, kwargs_vars, upstream_block_uuids = self.fetch_input_variables(
+                    None,
+                    dynamic_block_index=dynamic_block_index,
+                    dynamic_block_indexes=dynamic_block_indexes,
+                    dynamic_upstream_block_uuids=dynamic_upstream_block_uuids,
+                    execution_partition=execution_partition,
+                    from_notebook=from_notebook,
+                    global_vars=global_vars,
+                    metadata=metadata,
+                    upstream_block_uuids=[parent_block.uuid] if parent_block else None,
+                    upstream_block_uuids_override=upstream_block_uuids_override,
+                )
+                # Copied logic from the method self.execute_block
+                outputs_from_input_vars = {}
+                upstream_block_uuids_length = len(upstream_block_uuids)
+                for idx, input_var in enumerate(input_vars):
+                    if idx < upstream_block_uuids_length:
+                        upstream_block_uuid = upstream_block_uuids[idx]
+                        outputs_from_input_vars[upstream_block_uuid] = input_var
+                        outputs_from_input_vars[f'df_{idx + 1}'] = input_var
+            else:
+                input_vars = []
+                kwargs_vars = []
+                upstream_block_uuids = []
 
             global_vars_copy = global_vars.copy()
             for kwargs_var in kwargs_vars:

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/models/block/outputs.py

@@ -343,7 +343,7 @@ def format_output_data(
                 columns=columns_to_display,
                 rows=[
                     list(row.values())
-                    for row in json.loads(data[columns_to_display].write_json(row_oriented=True))
+                    for row in json.loads(json.dumps(data[columns_to_display].to_dicts()))
                 ],
             ),
             resource_usage=resource_usage,

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/models/constants.py

@@ -82,6 +82,7 @@ class BlockColor(StrEnum):
 
 
 class CallbackStatus(StrEnum):
+    CANCELLED = 'cancelled'
     FAILURE = 'failure'
     SUCCESS = 'success'
 
@@ -126,6 +127,7 @@ BLOCK_LANGUAGE_TO_FILE_EXTENSION = {
 
 
 CALLBACK_STATUSES = [
+    CallbackStatus.CANCELLED,
     CallbackStatus.FAILURE,
     CallbackStatus.SUCCESS,
 ]

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/data_preparation/storage/local_storage.py

@@ -121,7 +121,10 @@ class LocalStorage(BaseStorage):
         return pd.read_parquet(file_path, engine='pyarrow')
 
     def read_polars_parquet(self, file_path: str, **kwargs) -> pl.DataFrame:
-        return pl.read_parquet(file_path, use_pyarrow=True)
+        try:
+            return pl.read_parquet(file_path, use_pyarrow=True)
+        except Exception:
+            return pl.read_parquet(file_path)
 
     def write_csv(self, df: pd.DataFrame, file_path: str) -> None:
         File.create_parent_directories(file_path)

{mage_ai-0.9.76 → mage_ai-0.9.77}/mage_ai/io/config.py

@@ -66,6 +66,7 @@ class ConfigKey(StrEnum):
     MONGODB_PORT = 'MONGODB_PORT'
     MONGODB_USER = 'MONGODB_USER'
 
+    MSSQL_AUTHENTICATION = 'MSSQL_AUTHENTICATION'
     MSSQL_DATABASE = 'MSSQL_DATABASE'
     MSSQL_DRIVER = 'MSSQL_DRIVER'
     MSSQL_HOST = 'MSSQL_HOST'