mage-ai 0.9.73__py3-none-any.whl → 0.9.74__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.
Files changed (247) hide show
  1. mage_ai/ai/constants.py +2 -2
  2. mage_ai/ai/llm_pipeline_wizard.py +5 -3
  3. mage_ai/ai/openai_client.py +11 -0
  4. mage_ai/api/constants.py +4 -5
  5. mage_ai/api/oauth_scope.py +2 -2
  6. mage_ai/api/operations/constants.py +2 -2
  7. mage_ai/api/presenters/SyncPresenter.py +4 -1
  8. mage_ai/authentication/oauth/constants.py +2 -2
  9. mage_ai/authentication/operation_history/constants.py +2 -2
  10. mage_ai/authentication/permissions/constants.py +5 -6
  11. mage_ai/cache/constants.py +2 -2
  12. mage_ai/cache/dbt/constants.py +2 -3
  13. mage_ai/cluster_manager/constants.py +2 -2
  14. mage_ai/cluster_manager/kubernetes/workload_manager.py +3 -2
  15. mage_ai/command_center/constants.py +13 -13
  16. mage_ai/data_cleaner/column_types/constants.py +2 -2
  17. mage_ai/data_cleaner/transformer_actions/column.py +19 -7
  18. mage_ai/data_cleaner/transformer_actions/constants.py +10 -9
  19. mage_ai/data_integrations/sources/constants.py +2 -0
  20. mage_ai/data_preparation/git/utils.py +4 -1
  21. mage_ai/data_preparation/logging/__init__.py +2 -2
  22. mage_ai/data_preparation/logging/logger.py +3 -3
  23. mage_ai/data_preparation/models/block/__init__.py +9 -3
  24. mage_ai/data_preparation/models/block/data_integration/constants.py +2 -2
  25. mage_ai/data_preparation/models/block/dbt/constants.py +3 -3
  26. mage_ai/data_preparation/models/block/dynamic/factory.py +40 -3
  27. mage_ai/data_preparation/models/block/dynamic/utils.py +31 -33
  28. mage_ai/data_preparation/models/block/dynamic/variables.py +1 -1
  29. mage_ai/data_preparation/models/block/settings/dynamic/constants.py +3 -3
  30. mage_ai/data_preparation/models/block/settings/dynamic/mixins.py +63 -4
  31. mage_ai/data_preparation/models/block/settings/global_data_products/models.py +2 -2
  32. mage_ai/data_preparation/models/constants.py +10 -10
  33. mage_ai/data_preparation/models/global_hooks/constants.py +7 -8
  34. mage_ai/data_preparation/models/global_hooks/models.py +5 -5
  35. mage_ai/data_preparation/models/project/constants.py +2 -2
  36. mage_ai/data_preparation/models/triggers/__init__.py +6 -4
  37. mage_ai/data_preparation/models/variables/constants.py +5 -5
  38. mage_ai/data_preparation/models/widget/constants.py +5 -5
  39. mage_ai/data_preparation/preferences.py +9 -16
  40. mage_ai/data_preparation/repo_manager.py +2 -2
  41. mage_ai/data_preparation/sync/__init__.py +2 -2
  42. mage_ai/data_preparation/templates/data_exporters/streaming/generic_python.py +1 -1
  43. mage_ai/data_preparation/templates/data_loaders/streaming/nats.yaml +3 -0
  44. mage_ai/errors/constants.py +2 -2
  45. mage_ai/io/base.py +28 -15
  46. mage_ai/io/config.py +3 -3
  47. mage_ai/io/export_utils.py +2 -2
  48. mage_ai/io/google_cloud_storage.py +3 -2
  49. mage_ai/io/io_config.py +3 -2
  50. mage_ai/io/postgres.py +2 -1
  51. mage_ai/kernels/magic/constants.py +4 -4
  52. mage_ai/orchestration/constants.py +2 -2
  53. mage_ai/orchestration/db/constants.py +2 -2
  54. mage_ai/orchestration/db/models/oauth.py +5 -5
  55. mage_ai/orchestration/db/models/schedules.py +28 -8
  56. mage_ai/orchestration/db/models/schedules_project_platform.py +10 -8
  57. mage_ai/orchestration/job_manager.py +2 -2
  58. mage_ai/orchestration/monitor/monitor_stats.py +2 -2
  59. mage_ai/orchestration/notification/config.py +2 -2
  60. mage_ai/orchestration/pipeline_scheduler_original.py +4 -12
  61. mage_ai/orchestration/queue/config.py +2 -2
  62. mage_ai/orchestration/queue/process_queue.py +3 -3
  63. mage_ai/presenters/charts/data_sources/constants.py +2 -2
  64. mage_ai/presenters/interactions/constants.py +4 -4
  65. mage_ai/presenters/pages/models/constants.py +4 -4
  66. mage_ai/server/constants.py +1 -1
  67. mage_ai/server/frontend_dist/404.html +2 -2
  68. mage_ai/server/frontend_dist/_next/static/chunks/pages/{_app-663c909dcda4c23a.js → _app-5bdff745074fb350.js} +2 -2
  69. mage_ai/server/frontend_dist/_next/static/chunks/{webpack-43534cc51fce8644.js → webpack-b9a067f3bd0a3a05.js} +1 -1
  70. mage_ai/server/frontend_dist/block-layout.html +2 -2
  71. mage_ai/server/frontend_dist/compute.html +2 -2
  72. mage_ai/server/frontend_dist/files.html +2 -2
  73. mage_ai/server/frontend_dist/global-data-products/[...slug].html +2 -2
  74. mage_ai/server/frontend_dist/global-data-products.html +2 -2
  75. mage_ai/server/frontend_dist/global-hooks/[...slug].html +2 -2
  76. mage_ai/server/frontend_dist/global-hooks.html +2 -2
  77. mage_ai/server/frontend_dist/index.html +2 -2
  78. mage_ai/server/frontend_dist/manage/files.html +2 -2
  79. mage_ai/server/frontend_dist/manage/overview.html +2 -2
  80. mage_ai/server/frontend_dist/manage/pipeline-runs.html +2 -2
  81. mage_ai/server/frontend_dist/manage/settings.html +2 -2
  82. mage_ai/server/frontend_dist/manage/users/[user].html +2 -2
  83. mage_ai/server/frontend_dist/manage/users/new.html +2 -2
  84. mage_ai/server/frontend_dist/manage/users.html +2 -2
  85. mage_ai/server/frontend_dist/manage.html +2 -2
  86. mage_ai/server/frontend_dist/oauth.html +3 -3
  87. mage_ai/server/frontend_dist/overview.html +2 -2
  88. mage_ai/server/frontend_dist/pipeline-runs.html +2 -2
  89. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills/[...slug].html +2 -2
  90. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills.html +2 -2
  91. mage_ai/server/frontend_dist/pipelines/[pipeline]/dashboard.html +2 -2
  92. mage_ai/server/frontend_dist/pipelines/[pipeline]/edit.html +2 -2
  93. mage_ai/server/frontend_dist/pipelines/[pipeline]/logs.html +2 -2
  94. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runs.html +2 -2
  95. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runtime.html +2 -2
  96. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors.html +2 -2
  97. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs/[run].html +2 -2
  98. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs.html +2 -2
  99. mage_ai/server/frontend_dist/pipelines/[pipeline]/settings.html +2 -2
  100. mage_ai/server/frontend_dist/pipelines/[pipeline]/syncs.html +2 -2
  101. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers/[...slug].html +2 -2
  102. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers.html +2 -2
  103. mage_ai/server/frontend_dist/pipelines/[pipeline].html +2 -2
  104. mage_ai/server/frontend_dist/pipelines.html +2 -2
  105. mage_ai/server/frontend_dist/platform/global-hooks/[...slug].html +2 -2
  106. mage_ai/server/frontend_dist/platform/global-hooks.html +2 -2
  107. mage_ai/server/frontend_dist/settings/account/profile.html +2 -2
  108. mage_ai/server/frontend_dist/settings/platform/preferences.html +2 -2
  109. mage_ai/server/frontend_dist/settings/platform/settings.html +2 -2
  110. mage_ai/server/frontend_dist/settings/workspace/permissions/[...slug].html +2 -2
  111. mage_ai/server/frontend_dist/settings/workspace/permissions.html +2 -2
  112. mage_ai/server/frontend_dist/settings/workspace/preferences.html +2 -2
  113. mage_ai/server/frontend_dist/settings/workspace/roles/[...slug].html +2 -2
  114. mage_ai/server/frontend_dist/settings/workspace/roles.html +2 -2
  115. mage_ai/server/frontend_dist/settings/workspace/sync-data.html +2 -2
  116. mage_ai/server/frontend_dist/settings/workspace/users/[...slug].html +2 -2
  117. mage_ai/server/frontend_dist/settings/workspace/users.html +2 -2
  118. mage_ai/server/frontend_dist/settings.html +2 -2
  119. mage_ai/server/frontend_dist/sign-in.html +5 -5
  120. mage_ai/server/frontend_dist/templates/[...slug].html +2 -2
  121. mage_ai/server/frontend_dist/templates.html +2 -2
  122. mage_ai/server/frontend_dist/terminal.html +2 -2
  123. mage_ai/server/frontend_dist/test.html +2 -2
  124. mage_ai/server/frontend_dist/triggers.html +2 -2
  125. mage_ai/server/frontend_dist/v2/canvas.html +2 -2
  126. mage_ai/server/frontend_dist/v2.html +2 -2
  127. mage_ai/server/frontend_dist/version-control.html +2 -2
  128. mage_ai/server/frontend_dist_base_path_template/404.html +2 -2
  129. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{_app-bb4a0e0d783622a8.js → _app-90de19bc03f1484b.js} +2 -2
  130. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{webpack-9eb1dd2ee735aaac.js → webpack-12ad70eb5c31aa92.js} +1 -1
  131. mage_ai/server/frontend_dist_base_path_template/block-layout.html +2 -2
  132. mage_ai/server/frontend_dist_base_path_template/compute.html +2 -2
  133. mage_ai/server/frontend_dist_base_path_template/files.html +2 -2
  134. mage_ai/server/frontend_dist_base_path_template/global-data-products/[...slug].html +2 -2
  135. mage_ai/server/frontend_dist_base_path_template/global-data-products.html +2 -2
  136. mage_ai/server/frontend_dist_base_path_template/global-hooks/[...slug].html +2 -2
  137. mage_ai/server/frontend_dist_base_path_template/global-hooks.html +2 -2
  138. mage_ai/server/frontend_dist_base_path_template/index.html +2 -2
  139. mage_ai/server/frontend_dist_base_path_template/manage/files.html +2 -2
  140. mage_ai/server/frontend_dist_base_path_template/manage/overview.html +2 -2
  141. mage_ai/server/frontend_dist_base_path_template/manage/pipeline-runs.html +2 -2
  142. mage_ai/server/frontend_dist_base_path_template/manage/settings.html +2 -2
  143. mage_ai/server/frontend_dist_base_path_template/manage/users/[user].html +2 -2
  144. mage_ai/server/frontend_dist_base_path_template/manage/users/new.html +2 -2
  145. mage_ai/server/frontend_dist_base_path_template/manage/users.html +2 -2
  146. mage_ai/server/frontend_dist_base_path_template/manage.html +2 -2
  147. mage_ai/server/frontend_dist_base_path_template/oauth.html +3 -3
  148. mage_ai/server/frontend_dist_base_path_template/overview.html +2 -2
  149. mage_ai/server/frontend_dist_base_path_template/pipeline-runs.html +2 -2
  150. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills/[...slug].html +2 -2
  151. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills.html +2 -2
  152. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/dashboard.html +2 -2
  153. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/edit.html +2 -2
  154. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/logs.html +2 -2
  155. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runs.html +2 -2
  156. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runtime.html +2 -2
  157. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors.html +2 -2
  158. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs/[run].html +2 -2
  159. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs.html +2 -2
  160. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/settings.html +2 -2
  161. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/syncs.html +2 -2
  162. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers/[...slug].html +2 -2
  163. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers.html +2 -2
  164. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline].html +2 -2
  165. mage_ai/server/frontend_dist_base_path_template/pipelines.html +2 -2
  166. mage_ai/server/frontend_dist_base_path_template/platform/global-hooks/[...slug].html +2 -2
  167. mage_ai/server/frontend_dist_base_path_template/platform/global-hooks.html +2 -2
  168. mage_ai/server/frontend_dist_base_path_template/settings/account/profile.html +2 -2
  169. mage_ai/server/frontend_dist_base_path_template/settings/platform/preferences.html +2 -2
  170. mage_ai/server/frontend_dist_base_path_template/settings/platform/settings.html +2 -2
  171. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions/[...slug].html +2 -2
  172. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions.html +2 -2
  173. mage_ai/server/frontend_dist_base_path_template/settings/workspace/preferences.html +2 -2
  174. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles/[...slug].html +2 -2
  175. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles.html +2 -2
  176. mage_ai/server/frontend_dist_base_path_template/settings/workspace/sync-data.html +2 -2
  177. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users/[...slug].html +2 -2
  178. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users.html +2 -2
  179. mage_ai/server/frontend_dist_base_path_template/settings.html +2 -2
  180. mage_ai/server/frontend_dist_base_path_template/sign-in.html +5 -5
  181. mage_ai/server/frontend_dist_base_path_template/templates/[...slug].html +2 -2
  182. mage_ai/server/frontend_dist_base_path_template/templates.html +2 -2
  183. mage_ai/server/frontend_dist_base_path_template/terminal.html +2 -2
  184. mage_ai/server/frontend_dist_base_path_template/test.html +2 -2
  185. mage_ai/server/frontend_dist_base_path_template/triggers.html +2 -2
  186. mage_ai/server/frontend_dist_base_path_template/v2/canvas.html +2 -2
  187. mage_ai/server/frontend_dist_base_path_template/v2.html +2 -2
  188. mage_ai/server/frontend_dist_base_path_template/version-control.html +2 -2
  189. mage_ai/server/kernel_output_parser.py +2 -3
  190. mage_ai/server/kernels.py +2 -3
  191. mage_ai/server/scheduler_manager.py +2 -2
  192. mage_ai/server/server.py +12 -0
  193. mage_ai/server/websockets/constants.py +4 -4
  194. mage_ai/services/compute/aws/constants.py +2 -2
  195. mage_ai/services/compute/aws/steps.py +16 -10
  196. mage_ai/services/compute/constants.py +4 -4
  197. mage_ai/services/compute/models.py +4 -4
  198. mage_ai/services/k8s/config.py +4 -0
  199. mage_ai/services/k8s/job_manager.py +2 -0
  200. mage_ai/services/spark/constants.py +3 -3
  201. mage_ai/services/spark/models/jobs.py +2 -2
  202. mage_ai/services/spark/models/sqls.py +2 -2
  203. mage_ai/services/spark/models/stages.py +4 -4
  204. mage_ai/services/spark/models/threads.py +2 -2
  205. mage_ai/settings/backends.py +3 -2
  206. mage_ai/settings/models/configuration_option.py +3 -3
  207. mage_ai/settings/server.py +5 -0
  208. mage_ai/shared/constants.py +3 -3
  209. mage_ai/shared/custom_logger.py +13 -13
  210. mage_ai/shared/enum.py +11 -0
  211. mage_ai/shared/environments.py +1 -1
  212. mage_ai/shared/logger.py +2 -2
  213. mage_ai/shared/models.py +2 -1
  214. mage_ai/streaming/constants.py +3 -3
  215. mage_ai/streaming/sinks/kafka.py +2 -2
  216. mage_ai/streaming/sinks/postgres.py +6 -0
  217. mage_ai/streaming/sources/amazon_sqs.py +2 -2
  218. mage_ai/streaming/sources/base.py +2 -2
  219. mage_ai/streaming/sources/kafka.py +20 -4
  220. mage_ai/streaming/sources/nats_js.py +10 -3
  221. mage_ai/streaming/sources/shared.py +3 -2
  222. mage_ai/system/constants.py +2 -2
  223. mage_ai/tests/api/policies/permissions/test_base_policy_with_permissions.py +2 -2
  224. mage_ai/tests/data_cleaner/transformer_actions/test_trim_transformer.py +161 -0
  225. mage_ai/tests/data_preparation/models/block/dbt/test_profiles.py +1 -1
  226. mage_ai/tests/data_preparation/models/block/dynamic/test_combos.py +1 -1
  227. mage_ai/tests/data_preparation/models/block/hook/test_hook_block.py +3 -2
  228. mage_ai/tests/data_preparation/models/test_blocks_helper.py +1 -1
  229. mage_ai/tests/data_preparation/models/variables/test_summarizer.py +1 -1
  230. mage_ai/tests/data_preparation/sync/test_git_sync.py +6 -6
  231. mage_ai/tests/orchestration/queue/test_process_queue.py +3 -2
  232. mage_ai/tests/orchestration/test_pipeline_scheduler.py +1 -0
  233. mage_ai/tests/settings/test_platform.py +2 -2
  234. mage_ai/tests/streaming/sinks/test_generic_io.py +25 -21
  235. mage_ai/usage_statistics/constants.py +4 -4
  236. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/METADATA +172 -171
  237. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/RECORD +247 -245
  238. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/WHEEL +1 -1
  239. /mage_ai/server/frontend_dist/_next/static/chunks/pages/{_app-663c909dcda4c23a.js.LICENSE.txt → _app-5bdff745074fb350.js.LICENSE.txt} +0 -0
  240. /mage_ai/server/frontend_dist/_next/static/{kxGpiudO3f9aX6FAiqydf → pLWT6Sqd09xYpufCVIqnz}/_buildManifest.js +0 -0
  241. /mage_ai/server/frontend_dist/_next/static/{kxGpiudO3f9aX6FAiqydf → pLWT6Sqd09xYpufCVIqnz}/_ssgManifest.js +0 -0
  242. /mage_ai/server/frontend_dist_base_path_template/_next/static/{H5pcGxWf1BkhXDRs2BqiI → JQewSAObpbhO0wrdAM6Ng}/_buildManifest.js +0 -0
  243. /mage_ai/server/frontend_dist_base_path_template/_next/static/{H5pcGxWf1BkhXDRs2BqiI → JQewSAObpbhO0wrdAM6Ng}/_ssgManifest.js +0 -0
  244. /mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{_app-bb4a0e0d783622a8.js.LICENSE.txt → _app-90de19bc03f1484b.js.LICENSE.txt} +0 -0
  245. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/LICENSE +0 -0
  246. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/entry_points.txt +0 -0
  247. {mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/top_level.txt +0 -0
{mage_ai-0.9.73.dist-info → mage_ai-0.9.74.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (72.1.0)
2
+ Generator: setuptools (75.1.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5