mage-ai 0.9.74__py3-none-any.whl → 0.9.76__py3-none-any.whl

mage_ai/tests/io/create_table/test_postgresql.py

@@ -1,7 +1,9 @@
 import pandas as pd
+import simplejson
 
 from mage_ai.io.export_utils import infer_dtypes
 from mage_ai.io.postgres import Postgres
+from mage_ai.shared.parsers import encode_complex
 from mage_ai.tests.base_test import DBTestCase
 
 
@@ -26,3 +28,28 @@ class TestTablePostgres(DBTestCase):
         )
         self.assertEqual('CREATE TABLE Test.Test ("varchar_time" text,"datetime_time" TIMESTAMP);',
                          query)
+
+    def test_clean_array_value(self):
+        test_cases = [
+            [],
+            [123],
+            [['abc', 'def']],
+            [['08:00', '12:00'], ['15:00', '20:00']],
+            [['08:00', '12:00'], []],
+        ]
+        expected = [
+            '{}',
+            '{123}',
+            '{{"abc", "def"}}',
+            '{{"08:00", "12:00"}, {"15:00", "20:00"}}',
+            '{{"08:00", "12:00"}, {}}',
+        ]
+        for val, expected_val in zip(test_cases, expected):
+            cleaned_array_value = Postgres._clean_array_value(
+                simplejson.dumps(
+                    val,
+                    default=encode_complex,
+                    ignore_nan=True,
+                )
+            )
+            self.assertEqual(cleaned_array_value, expected_val)

mage_ai/tests/orchestration/notification/test_config.py

@@ -2,10 +2,10 @@ from mage_ai.orchestration.notification.config import NotificationConfig
 from mage_ai.tests.base_test import TestCase
 from mage_ai.tests.orchestration.notification.constants import (
     EMAIL_NOTIFICATION_CONFIG,
+    GOOGLE_CHAT_NOTIFICATION_CONFIG,
+    OPSGENIE_NOTIFICATION_CONFIG,
     SLACK_NOTIFICATION_CONFIG,
     TEAMS_NOTIFICATION_CONFIG,
-    GOOGLE_CHAT_NOTIFICATION_CONFIG,
-    OPSGENIE_NOTIFICATION_CONFIG
 )
 
 
@@ -43,7 +43,7 @@ class NotificationConfigTests(TestCase):
         self.assertIsNone(config3.email_config)
         self.assertEqual(config3.slack_config.webhook_url, 'test_webhook_url')
 
-        self.assertEqual(config4.teams_config.webhook_url, 'test_webhook_url')
+        self.assertEqual(config4.teams_config.webhook_url, ['test_webhook_url'])
 
         self.assertEqual(config5.google_chat_config.webhook_url, 'test_webhook_url')
 

mage_ai/usage_statistics/logger.py

@@ -7,6 +7,7 @@ from typing import Callable, Dict, List, Union
 
 import aiohttp
 import pytz
+import requests
 
 from mage_ai.api.operations.constants import OperationType
 from mage_ai.cache.block_action_object.constants import (
@@ -43,6 +44,8 @@ from mage_ai.usage_statistics.constants import (
 )
 from mage_ai.usage_statistics.utils import build_event_data_for_chart
 
+AMPLITUDE_API_KEY = 'KwnbpKJNe6gOjC2X5ilxafFvxbNppiIfGejB2hlY'
+
 
 class UsageStatisticLogger():
     def __init__(
@@ -229,6 +232,17 @@ class UsageStatisticLogger():
             ),
         )
 
+    def pipeline_runs_impression_sync(self, count_func: Callable) -> bool:
+        if not self.help_improve_mage:
+            return False
+        return self.__send_message_sync(
+            dict(
+                object=EventObjectType.PIPELINE_RUN,
+                action=EventActionType.IMPRESSION,
+                pipeline_runs=count_func(),
+            ),
+        )
+
     @safe_db_query
     async def pipeline_runs_impression(self, count_func: Callable) -> bool:
         if not self.help_improve_mage:
@@ -301,19 +315,7 @@ class UsageStatisticLogger():
         )
 
     @safe_db_query
-    async def pipeline_run_ended(self, pipeline_run: PipelineRun) -> bool:
-        """
-        Write "pipeline_run_ended" event to Amplitude for the given PipelineRun.
-
-        Args:
-            pipeline_run (PipelineRun): pipeline run to use to populate the event
-
-        Returns:
-            bool: True if event was successfully uploaded
-        """
-        if not self.help_improve_mage:
-            return False
-
+    def pipeline_run_ended_data(self, pipeline_run: PipelineRun):
         pipeline = pipeline_run.pipeline
 
         if pipeline.type == PipelineType.INTEGRATION:
