mage-ai 0.9.13__py3-none-any.whl → 0.9.15__py3-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 (187) hide show
  1. mage_ai/ai/generator.py +1 -1
  2. mage_ai/ai/generator_cmds.py +1 -1
  3. mage_ai/ai/llm_pipeline_wizard.py +31 -17
  4. mage_ai/api/policies/PipelineRunPolicy.py +1 -0
  5. mage_ai/api/presenters/BlockRunPresenter.py +4 -1
  6. mage_ai/api/presenters/PipelineRunPresenter.py +2 -0
  7. mage_ai/api/resources/BlockRunResource.py +18 -12
  8. mage_ai/api/resources/GitBranchResource.py +8 -1
  9. mage_ai/api/resources/PipelineResource.py +30 -14
  10. mage_ai/api/resources/PipelineRunResource.py +8 -5
  11. mage_ai/data_integrations/sources/constants.py +1 -0
  12. mage_ai/data_preparation/executors/block_executor.py +5 -7
  13. mage_ai/data_preparation/git/__init__.py +18 -0
  14. mage_ai/data_preparation/models/block/__init__.py +13 -1
  15. mage_ai/data_preparation/models/block/dbt/utils/__init__.py +5 -2
  16. mage_ai/data_preparation/models/block/integration/__init__.py +1 -1
  17. mage_ai/data_preparation/models/custom_templates/utils.py +14 -4
  18. mage_ai/data_preparation/models/variable.py +5 -2
  19. mage_ai/data_preparation/templates/constants.py +16 -0
  20. mage_ai/data_preparation/templates/data_exporters/mssql.py +32 -0
  21. mage_ai/data_preparation/templates/data_loaders/mssql.py +27 -0
  22. mage_ai/io/base.py +5 -0
  23. mage_ai/io/mssql.py +1 -0
  24. mage_ai/orchestration/db/models/base.py +15 -15
  25. mage_ai/orchestration/db/models/oauth.py +35 -4
  26. mage_ai/orchestration/db/models/schedules.py +4 -0
  27. mage_ai/orchestration/pipeline_scheduler.py +12 -6
  28. mage_ai/server/constants.py +1 -1
  29. mage_ai/server/frontend_dist/404.html +2 -2
  30. mage_ai/server/frontend_dist/404.html.html +2 -2
  31. mage_ai/server/frontend_dist/_next/static/chunks/{1005-ee665ba499795660.js → 1005-a2f0e3ee378ef02b.js} +1 -1
  32. mage_ai/server/frontend_dist/_next/static/chunks/1424-fca78f21a81a7183.js +1 -0
  33. mage_ai/server/frontend_dist/_next/static/chunks/2786-756b3834d10b8711.js +1 -0
  34. mage_ai/server/frontend_dist/_next/static/chunks/3391-e1578fe1bdb5a557.js +1 -0
  35. mage_ai/server/frontend_dist/_next/static/chunks/6299-9b785c07dacf22b3.js +1 -0
  36. mage_ai/server/frontend_dist/_next/static/chunks/7496-7e4dd11e3f3b8e79.js +1 -0
  37. mage_ai/server/frontend_dist/_next/static/chunks/8952-050e60a8b1eaa32a.js +1 -0
  38. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/9605-e40d076d9fc36d83.js → frontend_dist/_next/static/chunks/9605-9332e1686c46da7d.js} +1 -1
  39. mage_ai/server/frontend_dist/_next/static/chunks/pages/{_app-ccc78f4c29a2a431.js → _app-9dae6fa5126cf001.js} +1 -1
  40. mage_ai/server/frontend_dist/_next/static/chunks/pages/files-03841b6dd7d240f6.js +1 -0
  41. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/settings-6e2c0e3f818fd4de.js +1 -0
  42. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-6e59dc4e57dd2680.js +1 -0
  43. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/logs-02295848c811e26a.js +1 -0
  44. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-4a2671811a153411.js +1 -0
  45. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/{syncs-ddaca2c267def72a.js → syncs-484581ae34a1c596.js} +1 -1
  46. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-ce11db27e4af7869.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-91889109af36a793.js} +1 -1
  47. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-ff1faac7a72b1af1.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-624a2d7cbe6303b4.js} +1 -1
  48. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/triggers-eee16a91f281f92d.js → frontend_dist/_next/static/chunks/pages/triggers-71bbb30dd9b57f80.js} +1 -1
  49. mage_ai/server/{frontend_dist_base_path_template/_next/static/gJQ7p6K0VGDZzX1HXtRED → frontend_dist/_next/static/h5mR3pjfn_EQy348CZ_ok}/_buildManifest.js +1 -1
  50. mage_ai/server/frontend_dist/files.html +2 -2
  51. mage_ai/server/frontend_dist/global-data-products/[...slug].html +2 -2
  52. mage_ai/server/frontend_dist/global-data-products.html +2 -2
  53. mage_ai/server/frontend_dist/index.html +2 -2
  54. mage_ai/server/frontend_dist/manage/settings.html +2 -2
  55. mage_ai/server/frontend_dist/manage/users/[user].html +2 -2
  56. mage_ai/server/frontend_dist/manage/users/new.html +2 -2
  57. mage_ai/server/frontend_dist/manage/users.html +2 -2
  58. mage_ai/server/frontend_dist/manage.html +2 -2
  59. mage_ai/server/frontend_dist/overview.html +2 -2
  60. mage_ai/server/frontend_dist/pipeline-runs.html +2 -2
  61. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills/[...slug].html +2 -2
  62. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills.html +2 -2
  63. mage_ai/server/frontend_dist/pipelines/[pipeline]/edit.html +2 -2
  64. mage_ai/server/frontend_dist/pipelines/[pipeline]/logs.html +2 -2
  65. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runs.html +2 -2
  66. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runtime.html +2 -2
  67. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors.html +2 -2
  68. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs/[run].html +2 -2
  69. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs.html +2 -2
  70. mage_ai/server/frontend_dist/pipelines/[pipeline]/settings.html +2 -2
  71. mage_ai/server/frontend_dist/pipelines/[pipeline]/syncs.html +2 -2
  72. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers/[...slug].html +2 -2
  73. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers.html +2 -2
  74. mage_ai/server/frontend_dist/pipelines/[pipeline].html +2 -2
  75. mage_ai/server/frontend_dist/pipelines.html +2 -2
  76. mage_ai/server/frontend_dist/settings/account/profile.html +2 -2
  77. mage_ai/server/frontend_dist/settings/workspace/preferences.html +2 -2
  78. mage_ai/server/frontend_dist/settings/workspace/sync-data.html +2 -2
  79. mage_ai/server/frontend_dist/settings/workspace/users.html +2 -2
  80. mage_ai/server/frontend_dist/settings.html +2 -2
  81. mage_ai/server/frontend_dist/sign-in.html +11 -11
  82. mage_ai/server/frontend_dist/templates/[...slug].html +2 -2
  83. mage_ai/server/frontend_dist/templates.html +2 -2
  84. mage_ai/server/frontend_dist/terminal.html +2 -2
  85. mage_ai/server/frontend_dist/test.html +3 -3
  86. mage_ai/server/frontend_dist/triggers.html +2 -2
  87. mage_ai/server/frontend_dist/version-control.html +2 -2
  88. mage_ai/server/frontend_dist_base_path_template/404.html +2 -2
  89. mage_ai/server/frontend_dist_base_path_template/404.html.html +2 -2
  90. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{1005-ee665ba499795660.js → 1005-a2f0e3ee378ef02b.js} +1 -1
  91. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1424-fca78f21a81a7183.js +1 -0
  92. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/2786-756b3834d10b8711.js +1 -0
  93. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3391-e1578fe1bdb5a557.js +1 -0
  94. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/6299-9b785c07dacf22b3.js +1 -0
  95. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/7496-7e4dd11e3f3b8e79.js +1 -0
  96. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8952-050e60a8b1eaa32a.js +1 -0
  97. mage_ai/server/{frontend_dist/_next/static/chunks/9605-e40d076d9fc36d83.js → frontend_dist_base_path_template/_next/static/chunks/9605-9332e1686c46da7d.js} +1 -1
  98. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{_app-ccc78f4c29a2a431.js → _app-9dae6fa5126cf001.js} +1 -1
  99. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/files-03841b6dd7d240f6.js +1 -0
  100. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/settings-6e2c0e3f818fd4de.js +1 -0
  101. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-6e59dc4e57dd2680.js +1 -0
  102. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/logs-02295848c811e26a.js +1 -0
  103. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-4a2671811a153411.js +1 -0
  104. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/{syncs-ddaca2c267def72a.js → syncs-484581ae34a1c596.js} +1 -1
  105. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-ce11db27e4af7869.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-91889109af36a793.js} +1 -1
  106. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-ff1faac7a72b1af1.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-624a2d7cbe6303b4.js} +1 -1
  107. mage_ai/server/{frontend_dist/_next/static/chunks/pages/triggers-eee16a91f281f92d.js → frontend_dist_base_path_template/_next/static/chunks/pages/triggers-71bbb30dd9b57f80.js} +1 -1
  108. mage_ai/server/{frontend_dist/_next/static/O4HzlRY2U3Q47jF8s3D-E → frontend_dist_base_path_template/_next/static/xykYLW_ARw5SevdiponLI}/_buildManifest.js +1 -1
  109. mage_ai/server/frontend_dist_base_path_template/files.html +2 -2
  110. mage_ai/server/frontend_dist_base_path_template/global-data-products/[...slug].html +2 -2
  111. mage_ai/server/frontend_dist_base_path_template/global-data-products.html +2 -2
  112. mage_ai/server/frontend_dist_base_path_template/index.html +2 -2
  113. mage_ai/server/frontend_dist_base_path_template/manage/settings.html +2 -2
  114. mage_ai/server/frontend_dist_base_path_template/manage/users/[user].html +2 -2
  115. mage_ai/server/frontend_dist_base_path_template/manage/users/new.html +2 -2
  116. mage_ai/server/frontend_dist_base_path_template/manage/users.html +2 -2
  117. mage_ai/server/frontend_dist_base_path_template/manage.html +2 -2
  118. mage_ai/server/frontend_dist_base_path_template/overview.html +2 -2
  119. mage_ai/server/frontend_dist_base_path_template/pipeline-runs.html +2 -2
  120. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills/[...slug].html +2 -2
  121. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills.html +2 -2
  122. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/edit.html +2 -2
  123. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/logs.html +2 -2
  124. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runs.html +2 -2
  125. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runtime.html +2 -2
  126. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors.html +2 -2
  127. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs/[run].html +2 -2
  128. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs.html +2 -2
  129. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/settings.html +2 -2
  130. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/syncs.html +2 -2
  131. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers/[...slug].html +2 -2
  132. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers.html +2 -2
  133. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline].html +2 -2
  134. mage_ai/server/frontend_dist_base_path_template/pipelines.html +2 -2
  135. mage_ai/server/frontend_dist_base_path_template/settings/account/profile.html +2 -2
  136. mage_ai/server/frontend_dist_base_path_template/settings/workspace/preferences.html +2 -2
  137. mage_ai/server/frontend_dist_base_path_template/settings/workspace/sync-data.html +2 -2
  138. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users.html +2 -2
  139. mage_ai/server/frontend_dist_base_path_template/settings.html +2 -2
  140. mage_ai/server/frontend_dist_base_path_template/sign-in.html +11 -11
  141. mage_ai/server/frontend_dist_base_path_template/templates/[...slug].html +2 -2
  142. mage_ai/server/frontend_dist_base_path_template/templates.html +2 -2
  143. mage_ai/server/frontend_dist_base_path_template/terminal.html +2 -2
  144. mage_ai/server/frontend_dist_base_path_template/test.html +3 -3
  145. mage_ai/server/frontend_dist_base_path_template/triggers.html +2 -2
  146. mage_ai/server/frontend_dist_base_path_template/version-control.html +2 -2
  147. mage_ai/server/server.py +12 -1
  148. mage_ai/services/spark/spark.py +75 -2
  149. mage_ai/shared/retry.py +1 -1
  150. mage_ai/tests/data_preparation/models/custom_templates/__init__.py +0 -0
  151. mage_ai/tests/data_preparation/models/custom_templates/test_utils.py +36 -0
  152. mage_ai/tests/data_preparation/models/test_block.py +48 -0
  153. mage_ai/tests/services/spark/__init__.py +0 -0
  154. mage_ai/tests/services/spark/test_spark.py +52 -0
  155. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/METADATA +1 -1
  156. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/RECORD +164 -159
  157. mage_ai/server/frontend_dist/_next/static/chunks/1424-846b754e84e6eac1.js +0 -1
  158. mage_ai/server/frontend_dist/_next/static/chunks/2786-ba6f486fcaf52ba0.js +0 -1
  159. mage_ai/server/frontend_dist/_next/static/chunks/3391-6f0a0a5c254cd7f2.js +0 -1
  160. mage_ai/server/frontend_dist/_next/static/chunks/6299-fcb702d0f3d3291b.js +0 -1
  161. mage_ai/server/frontend_dist/_next/static/chunks/7496-7d0f9adf0b333801.js +0 -1
  162. mage_ai/server/frontend_dist/_next/static/chunks/8952-9d6fa18fa9378989.js +0 -1
  163. mage_ai/server/frontend_dist/_next/static/chunks/pages/files-2dc2a0dfc0ff620d.js +0 -1
  164. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/settings-6577159a0af52a78.js +0 -1
  165. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-615ab9e61f0f39ed.js +0 -1
  166. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/logs-d331e98ad103b13e.js +0 -1
  167. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-df014ddb14ebcef4.js +0 -1
  168. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1424-846b754e84e6eac1.js +0 -1
  169. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/2786-ba6f486fcaf52ba0.js +0 -1
  170. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3391-6f0a0a5c254cd7f2.js +0 -1
  171. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/6299-fcb702d0f3d3291b.js +0 -1
  172. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/7496-7d0f9adf0b333801.js +0 -1
  173. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8952-9d6fa18fa9378989.js +0 -1
  174. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/files-2dc2a0dfc0ff620d.js +0 -1
  175. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/settings-6577159a0af52a78.js +0 -1
  176. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-615ab9e61f0f39ed.js +0 -1
  177. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/logs-d331e98ad103b13e.js +0 -1
  178. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-df014ddb14ebcef4.js +0 -1
  179. mage_ai/tests/orchestration/db/test_models.py +0 -710
  180. /mage_ai/server/frontend_dist/_next/static/{O4HzlRY2U3Q47jF8s3D-E → h5mR3pjfn_EQy348CZ_ok}/_middlewareManifest.js +0 -0
  181. /mage_ai/server/frontend_dist/_next/static/{O4HzlRY2U3Q47jF8s3D-E → h5mR3pjfn_EQy348CZ_ok}/_ssgManifest.js +0 -0
  182. /mage_ai/server/frontend_dist_base_path_template/_next/static/{gJQ7p6K0VGDZzX1HXtRED → xykYLW_ARw5SevdiponLI}/_middlewareManifest.js +0 -0
  183. /mage_ai/server/frontend_dist_base_path_template/_next/static/{gJQ7p6K0VGDZzX1HXtRED → xykYLW_ARw5SevdiponLI}/_ssgManifest.js +0 -0
  184. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/LICENSE +0 -0
  185. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/WHEEL +0 -0
  186. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/entry_points.txt +0 -0
  187. {mage_ai-0.9.13.dist-info → mage_ai-0.9.15.dist-info}/top_level.txt +0 -0