PyPI - mage-ai - Versions diffs - 0.9.75__py3-none-any.whl → 0.9.77__py3-none-any.whl - Mend

mage-ai 0.9.75py3-none-any.whl → 0.9.77py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of mage-ai might be problematic. Click here for more details.

Files changed (226) hide show

mage_ai/server/frontend_dist_base_path_template/version-control.html CHANGED Viewed

@@ -27,7 +27,7 @@
         };
     }
 })();
-              </script><link rel="preload" href="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/css/18782af6d2c4e826.css" as="style"/><link rel="stylesheet" href="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/css/18782af6d2c4e826.css" data-n-g=""/><noscript data-n-css=""></noscript><script defer="" nomodule="" src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/polyfills-c67a75d1b6f99dc8.js"></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/webpack-12ad70eb5c31aa92.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/framework-ea07653270784974.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/main-342bedfde9cd81cf.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/pages/_app-0392ef723ea2c6f8.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/449-f689774546860ca4.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/pages/version-control-573f0225d7a703ed.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/dxnSzgIvSG4Ke5LM-tPQX/_buildManifest.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/dxnSzgIvSG4Ke5LM-tPQX/_ssgManifest.js" defer=""></script><style data-styled="" data-styled-version="5.3.11">html{-webkit-box-sizing:border-box;box-sizing:border-box;-ms-overflow-style:scrollbar;}/*!sc*/
+              </script><link rel="preload" href="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/css/18782af6d2c4e826.css" as="style"/><link rel="stylesheet" href="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/css/18782af6d2c4e826.css" data-n-g=""/><noscript data-n-css=""></noscript><script defer="" nomodule="" src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/polyfills-c67a75d1b6f99dc8.js"></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/webpack-5f4be622608d9267.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/framework-ea07653270784974.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/main-342bedfde9cd81cf.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/pages/_app-ee5e328aaf51c698.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/449-5e2253c6aba42557.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/chunks/pages/version-control-573f0225d7a703ed.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/iCySon3_GCldnbC5U7C-s/_buildManifest.js" defer=""></script><script src="/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_/_next/static/iCySon3_GCldnbC5U7C-s/_ssgManifest.js" defer=""></script><style data-styled="" data-styled-version="5.3.11">html{-webkit-box-sizing:border-box;box-sizing:border-box;-ms-overflow-style:scrollbar;}/*!sc*/
 *,*::before,*::after{-webkit-box-sizing:inherit;box-sizing:inherit;}/*!sc*/
 data-styled.g4[id="sc-global-czSCUT1"]{content:"sc-global-czSCUT1,"}/*!sc*/
 .esCWvc{display:-webkit-box;display:-webkit-flex;display:-ms-flexbox;display:flex;-webkit-flex:1;-ms-flex:1;flex:1;-webkit-flex-direction:column;-ms-flex-direction:column;flex-direction:column;}/*!sc*/