@@ -333,7 +335,7 @@ class UsageStatisticLogger():
 
         encoded_pipeline_uuid = pipeline.uuid.encode('utf-8')
         pipeline_schedule = pipeline_run.pipeline_schedule
-        data = dict(
+        return dict(
             landing_time_enabled=1 if pipeline_schedule.landing_time_enabled() else 0,
             num_pipeline_blocks=len(block_configs),
             pipeline_run_uuid=pipeline_run.id,
@@ -346,11 +348,48 @@ class UsageStatisticLogger():
             unique_languages=list(set([b.get('language') for b in block_configs])),
         )
 
+    @safe_db_query
+    async def pipeline_run_ended(self, pipeline_run: PipelineRun) -> bool:
+        """
+        Write "pipeline_run_ended" event to Amplitude for the given PipelineRun.
+
+        Args:
+            pipeline_run (PipelineRun): pipeline run to use to populate the event
+
+        Returns:
+            bool: True if event was successfully uploaded
+        """
+        if not self.help_improve_mage:
+            return False
+
+        data = self.pipeline_run_ended_data(pipeline_run)
+
         return await self.__send_message(
             data,
             event_name=EventNameType.PIPELINE_RUN_ENDED,
         )
 
+    @safe_db_query
+    def pipeline_run_ended_sync(self, pipeline_run: PipelineRun) -> bool:
+        """
+        Write "pipeline_run_ended" event to Amplitude for the given PipelineRun.
+
+        Args:
+            pipeline_run (PipelineRun): pipeline run to use to populate the event
+
+        Returns:
+            bool: True if event was successfully uploaded
+        """
+        if not self.help_improve_mage:
+            return False
+
+        data = self.pipeline_run_ended_data(pipeline_run)
+
+        return self.__send_message_sync(
+            data,
+            event_name=EventNameType.PIPELINE_RUN_ENDED,
+        )
+
     @safe_db_query
     async def block_run_ended(self, block_run: BlockRun) -> bool:
         if not self.help_improve_mage:
@@ -452,7 +491,7 @@ class UsageStatisticLogger():
                 async with session.post(
                     API_ENDPOINT,
                     json=dict(
-                        api_key='KwnbpKJNe6gOjC2X5ilxafFvxbNppiIfGejB2hlY',
+                        api_key=AMPLITUDE_API_KEY,
                         event_name=event_name,
                         usage_statistics=data_to_send,
                     ),
@@ -467,6 +506,51 @@ class UsageStatisticLogger():
             print(f'[Statistics] Message: {err}')
         return False
 
+    def __send_message_sync(
+        self,
+        data: Dict,
+        event_name: EventNameType = EventNameType.USAGE_STATISTIC_CREATE,
+        override_validation: bool = False,
+        project_uuid: str = None,
+    ) -> bool:
+        if not override_validation:
+            if is_test():
+                return False
+
+            if not self.help_improve_mage:
+                return False
+
+        if data is None:
+            data = {}
+
+        data_to_send = merge_dict(
+            self.__shared_metadata(),
+            data,
+        )
+
+        if project_uuid and not data_to_send.get('project_uuid'):
+            data_to_send['project_uuid'] = project_uuid
+
+        try:
+            response = requests.post(
+                API_ENDPOINT,
+                json={
+                    'api_key': AMPLITUDE_API_KEY,
+                    'event_name': event_name,
+                    'usage_statistics': data_to_send,
+                },
+                timeout=3
+            )
+            response.raise_for_status()
+            response_json = response.json()
+            if response_json.get('success'):
+                if is_debug():
+                    print(json.dumps(data_to_send, indent=2))
+                return True
+        except Exception as err:
+            print(f'[Statistics] Message: {err}')
+        return False
+
     def __shared_metadata(self) -> Dict:
         return dict(
             environment=get_env(),

mage_ai-0.9.76.dist-info/METADATA

@@ -0,0 +1,393 @@
+Metadata-Version: 2.4
+Name: mage-ai
+Version: 0.9.76
+Summary: Mage is a tool for building and deploying data pipelines.
+Home-page: https://github.com/mage-ai/mage-ai
+Author: Mage
+Author-email: eng@mage.ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cachetools
+Requires-Dist: Faker==4.14.0
+Requires-Dist: GitPython==3.1.41
+Requires-Dist: Jinja2==3.1.3
+Requires-Dist: Pillow==10.3.0
+Requires-Dist: PyGithub==1.59.0
+Requires-Dist: PyJWT==2.6.0
+Requires-Dist: aiofiles==22.1.0
+Requires-Dist: aiohttp==3.10.0
+Requires-Dist: alembic>=1.7.5
+Requires-Dist: bcrypt==4.0.1
+Requires-Dist: croniter==1.3.7
+Requires-Dist: cryptography==42.0.4
+Requires-Dist: dask>=2022.2.0
+Requires-Dist: datadog==0.44.0
+Requires-Dist: freezegun==1.2.2
+Requires-Dist: great-expectations==0.18.12
+Requires-Dist: httpx==0.25.0
+Requires-Dist: inflection==0.5.1
+Requires-Dist: ipykernel==6.15.0
+Requires-Dist: ipython==8.10.0
+Requires-Dist: itsdangerous~=1.1.0
+Requires-Dist: joblib>=1.1.0
+Requires-Dist: jupyter-server==2.14.1
+Requires-Dist: jupyter_client==7.4.4
+Requires-Dist: ldap3==2.9.1
+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: protobuf>=4.25.0
+Requires-Dist: psutil==5.9.8
+Requires-Dist: pyairtable>=2.3.3
+Requires-Dist: pyarrow>=14.0.1
+Requires-Dist: python-dateutil==2.8.2
+Requires-Dist: pytz>=2022.2.1
+Requires-Dist: pyyaml~=6.0
+Requires-Dist: redis~=5.0.1
+Requires-Dist: requests~=2.31.0
+Requires-Dist: ruamel.yaml==0.17.17
+Requires-Dist: scikit-learn>=1.0
+Requires-Dist: sentry-sdk==1.19.1
+Requires-Dist: setuptools==70.0.0
+Requires-Dist: simplejson
+Requires-Dist: six>=1.15.0
+Requires-Dist: sqlalchemy<2.0.0,>=1.4.20
+Requires-Dist: sqlglot[rs]
+Requires-Dist: terminado==0.17.1
+Requires-Dist: thefuzz[speedup]==0.19.0
+Requires-Dist: tornado==6.4.2
+Requires-Dist: typer[all]==0.9.0
+Requires-Dist: typing_extensions==4.10.0
+Requires-Dist: watchdog==4.0.0
+Requires-Dist: Werkzeug==3.0.3
+Provides-Extra: ai
+Requires-Dist: astor>=0.8.1; extra == "ai"
+Requires-Dist: langchain==0.1.6; extra == "ai"
+Requires-Dist: langchain_community<0.0.20; extra == "ai"
+Requires-Dist: openai>=1.0.0; extra == "ai"
+Provides-Extra: airtable
+Requires-Dist: pyairtable>=2.3.3; extra == "airtable"
+Provides-Extra: azure
+Requires-Dist: azure-eventhub==5.11.2; extra == "azure"
+Requires-Dist: azure-identity==1.12.0; extra == "azure"
+Requires-Dist: azure-keyvault-secrets==4.6.0; extra == "azure"
+Requires-Dist: azure-keyvault-certificates==4.6.0; extra == "azure"
+Requires-Dist: azure-mgmt-containerinstance==10.1.0; extra == "azure"
+Requires-Dist: azure-storage-blob==12.14.1; extra == "azure"
+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: chromadb>=0.4.17; extra == "chroma"
+Provides-Extra: clickhouse
+Requires-Dist: clickhouse-connect~=0.6.23; extra == "clickhouse"
+Provides-Extra: dbt
+Requires-Dist: dbt-adapters==1.8.0; extra == "dbt"
+Requires-Dist: dbt-bigquery==1.8.3; extra == "dbt"
+Requires-Dist: dbt-clickhouse==1.8.5; extra == "dbt"
+Requires-Dist: dbt-core==1.8.7; extra == "dbt"
+Requires-Dist: dbt-duckdb==1.8.4; extra == "dbt"
+Requires-Dist: dbt-postgres==1.8.2; extra == "dbt"
+Requires-Dist: dbt-redshift==1.8.1; extra == "dbt"
+Requires-Dist: dbt-snowflake==1.8.4; extra == "dbt"
+Requires-Dist: dbt-spark==1.8.0; extra == "dbt"
+Requires-Dist: dbt-sqlserver==1.8.4; extra == "dbt"
+Requires-Dist: dbt-synapse==1.8.2; extra == "dbt"
+Requires-Dist: dbt-trino==1.8.4; extra == "dbt"
+Requires-Dist: trino~=0.326; extra == "dbt"
+Provides-Extra: google-cloud-storage
+Requires-Dist: google-cloud-storage==2.5.0; extra == "google-cloud-storage"
+Requires-Dist: gspread==5.7.2; extra == "google-cloud-storage"
+Provides-Extra: hdf5
+Requires-Dist: tables==3.7.0; python_version < "3.11" and extra == "hdf5"
+Requires-Dist: tables==3.10.1; python_version >= "3.11" and extra == "hdf5"
+Provides-Extra: mysql
+Requires-Dist: mysql-connector-python~=8.4.0; python_version < "3.11" and extra == "mysql"
+Requires-Dist: mysql-connector-python~=9.0.0; python_version >= "3.11" and extra == "mysql"
+Provides-Extra: oracle
+Requires-Dist: oracledb==1.3.1; python_version < "3.12" and extra == "oracle"
+Requires-Dist: oracledb==2.4.1; python_version >= "3.12" and extra == "oracle"
+Provides-Extra: postgres
+Requires-Dist: psycopg2==2.9.3; extra == "postgres"
+Requires-Dist: psycopg2-binary==2.9.3; extra == "postgres"
+Requires-Dist: sshtunnel==0.4.0; extra == "postgres"
+Provides-Extra: qdrant
+Requires-Dist: qdrant-client==1.6.9; extra == "qdrant"
+Requires-Dist: sentence-transformers==2.2.2; extra == "qdrant"
+Provides-Extra: redshift
+Requires-Dist: boto3==1.26.60; extra == "redshift"
+Requires-Dist: redshift-connector==2.0.915; extra == "redshift"
+Requires-Dist: lxml==4.9.4; extra == "redshift"
+Provides-Extra: s3
+Requires-Dist: boto3==1.26.60; extra == "s3"
+Requires-Dist: botocore==1.29.60; extra == "s3"
+Provides-Extra: snowflake
+Requires-Dist: snowflake-connector-python==3.7.1; extra == "snowflake"
+Provides-Extra: spark
+Requires-Dist: boto3==1.26.60; extra == "spark"
+Requires-Dist: botocore==1.29.60; extra == "spark"
+Provides-Extra: streaming
+Requires-Dist: confluent-avro~=1.8.0; extra == "streaming"
+Requires-Dist: elasticsearch==8.15.1; extra == "streaming"
+Requires-Dist: influxdb_client==1.36.1; extra == "streaming"
+Requires-Dist: kafka-python==2.0.2; extra == "streaming"
+Requires-Dist: nats-py==2.6.0; extra == "streaming"
+Requires-Dist: nkeys~=0.2.0; extra == "streaming"
+Requires-Dist: opensearch-py==2.0.0; extra == "streaming"
+Requires-Dist: pika==1.3.1; extra == "streaming"
+Requires-Dist: pymongo==4.3.3; extra == "streaming"
+Requires-Dist: requests_aws4auth==1.1.2; extra == "streaming"
+Requires-Dist: stomp.py==8.1.0; extra == "streaming"
+Provides-Extra: all
+Requires-Dist: PyGithub==1.59.0; extra == "all"
+Requires-Dist: astor>=0.8.1; extra == "all"
+Requires-Dist: aws-secretsmanager-caching==1.1.1.5; extra == "all"
+Requires-Dist: azure-eventhub==5.11.2; extra == "all"
+Requires-Dist: azure-identity==1.12.0; extra == "all"
+Requires-Dist: azure-keyvault-certificates==4.6.0; extra == "all"
+Requires-Dist: azure-keyvault-secrets==4.6.0; extra == "all"
+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: clickhouse-connect~=0.6.23; extra == "all"
+Requires-Dist: confluent-avro~=1.8.0; extra == "all"
+Requires-Dist: db-dtypes==1.0.5; extra == "all"
+Requires-Dist: dbt-adapters==1.8.0; extra == "all"
+Requires-Dist: dbt-bigquery==1.8.3; extra == "all"
+Requires-Dist: dbt-clickhouse==1.8.5; extra == "all"
+Requires-Dist: dbt-core==1.8.7; extra == "all"
+Requires-Dist: dbt-duckdb==1.8.4; extra == "all"
+Requires-Dist: dbt-postgres==1.8.2; extra == "all"
+Requires-Dist: dbt-redshift==1.8.1; extra == "all"
+Requires-Dist: dbt-snowflake==1.8.4; extra == "all"
+Requires-Dist: dbt-spark==1.8.0; extra == "all"
+Requires-Dist: dbt-sqlserver==1.8.4; extra == "all"
+Requires-Dist: dbt-synapse==1.8.2; extra == "all"
+Requires-Dist: dbt-trino==1.8.4; extra == "all"
+Requires-Dist: duckdb==1.0.0; extra == "all"
+Requires-Dist: elasticsearch==8.15.1; extra == "all"
+Requires-Dist: google-api-core~=2.15.0; extra == "all"
+Requires-Dist: google-api-python-client~=2.70.0; extra == "all"
+Requires-Dist: google-cloud-bigquery~=3.14.1; extra == "all"
+Requires-Dist: google-cloud-iam~=2.13.0; extra == "all"
+Requires-Dist: google-cloud-pubsub~=2.19.0; extra == "all"
+Requires-Dist: google-cloud-run~=0.10.1; extra == "all"
+Requires-Dist: google-cloud-storage~=2.5.0; extra == "all"
+Requires-Dist: great-expectations==0.18.12; extra == "all"
+Requires-Dist: gspread==5.7.2; extra == "all"
+Requires-Dist: influxdb_client==1.36.1; extra == "all"
+Requires-Dist: kafka-python==2.0.2; extra == "all"
+Requires-Dist: kubernetes>=28.1.0; extra == "all"
+Requires-Dist: langchain==0.1.6; extra == "all"
+Requires-Dist: langchain_community<0.0.20; extra == "all"
+Requires-Dist: ldap3==2.9.1; extra == "all"
+Requires-Dist: nats-py==2.6.0; extra == "all"
+Requires-Dist: nkeys~=0.2.0; extra == "all"
+Requires-Dist: openai>=1.0.0; extra == "all"
+Requires-Dist: opensearch-py==2.0.0; extra == "all"
+Requires-Dist: opentelemetry-api==1.22.0; extra == "all"
+Requires-Dist: opentelemetry-exporter-prometheus==0.43b0; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-tornado==0.43b0; extra == "all"
+Requires-Dist: opentelemetry-exporter-otlp==1.22.0; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy==0.43b0; extra == "all"
+Requires-Dist: oracledb==1.3.1; extra == "all"
+Requires-Dist: pika==1.3.1; extra == "all"
+Requires-Dist: pinotdb==5.6.0; extra == "all"
+Requires-Dist: prometheus_client>=0.18.0; extra == "all"
+Requires-Dist: protobuf>=4.25.0; extra == "all"
+Requires-Dist: psycopg2-binary==2.9.3; extra == "all"
+Requires-Dist: psycopg2==2.9.3; extra == "all"
+Requires-Dist: pyairtable==2.3.3; extra == "all"
+Requires-Dist: pydruid==0.6.5; extra == "all"
+Requires-Dist: pymongo==4.3.3; extra == "all"
+Requires-Dist: pyodbc==4.0.35; python_version < "3.12" and extra == "all"
+Requires-Dist: pyodbc==5.0.1; python_version >= "3.12" and extra == "all"
+Requires-Dist: redshift-connector==2.0.915; extra == "all"
+Requires-Dist: lxml==4.9.4; extra == "all"
+Requires-Dist: requests_aws4auth==1.1.2; extra == "all"
+Requires-Dist: snowflake-connector-python==3.7.1; extra == "all"
+Requires-Dist: sshtunnel==0.4.0; extra == "all"
+Requires-Dist: stomp.py==8.1.0; extra == "all"
+Requires-Dist: thefuzz[speedup]==0.19.0; extra == "all"
+Requires-Dist: trino~=0.326; extra == "all"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+<div align="left">
+  <h1>Mage AI</h1>
+
+  <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>
+
+<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>
+
+<br />
+<br />
+
+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.
+
+<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.
+
+<br />
+
+Plus hundreds of enterprise-class features, infrastructure innovations, and magical surprises.
+
+<h1></h1>
+
+#### Available in two spellbinding versions
+
+<br />
+
+<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>
+
+<br />
+
+<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>
+
+<br />
+<br />
+
+<div align="left">
+
+# It’s magic.
+
+<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>
+
+<br />
+
+## 🏃‍♀️ Install
+
+The recommended way to install the latest version of Mage is through Docker with the following command:
+
+```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.
+
+```bash
+pip install mage-ai
+```
+```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).
+
+## 🎮 Demo
+
+### Live demo
+
+Build and run a data pipeline with our <b>[demo app](https://demo.mage.ai/)</b>.
+
+> WARNING
+>
+> The live demo is public to everyone, please don’t save anything sensitive (e.g. passwords, secrets, etc).
+### Demo video (5 min)
+
+[![Mage quick start demo](https://github.com/mage-ai/assets/blob/main/overview/overview-video.png?raw=True)](https://youtu.be/GswOdShLGmg)
+
+<sub><i>Click the image to play video</i></sub>
+
+<br />
+
+## 🔮 [Features](https://docs.mage.ai/about/features)
+
+|   |   |   |
+| --- | --- | --- |
+| 🎶 | <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. |
+
+<br />
+
+<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')
+    ```
+
+<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/)