@@ -50,4 +50,4 @@ data-styled.g242[id="indexstyle__VerticalNavigationStyleComponent-sc-ecogjt-1"]{
 .dionUf .Toastify__toast--success{background:#00A81A !important;color:#FFFFFF !important;}/*!sc*/
 .dionUf .Toastify__toast--warning{background:#DD9900 !important;color:#FFFFFF !important;}/*!sc*/
 data-styled.g279[id="ToastWrapper-sc-1a33ph1-0"]{content:"dionUf,"}/*!sc*/
-</style></head><body><div id="__next"><div class="" style="position:fixed;top:0;left:0;height:2px;background:transparent;z-index:99999999999;width:100%"><div class="" style="height:100%;background:#FF144D;transition:all 500ms ease;width:0%"><div style="box-shadow:0 0 10px #FF144D, 0 0 10px #FF144D;width:5%;opacity:1;position:absolute;height:100%;transition:all 500ms ease;transform:rotate(3deg) translate(0px, -4px);left:-10rem"></div></div></div><div class="indexstyle__HeaderStyle-sc-1bk8irg-0 bVoYMS"></div><div class="indexstyle__ContainerStyle-sc-ecogjt-0 jrjFfx"><div class="indexstyle__VerticalNavigationStyleComponent-sc-ecogjt-1 drmgmw"><div class="Spacing__SpacingStyle-sc-1mpmtgj-0 kgseVY"></div></div><div class="Flex-sc-sgfnl9-0 esCWvc"></div></div><div id="command-center-root"></div><div></div><div></div><div></div><div class="ToastWrapper-sc-1a33ph1-0 dionUf"><div class="Toastify"></div></div></div><script id="__NEXT_DATA__" type="application/json">{"props":{"pageProps":{"auth":{"decodedToken":{"expires":0,"token":null}}},"currentTheme":{"accent":{"alert":"#F6540B","blue":"#4877FF","blueLight":"rgba(72, 119, 255, 0.5)","contentDefaultTransparent":"rgba(174, 174, 174, 0.5)","cyan":"#65E3FF","cyanLight":"rgba(101, 227, 255, 0.3)","cyanTransparent":"rgba(101, 227, 255, 0.12)","dbt":"#fc6949","dbtDark":"rgba(252, 105, 73, 0.3)","dbtLight":"rgba(252, 105, 73, 0.5)","info":"#00ABFF","infoTransparent":"rgba(0, 171, 255, 0.5)","negative":"#FF1E59","negativeTransparent":"rgba(255, 30, 89, 0.3)","pink":"#FF4FF8","pinkLight":"rgb(255, 79, 248, 0.5)","positive":"#00A81A","primaryTransparent":"rgba(155, 108, 167, 0.5)","purple":"#7D55EC","purpleLight":"rgba(125, 85, 236, 0.5)","rose":"#D1A2AB","roseLight":"rgba(209, 162, 171, 0.5)","sky":"#6AA1E0","skyLight":"rgba(106, 161, 224, 0.5)","teal":"#00B4CC","tealLight":"rgba(0, 180, 204, 0.5)","warning":"#DD9900","warningTransparent":"rgba(221, 153, 0, 0.5)","yellow":"#FFCC19","yellowLight":"rgba(255, 204, 25, 0.5)"},"background":{"blackTransparent":"rgba(0, 0, 0, 0.6)","blackTransparentDark":"rgba(0, 0, 0, 0.8)","chartBlock":"#2E3036","codeArea":"#1E1F24","codeTextarea":"#000000","content":"#1B1C20","danger":"#FFD0DB","dark":"#B1B8C3","dashboard":"#18181C","dashboardTransparent":"rgba(24, 24, 28, 0.1)","header":"#1B1B1B","menu":"#0F4CFF","muted":"#F9FAFC","navigation":"#EDEDED","output":"#2E3036","page":"#1E1F24","panel":"#232429","panelTransparent":"rgba(35, 36, 41, 0.7)","popup":"#27292E","row":"#2C2C2C","row2":"#51535C","scrollbarThumb":"rgba(100, 100, 100, 0.5)","scrollbarThumbHover":"rgba(255, 255, 255, 0.3)","scrollbarTrack":"#2E3036","success":"#8ADE00","successLight":"rgb(138, 222, 0, 0.3)","table":"#292A2F","tag":"#3A4550"},"borders":{"button":"#454850","contrast":"#FFFFFF","danger":"#FF144D","dark":"#000000","darkLight":"#2E3036","info":"#FFCC19","light":"#2F3034","medium":"#1C1C1C","medium2":"#141414","success":"#2FCB52"},"brand":{"earth100":"#C6EEDB","earth200":"#9DDFBF","earth300":"#6BBF96","earth400":"#37A46F","earth400Transparent":"rgba(55, 164, 111, 0.4)","earth500":"#00954C","energy100":"#FFF4BA","energy200":"#FFED92","energy300":"#FFE662","energy400":"#FFDA19","energy400Transparent":"rgba(255, 218, 25, 0.04)","energy500":"#F6C000","fire100":"#FFD7E0","fire200":"#FFA3B9","fire300":"#FF547D","fire400":"#FF144D","fire400Transparent":"rgba(255, 20, 77, 0.4)","fire500":"#EB0032","stone100":"#F3E6D7","stone200":"#E3D4C2","stone400":"#BFA78B","stone500":"#AF8859","water100":"#BDCEFF","water200":"#81A1FF","water300":"#517DFF","water400":"#2A60FE","water400Transparent":"rgba(42, 96, 254, 0.4)","water500":"#0F4CFF","wind100":"#EEEAFF","wind200":"#CCC1F4","wind300":"#A698DD","wind400":"#6B50D7","wind400SuperTransparent":"rgba(107, 80, 215, 0.12)","wind400Transparent":"rgba(107, 80, 215, 0.4)","wind500":"#4E32BC"},"chart":{"backgroundPrimary":"#7D55EC","backgroundSecondary":"#FF144D","backgroundTertiary":"#86E2FF","button1":"#4877FF","button2":"#FFCC19","button3":"#8ADE00","button4":"#FF4FF8","button5":"#B98D95","lines":"#9B6CA7","primary":"#6B50D7","secondary":"#FF144D","tertiary":"#2A60FE"},"content":{"active":"#FFFFFF","default":"#AEAEAE","disabled":"rgba(255, 255, 255, 0.3)","inverted":"#2C2C2C","muted":"#787A85"},"elevation":{"visualizationAccent":"#996CFF","visualizationAccentAlt":"#C1ACF7"},"feature":{"active":"rgba(250, 248, 254, 0.14)","disabled":"rgba(201, 206, 218, 0.12)"},"icons":{"neutral":"#787878"},"interactive":{"activeBorder":"#060606","blackBackgroundTransparent":"rgba(0, 0, 0, 0.5)","checked":"#060606","dangerBorder":"#FF144D","defaultBackground":"#36383F","defaultBackgroundTransparent":"rgba(54, 56, 63, 0.5)","defaultBorder":"#ffffff1a","disabledBorder":"#B1B8C3","focusBackground":"#B1B8C3","focusBorder":"#86E2FF","hoverBackground":"#4E4E4E","hoverBackgroundTransparent":"rgba(78, 78, 78, 0.5)","hoverBlackBackgroundTransparent":"rgba(0, 0, 0, 0.7)","hoverBorder":"#B9BFCA","hoverOverlay":"rgba(255, 255, 255, 0.1)","linkPrimary":"#1752FF","linkPrimaryHover":"#4877FF","linkPrimaryLight":"#5982ff","linkSecondary":"#6B50D7","linkSecondaryDisabled":"#C4B9EF","linkText":"#6AA1E0","linkTextLight":"#9ECBFF","purple":"#885EFF","rowHoverBackground":"rgba(0, 0, 0, 0.1)","transparent":"rgba(255, 255, 255, 0)"},"loader":{"color":"#EB0032","colorInverted":"#8ADE00"},"logo":{"color":"#FFFFFF"},"monotone":{"black":"#060606","blackTransparent":"rgba(0, 0, 0, 0.6)","gray":"#B1B8C3","grey100":"#F2F2F2","grey200":"#D5D7DC","grey300":"#B4B8C0","grey400":"#70747C","grey500":"#51535C","purple":"#6B50D7","white":"#FFFFFF","whiteTransparent":"rgba(255, 255, 255, 0.6)"},"neutral":{"n100":"#E7E8EA","n200":"#D8DADE","n300":"#CBCCD0","n400":"#BCBEC4","n500":"#AEB0B6"},"progress":{"negative":"#FF144D","positive":"#6B50D7"},"shadow":{"base":"12px 40px 120px rgba(0, 0, 0, 0.3)","frame":"0px 10px 40px rgba(0, 0, 0, 0.26)","menu":"4px 10px 20px rgba(6, 6, 6, 0.12)","popup":"10px 20px 40px rgba(0, 0, 0, 0.2)","small":"0px, 4px, rgba(0, 0, 0, 0.25)","window":"0px 10px 60px rgba(0, 0, 0, 0.7)"},"status":{"negative":"#FF144D","positive":"#24B400"},"text":{"fileBrowser":"#787A85"}}},"page":"/version-control","query":{},"buildId":"dxnSzgIvSG4Ke5LM-tPQX","assetPrefix":"/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_","nextExport":true,"isFallback":false,"gip":true,"appGip":true,"scriptLoader":[]}</script></body></html>
+</style></head><body><div id="__next"><div class="" style="position:fixed;top:0;left:0;height:2px;background:transparent;z-index:99999999999;width:100%"><div class="" style="height:100%;background:#FF144D;transition:all 500ms ease;width:0%"><div style="box-shadow:0 0 10px #FF144D, 0 0 10px #FF144D;width:5%;opacity:1;position:absolute;height:100%;transition:all 500ms ease;transform:rotate(3deg) translate(0px, -4px);left:-10rem"></div></div></div><div class="indexstyle__HeaderStyle-sc-1bk8irg-0 bVoYMS"></div><div class="indexstyle__ContainerStyle-sc-ecogjt-0 jrjFfx"><div class="indexstyle__VerticalNavigationStyleComponent-sc-ecogjt-1 drmgmw"><div class="Spacing__SpacingStyle-sc-1mpmtgj-0 kgseVY"></div></div><div class="Flex-sc-sgfnl9-0 esCWvc"></div></div><div id="command-center-root"></div><div></div><div></div><div></div><div class="ToastWrapper-sc-1a33ph1-0 dionUf"><div class="Toastify"></div></div></div><script id="__NEXT_DATA__" type="application/json">{"props":{"pageProps":{"auth":{"decodedToken":{"expires":0,"token":null}}},"currentTheme":{"accent":{"alert":"#F6540B","blue":"#4877FF","blueLight":"rgba(72, 119, 255, 0.5)","contentDefaultTransparent":"rgba(174, 174, 174, 0.5)","cyan":"#65E3FF","cyanLight":"rgba(101, 227, 255, 0.3)","cyanTransparent":"rgba(101, 227, 255, 0.12)","dbt":"#fc6949","dbtDark":"rgba(252, 105, 73, 0.3)","dbtLight":"rgba(252, 105, 73, 0.5)","info":"#00ABFF","infoTransparent":"rgba(0, 171, 255, 0.5)","negative":"#FF1E59","negativeTransparent":"rgba(255, 30, 89, 0.3)","pink":"#FF4FF8","pinkLight":"rgb(255, 79, 248, 0.5)","positive":"#00A81A","primaryTransparent":"rgba(155, 108, 167, 0.5)","purple":"#7D55EC","purpleLight":"rgba(125, 85, 236, 0.5)","rose":"#D1A2AB","roseLight":"rgba(209, 162, 171, 0.5)","sky":"#6AA1E0","skyLight":"rgba(106, 161, 224, 0.5)","teal":"#00B4CC","tealLight":"rgba(0, 180, 204, 0.5)","warning":"#DD9900","warningTransparent":"rgba(221, 153, 0, 0.5)","yellow":"#FFCC19","yellowLight":"rgba(255, 204, 25, 0.5)"},"background":{"blackTransparent":"rgba(0, 0, 0, 0.6)","blackTransparentDark":"rgba(0, 0, 0, 0.8)","chartBlock":"#2E3036","codeArea":"#1E1F24","codeTextarea":"#000000","content":"#1B1C20","danger":"#FFD0DB","dark":"#B1B8C3","dashboard":"#18181C","dashboardTransparent":"rgba(24, 24, 28, 0.1)","header":"#1B1B1B","menu":"#0F4CFF","muted":"#F9FAFC","navigation":"#EDEDED","output":"#2E3036","page":"#1E1F24","panel":"#232429","panelTransparent":"rgba(35, 36, 41, 0.7)","popup":"#27292E","row":"#2C2C2C","row2":"#51535C","scrollbarThumb":"rgba(100, 100, 100, 0.5)","scrollbarThumbHover":"rgba(255, 255, 255, 0.3)","scrollbarTrack":"#2E3036","success":"#8ADE00","successLight":"rgb(138, 222, 0, 0.3)","table":"#292A2F","tag":"#3A4550"},"borders":{"button":"#454850","contrast":"#FFFFFF","danger":"#FF144D","dark":"#000000","darkLight":"#2E3036","info":"#FFCC19","light":"#2F3034","medium":"#1C1C1C","medium2":"#141414","success":"#2FCB52"},"brand":{"earth100":"#C6EEDB","earth200":"#9DDFBF","earth300":"#6BBF96","earth400":"#37A46F","earth400Transparent":"rgba(55, 164, 111, 0.4)","earth500":"#00954C","energy100":"#FFF4BA","energy200":"#FFED92","energy300":"#FFE662","energy400":"#FFDA19","energy400Transparent":"rgba(255, 218, 25, 0.04)","energy500":"#F6C000","fire100":"#FFD7E0","fire200":"#FFA3B9","fire300":"#FF547D","fire400":"#FF144D","fire400Transparent":"rgba(255, 20, 77, 0.4)","fire500":"#EB0032","stone100":"#F3E6D7","stone200":"#E3D4C2","stone400":"#BFA78B","stone500":"#AF8859","water100":"#BDCEFF","water200":"#81A1FF","water300":"#517DFF","water400":"#2A60FE","water400Transparent":"rgba(42, 96, 254, 0.4)","water500":"#0F4CFF","wind100":"#EEEAFF","wind200":"#CCC1F4","wind300":"#A698DD","wind400":"#6B50D7","wind400SuperTransparent":"rgba(107, 80, 215, 0.12)","wind400Transparent":"rgba(107, 80, 215, 0.4)","wind500":"#4E32BC"},"chart":{"backgroundPrimary":"#7D55EC","backgroundSecondary":"#FF144D","backgroundTertiary":"#86E2FF","button1":"#4877FF","button2":"#FFCC19","button3":"#8ADE00","button4":"#FF4FF8","button5":"#B98D95","lines":"#9B6CA7","primary":"#6B50D7","secondary":"#FF144D","tertiary":"#2A60FE"},"content":{"active":"#FFFFFF","default":"#AEAEAE","disabled":"rgba(255, 255, 255, 0.3)","inverted":"#2C2C2C","muted":"#787A85"},"elevation":{"visualizationAccent":"#996CFF","visualizationAccentAlt":"#C1ACF7"},"feature":{"active":"rgba(250, 248, 254, 0.14)","disabled":"rgba(201, 206, 218, 0.12)"},"icons":{"neutral":"#787878"},"interactive":{"activeBorder":"#060606","blackBackgroundTransparent":"rgba(0, 0, 0, 0.5)","checked":"#060606","dangerBorder":"#FF144D","defaultBackground":"#36383F","defaultBackgroundTransparent":"rgba(54, 56, 63, 0.5)","defaultBorder":"#ffffff1a","disabledBorder":"#B1B8C3","focusBackground":"#B1B8C3","focusBorder":"#86E2FF","hoverBackground":"#4E4E4E","hoverBackgroundTransparent":"rgba(78, 78, 78, 0.5)","hoverBlackBackgroundTransparent":"rgba(0, 0, 0, 0.7)","hoverBorder":"#B9BFCA","hoverOverlay":"rgba(255, 255, 255, 0.1)","linkPrimary":"#1752FF","linkPrimaryHover":"#4877FF","linkPrimaryLight":"#5982ff","linkSecondary":"#6B50D7","linkSecondaryDisabled":"#C4B9EF","linkText":"#6AA1E0","linkTextLight":"#9ECBFF","purple":"#885EFF","rowHoverBackground":"rgba(0, 0, 0, 0.1)","transparent":"rgba(255, 255, 255, 0)"},"loader":{"color":"#EB0032","colorInverted":"#8ADE00"},"logo":{"color":"#FFFFFF"},"monotone":{"black":"#060606","blackTransparent":"rgba(0, 0, 0, 0.6)","gray":"#B1B8C3","grey100":"#F2F2F2","grey200":"#D5D7DC","grey300":"#B4B8C0","grey400":"#70747C","grey500":"#51535C","purple":"#6B50D7","white":"#FFFFFF","whiteTransparent":"rgba(255, 255, 255, 0.6)"},"neutral":{"n100":"#E7E8EA","n200":"#D8DADE","n300":"#CBCCD0","n400":"#BCBEC4","n500":"#AEB0B6"},"progress":{"negative":"#FF144D","positive":"#6B50D7"},"shadow":{"base":"12px 40px 120px rgba(0, 0, 0, 0.3)","frame":"0px 10px 40px rgba(0, 0, 0, 0.26)","menu":"4px 10px 20px rgba(6, 6, 6, 0.12)","popup":"10px 20px 40px rgba(0, 0, 0, 0.2)","small":"0px, 4px, rgba(0, 0, 0, 0.25)","window":"0px 10px 60px rgba(0, 0, 0, 0.7)"},"status":{"negative":"#FF144D","positive":"#24B400"},"text":{"fileBrowser":"#787A85"}}},"page":"/version-control","query":{},"buildId":"iCySon3_GCldnbC5U7C-s","assetPrefix":"/CLOUD_NOTEBOOK_BASE_PATH_PLACEHOLDER_","nextExport":true,"isFallback":false,"gip":true,"appGip":true,"scriptLoader":[]}</script></body></html>

mage_ai/server/scheduler_manager.py CHANGED Viewed

@@ -12,6 +12,7 @@ from mage_ai.server.logger import Logger
 from mage_ai.services.newrelic import initialize_new_relic
 from mage_ai.settings import (
     SENTRY_DSN,
+    SENTRY_SERVER_NAME,
     SENTRY_TRACES_SAMPLE_RATE,
     SERVER_LOGGING_FORMAT,
     SERVER_VERBOSITY,
@@ -35,6 +36,7 @@ def run_scheduler():
         sentry_sdk.init(
             sentry_dsn,
             traces_sample_rate=SENTRY_TRACES_SAMPLE_RATE,
+            server_name=SENTRY_SERVER_NAME,
         )
     (enable_new_relic, application) = initialize_new_relic()
     try:

mage_ai/server/terminal_server.py CHANGED Viewed

@@ -126,6 +126,9 @@ class TerminalWebsocketServer(terminado.TermSocket):
                             Entity.PROJECT,
                             get_project_uuid(),
                         )
+                    else:
+                        # If the user is deleted, set valid to False
+                        valid = False
             if not valid or is_disable_pipeline_edit_access():
                 return self.send_json_message(
                     ['stdout', f'{command[1]}\nUnauthorized access to the terminal.'])

mage_ai/settings/server.py CHANGED Viewed

@@ -121,6 +121,8 @@ CONCURRENCY_CONFIG_PIPELINE_RUN_LIMIT = get_int_value(
 )
 DISABLE_AUTO_BROWSER_OPEN = get_bool_value(os.getenv('DISABLE_AUTO_BROWSER_OPEN', 'False'))
 DISABLE_AUTORELOAD = get_bool_value(os.getenv('DISABLE_AUTORELOAD', 'False'))
+HIDE_API_TRIGGER_TOKEN = get_bool_value(os.getenv('HIDE_API_TRIGGER_TOKEN', 'False'))
 # The hostname in Kubernetes or AWS ECS
 HOSTNAME = os.getenv('HOSTNAME')
 INITIAL_METADATA = os.getenv('INITIAL_METADATA')
@@ -152,6 +154,7 @@ USE_UNIQUE_TERMINAL = os.getenv('USE_UNIQUE_TERMINAL', None)
 # Sentry Configuration
 SENTRY_DSN = os.getenv('SENTRY_DSN')
 SENTRY_TRACES_SAMPLE_RATE = os.getenv('SENTRY_TRACES_SAMPLE_RATE', 1.0)
+SENTRY_SERVER_NAME = os.getenv('SENTRY_SERVER_NAME', None)
 # New Relic Configuration
 ENABLE_NEW_RELIC = os.getenv('ENABLE_NEW_RELIC', False)
@@ -263,6 +266,7 @@ MAGE_SETTINGS_ENVIRONMENT_VARIABLES = [
     'USE_UNIQUE_TERMINAL',
     'SENTRY_DSN',
     'SENTRY_TRACES_SAMPLE_RATE',
+    'SENTRY_SERVER_NAME',
     'MAGE_PUBLIC_HOST',
     'SCHEDULER_TRIGGER_INTERVAL',
     'REQUIRE_USER_PERMISSIONS',

mage_ai/streaming/sources/kafka.py CHANGED Viewed

@@ -91,7 +91,7 @@ class KafkaSource(BaseSource):
             api_version=self.config.api_version,
             auto_offset_reset=self.config.auto_offset_reset,
             max_partition_fetch_bytes=self.config.max_partition_fetch_bytes,
-            enable_auto_commit=True,
+            enable_auto_commit=False,
         )
         if self.config.security_protocol == SecurityProtocol.SSL:
             consumer_kwargs['security_protocol'] = SecurityProtocol.SSL
@@ -270,6 +270,7 @@ class KafkaSource(BaseSource):
                     message_values.append(message)
             if len(message_values) > 0:
                 handler(message_values)
+            self.consumer.commit()
     def test_connection(self):
         self.consumer._client.check_version(timeout=5)

mage_ai/tests/data_preparation/executors/test_block_executor.py CHANGED Viewed

@@ -98,7 +98,7 @@ class BlockExecutorTest(BaseApiTestCase):
         self.block_executor._execute_conditional = MagicMock(return_value=True)
         self.block_executor._execute = MagicMock(return_value={'result': 'success'})
         # self.block.run_tests = MagicMock()
-        self.block_executor._execute_callback = MagicMock()
+        self.block_executor.execute_callback = MagicMock()
         result = self.block_executor.execute(
             analyze_outputs=analyze_outputs,
@@ -167,7 +167,7 @@ class BlockExecutorTest(BaseApiTestCase):
         #     update_tests=False,
         #     dynamic_block_uuid=dynamic_block_uuid,
         # )
-        self.block_executor._execute_callback.assert_called_with(
+        self.block_executor.execute_callback.assert_called_with(
             'on_success',
             block_run_id=None,
             callback_kwargs=dict(retry=dict(attempts=1)),
@@ -245,7 +245,7 @@ class BlockExecutorTest(BaseApiTestCase):
         self.block.callback_blocks = [MagicMock(), MagicMock()]
         self.block.callback_block = MagicMock()
-        self.block_executor._execute_callback(
+        self.block_executor.execute_callback(
             callback='on_success',
             global_vars=dict(retry=dict(attempts=1)),
             logging_tags={},

mage_ai/tests/data_preparation/models/block/dynamic/test_counter.py CHANGED Viewed

@@ -1,14 +1,12 @@
 import os
 from unittest.mock import patch
-from mage_ai.data_preparation.models.block.dynamic.constants import (
-    CHILD_DATA_VARIABLE_UUID,
-)
 from mage_ai.data_preparation.models.block.dynamic.counter import (
     DynamicBlockItemCounter,
     DynamicChildItemCounter,
     DynamicDuoItemCounter,
 )
+from mage_ai.data_preparation.models.constants import CHILD_DATA_VARIABLE_UUID
 from mage_ai.data_preparation.models.variables.constants import (
     VariableAggregateDataTypeFilename,
 )

mage_ai/tests/data_preparation/models/test_variable.py CHANGED Viewed

@@ -180,6 +180,7 @@ class VariableTest(DBTestCase):
                 [1, 'test'],
                 [2, 'test2'],
             ],
+            orient="row",
             schema=['col1', 'col2'],
         )
         df2 = pl.DataFrame(
@@ -187,6 +188,7 @@ class VariableTest(DBTestCase):
                 [1, 'test', 3.123, 41414123123124],
                 [2, 'test2', 4.321, 12111111],
             ],
+            orient="row",
             schema=['col1', 'col2', 'col3', 'col4'],
         )
         df2 = df2.cast({'col4': pl.Int64})

mage_ai/tests/io/create_table/test_postgresql.py CHANGED Viewed

@@ -33,14 +33,14 @@ class TestTablePostgres(DBTestCase):
         test_cases = [
             [],
             [123],
-            [['abc', 'def']],
+            [['àabc', 'deèéf']],
             [['08:00', '12:00'], ['15:00', '20:00']],
             [['08:00', '12:00'], []],
         ]
         expected = [
             '{}',
             '{123}',
-            '{{"abc", "def"}}',
+            '{{"àabc", "deèéf"}}',
             '{{"08:00", "12:00"}, {"15:00", "20:00"}}',
             '{{"08:00", "12:00"}, {}}',
         ]
@@ -49,6 +49,7 @@ class TestTablePostgres(DBTestCase):
                 simplejson.dumps(
                     val,
                     default=encode_complex,
+                    ensure_ascii=False,
                     ignore_nan=True,
                 )
             )

mage_ai/tests/orchestration/notification/test_sender.py CHANGED Viewed

@@ -112,9 +112,11 @@ class NotificationSenderTests(DBTestCase):
             f'`{pipeline_run.pipeline_schedule.name}` '
             f'at execution time `{pipeline_run.execution_date}`. Error: None'
         )
+        title = 'Failed to run Mage pipeline test_pipeline'
         mock_send_teams_message.assert_called_once_with(
             notification_config.teams_config,
             message,
+            title
         )
     @patch('mage_ai.orchestration.notification.sender.send_teams_message')
@@ -131,9 +133,11 @@ class NotificationSenderTests(DBTestCase):
             f'`{pipeline_run.pipeline_schedule.name}` '
             f'at execution time `{pipeline_run.execution_date}`.'
         )
+        title = 'Successfully ran Pipeline test_pipeline'
         mock_send_teams_message.assert_called_once_with(
             notification_config.teams_config,
             message,
+            title
         )
     @patch('mage_ai.orchestration.notification.sender.send_teams_message')
@@ -162,4 +166,4 @@ class NotificationSenderTests(DBTestCase):
             notification_config.opsgenie_config,
             message=ANY,
             description=ANY,
-        )
+        )

mage_ai/tests/streaming/sources/test_kafka.py CHANGED Viewed

@@ -33,7 +33,7 @@ class KafkaTests(TestCase):
             api_version='0.10.2',
             auto_offset_reset='latest',
             max_partition_fetch_bytes=1048576,
-            enable_auto_commit=True,
+            enable_auto_commit=False,
         )
     def test_init_client_with_topics(self):
@@ -56,7 +56,7 @@ class KafkaTests(TestCase):
             api_version='0.10.2',
             auto_offset_reset='latest',
             max_partition_fetch_bytes=1048576,
-            enable_auto_commit=True,
+            enable_auto_commit=False,
         )
     def test_init_client_with_missing_topic_or_topics(self):

mage_ai/usage_statistics/logger.py CHANGED Viewed

@@ -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.75__py3-none-any.whl → 0.9.77__py3-none-any.whl

Potentially problematic release.

mage-ai 0.9.75py3-none-any.whl → 0.9.77py3-none-any.whl