mage-ai 0.9.74__py3-none-any.whl → 0.9.75__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 (329) hide show
  1. mage_ai/ai/llm_pipeline_wizard.py +6 -4
  2. mage_ai/ai/openai_client.py +7 -5
  3. mage_ai/data_integrations/destinations/constants.py +1 -0
  4. mage_ai/data_preparation/models/block/__init__.py +4 -4
  5. mage_ai/data_preparation/models/constants.py +1 -1
  6. mage_ai/data_preparation/models/pipeline.py +5 -0
  7. mage_ai/data_preparation/templates/data_exporters/streaming/elasticsearch.yaml +3 -0
  8. mage_ai/data_preparation/templates/data_loaders/streaming/nats.yaml +6 -3
  9. mage_ai/io/base.py +29 -1
  10. mage_ai/io/oracledb.py +2 -4
  11. mage_ai/io/postgres.py +38 -19
  12. mage_ai/io/utils.py +18 -0
  13. mage_ai/orchestration/db/__init__.py +3 -3
  14. mage_ai/server/constants.py +1 -1
  15. mage_ai/server/frontend_dist/404.html +3 -3
  16. mage_ai/server/frontend_dist/_next/static/38-PtskJFUTYUpRhT1qF_/_buildManifest.js +1 -0
  17. mage_ai/server/frontend_dist/_next/static/chunks/{1187-839336d276186105.js → 1187-4560c3895e1d7099.js} +1 -1
  18. mage_ai/server/frontend_dist/_next/static/chunks/{1598-0adca9dce3ba4c60.js → 1598-cbf3f5a6078fc3f5.js} +1 -1
  19. mage_ai/server/frontend_dist/_next/static/chunks/2717-638a944d24d5abde.js +1 -0
  20. mage_ai/server/frontend_dist/_next/static/chunks/3548-36f746b1824004f2.js +1 -0
  21. mage_ai/server/frontend_dist/_next/static/chunks/{3763-39a5174f6a3924db.js → 3763-aabe2703076636b0.js} +1 -1
  22. mage_ai/server/frontend_dist/_next/static/chunks/3782-3e2acb5ed45b582b.js +1 -0
  23. mage_ai/server/frontend_dist/_next/static/chunks/449-f689774546860ca4.js +1 -0
  24. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/5627-d5e559859dd0e1e0.js → frontend_dist/_next/static/chunks/5627-10e76bafa5a26f5f.js} +1 -1
  25. mage_ai/server/frontend_dist/_next/static/chunks/{5699-e49718dfc9eb2854.js → 5699-e99379e332bd0b41.js} +1 -1
  26. mage_ai/server/frontend_dist/_next/static/chunks/{7966-163da2621b8c987c.js → 7966-a5a7db345ce81263.js} +1 -1
  27. mage_ai/server/frontend_dist/_next/static/chunks/pages/{_app-5bdff745074fb350.js → _app-13bf3b7dcef50c29.js} +2 -2
  28. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/compute-ed67fa8e81662e8b.js → frontend_dist/_next/static/chunks/pages/compute-9e2dea78024e3bb4.js} +1 -1
  29. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/files-e0ecd7ced09a63b2.js → frontend_dist/_next/static/chunks/pages/files-e08c7fe76f968f9c.js} +1 -1
  30. mage_ai/server/frontend_dist/_next/static/chunks/pages/global-data-products/{[...slug]-c7a729477ecda50e.js → [...slug]-30c3807057a4e65b.js} +1 -1
  31. mage_ai/server/frontend_dist/_next/static/chunks/pages/{global-data-products-fd6ae6a358a60a0c.js → global-data-products-8dcb3b31af9e0e39.js} +1 -1
  32. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/global-hooks/[...slug]-8e50243797a7fe59.js → frontend_dist/_next/static/chunks/pages/global-hooks/[...slug]-85a64b64d27214b6.js} +1 -1
  33. mage_ai/server/frontend_dist/_next/static/chunks/pages/{global-hooks-d0c003446332dc0d.js → global-hooks-4ff959d51b8a9502.js} +1 -1
  34. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{files-a69ed8e9f814490c.js → files-d08a460641d0efaa.js} +1 -1
  35. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{overview-1aad7093c6d39257.js → overview-aae747f487e08d51.js} +1 -1
  36. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{pipeline-runs-528d30e0d13b0cc7.js → pipeline-runs-09a842d64a6ada62.js} +1 -1
  37. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{settings-fb9201d9cf63031d.js → settings-2e98e57d9376a458.js} +1 -1
  38. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/users/{[user]-000f5a980a07da39.js → [user]-7be6e41ad66089bb.js} +1 -1
  39. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/new-e4e613f6e817a733.js → frontend_dist/_next/static/chunks/pages/manage/users/new-4c088833063bfa07.js} +1 -1
  40. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{users-a5e9d77ed5b50205.js → users-b99379d0aa6a8c25.js} +1 -1
  41. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/manage-34d718b8a4066c23.js → frontend_dist/_next/static/chunks/pages/manage-868fcd8cbeb265f0.js} +1 -1
  42. mage_ai/server/frontend_dist/_next/static/chunks/pages/{oauth-3bfd1b8d7f036726.js → oauth-6ceceb62191dfe8a.js} +1 -1
  43. mage_ai/server/frontend_dist/_next/static/chunks/pages/overview-e51cd04bd4d1fffe.js +1 -0
  44. mage_ai/server/frontend_dist/_next/static/chunks/pages/{pipeline-runs-5f8c100e648efa8a.js → pipeline-runs-2d0136b51b57de93.js} +1 -1
  45. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills/{[...slug]-688c652f3296bb9c.js → [...slug]-1ad5238742e25b4c.js} +1 -1
  46. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills-bd11e87d026bfbf9.js +1 -0
  47. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/{dashboard-1236e36d39b1637d.js → dashboard-0f4f47f721b0723f.js} +1 -1
  48. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-432da20df91511fb.js +1 -0
  49. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/logs-fe91dfb0091f6bc6.js +1 -0
  50. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-0d68d4bf6290fefb.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-cf794b2d22a80f31.js} +1 -1
  51. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-9254358d58f07714.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-a964caef91bed9e1.js} +1 -1
  52. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/{monitors-821001e690caebe2.js → monitors-80bebb4401eefe25.js} +1 -1
  53. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-2eae7cb017027682.js +1 -0
  54. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-2d4b2a0800a66b33.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-e3d7fdfeb7f806f7.js} +1 -1
  55. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/{settings-cd49372ae1702963.js → settings-0abf8a1b7243f93b.js} +1 -1
  56. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/syncs-707ed8ca942ca802.js +1 -0
  57. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers/{[...slug]-259143ed3cf59e31.js → [...slug]-8429f17d4146e1ec.js} +1 -1
  58. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-193045d9836d8d80.js +1 -0
  59. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines-b578b075a8d857e3.js → frontend_dist/_next/static/chunks/pages/pipelines-38187954b6ec4b40.js} +1 -1
  60. mage_ai/server/frontend_dist/_next/static/chunks/pages/platform/global-hooks/{[...slug]-5eeec927e4202b63.js → [...slug]-6834ae87bd668cb2.js} +1 -1
  61. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/platform/global-hooks-fbe9ad995d46d837.js → frontend_dist/_next/static/chunks/pages/platform/global-hooks-b3f7309a23e592b2.js} +1 -1
  62. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/account/{profile-fc659962d4015cb3.js → profile-f8b7374385e1f1bf.js} +1 -1
  63. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/platform/preferences-8de68502a9afa299.js +1 -0
  64. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/settings/platform/settings-a4f88c334414402b.js → frontend_dist/_next/static/chunks/pages/settings/platform/settings-50fb6a34f3913f1f.js} +1 -1
  65. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/permissions/{[...slug]-4deb9579ef99a3c6.js → [...slug]-2e5c098c21ea32b7.js} +1 -1
  66. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{permissions-e0cda2f2bfce8d61.js → permissions-54e4b15b9585bfc4.js} +1 -1
  67. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/preferences-040f83d75d0f6537.js +1 -0
  68. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/roles/{[...slug]-910257d16c604ebd.js → [...slug]-95088f43034e3c95.js} +1 -1
  69. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{roles-4f7a0756806cee34.js → roles-e9149e1fcf218f42.js} +1 -1
  70. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{sync-data-208d6f955204d704.js → sync-data-75b67ae4a00818ef.js} +1 -1
  71. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/users/{[...slug]-c89dc67e5a1706a8.js → [...slug]-557dda05ca6c6124.js} +1 -1
  72. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{users-5212c01a9dc558da.js → users-3ee783f5139c76a1.js} +1 -1
  73. mage_ai/server/frontend_dist/_next/static/chunks/pages/{sign-in-054b33312d3193c3.js → sign-in-593c40985d63fcf7.js} +1 -1
  74. mage_ai/server/frontend_dist/_next/static/chunks/pages/templates/{[...slug]-b6ed6a5d818bfd20.js → [...slug]-252c4b6b818345d5.js} +1 -1
  75. mage_ai/server/frontend_dist/_next/static/chunks/pages/{templates-852357bc983af2ea.js → templates-ca528bc607753ab8.js} +1 -1
  76. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/terminal-1f9c56d671bbc67d.js → frontend_dist/_next/static/chunks/pages/terminal-287362c1defcc96b.js} +1 -1
  77. mage_ai/server/frontend_dist/_next/static/chunks/pages/test-2f83af8c9f1378fe.js +1 -0
  78. mage_ai/server/frontend_dist/_next/static/chunks/pages/triggers-d9de73fb799efed8.js +1 -0
  79. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/version-control-ae3469b992a341d6.js → frontend_dist/_next/static/chunks/pages/version-control-573f0225d7a703ed.js} +1 -1
  80. mage_ai/server/frontend_dist/_next/static/chunks/{webpack-b9a067f3bd0a3a05.js → webpack-0bc44da590c7cf85.js} +1 -1
  81. mage_ai/server/frontend_dist/block-layout.html +3 -3
  82. mage_ai/server/frontend_dist/compute.html +6 -6
  83. mage_ai/server/frontend_dist/files.html +6 -6
  84. mage_ai/server/frontend_dist/global-data-products/[...slug].html +6 -6
  85. mage_ai/server/frontend_dist/global-data-products.html +6 -6
  86. mage_ai/server/frontend_dist/global-hooks/[...slug].html +6 -6
  87. mage_ai/server/frontend_dist/global-hooks.html +6 -6
  88. mage_ai/server/frontend_dist/index.html +3 -3
  89. mage_ai/server/frontend_dist/manage/files.html +6 -6
  90. mage_ai/server/frontend_dist/manage/overview.html +6 -6
  91. mage_ai/server/frontend_dist/manage/pipeline-runs.html +6 -6
  92. mage_ai/server/frontend_dist/manage/settings.html +6 -6
  93. mage_ai/server/frontend_dist/manage/users/[user].html +6 -6
  94. mage_ai/server/frontend_dist/manage/users/new.html +6 -6
  95. mage_ai/server/frontend_dist/manage/users.html +6 -6
  96. mage_ai/server/frontend_dist/manage.html +6 -6
  97. mage_ai/server/frontend_dist/oauth.html +5 -5
  98. mage_ai/server/frontend_dist/overview.html +6 -6
  99. mage_ai/server/frontend_dist/pipeline-runs.html +6 -6
  100. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills/[...slug].html +6 -6
  101. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills.html +6 -6
  102. mage_ai/server/frontend_dist/pipelines/[pipeline]/dashboard.html +6 -6
  103. mage_ai/server/frontend_dist/pipelines/[pipeline]/edit.html +3 -3
  104. mage_ai/server/frontend_dist/pipelines/[pipeline]/logs.html +6 -6
  105. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runs.html +6 -6
  106. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runtime.html +6 -6
  107. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors.html +6 -6
  108. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs/[run].html +6 -6
  109. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs.html +6 -6
  110. mage_ai/server/frontend_dist/pipelines/[pipeline]/settings.html +6 -6
  111. mage_ai/server/frontend_dist/pipelines/[pipeline]/syncs.html +6 -6
  112. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers/[...slug].html +6 -6
  113. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers.html +6 -6
  114. mage_ai/server/frontend_dist/pipelines/[pipeline].html +3 -3
  115. mage_ai/server/frontend_dist/pipelines.html +6 -6
  116. mage_ai/server/frontend_dist/platform/global-hooks/[...slug].html +6 -6
  117. mage_ai/server/frontend_dist/platform/global-hooks.html +6 -6
  118. mage_ai/server/frontend_dist/settings/account/profile.html +6 -6
  119. mage_ai/server/frontend_dist/settings/platform/preferences.html +6 -6
  120. mage_ai/server/frontend_dist/settings/platform/settings.html +6 -6
  121. mage_ai/server/frontend_dist/settings/workspace/permissions/[...slug].html +6 -6
  122. mage_ai/server/frontend_dist/settings/workspace/permissions.html +6 -6
  123. mage_ai/server/frontend_dist/settings/workspace/preferences.html +6 -6
  124. mage_ai/server/frontend_dist/settings/workspace/roles/[...slug].html +6 -6
  125. mage_ai/server/frontend_dist/settings/workspace/roles.html +6 -6
  126. mage_ai/server/frontend_dist/settings/workspace/sync-data.html +6 -6
  127. mage_ai/server/frontend_dist/settings/workspace/users/[...slug].html +6 -6
  128. mage_ai/server/frontend_dist/settings/workspace/users.html +6 -6
  129. mage_ai/server/frontend_dist/settings.html +3 -3
  130. mage_ai/server/frontend_dist/sign-in.html +7 -7
  131. mage_ai/server/frontend_dist/templates/[...slug].html +6 -6
  132. mage_ai/server/frontend_dist/templates.html +6 -6
  133. mage_ai/server/frontend_dist/terminal.html +6 -6
  134. mage_ai/server/frontend_dist/test.html +3 -3
  135. mage_ai/server/frontend_dist/triggers.html +6 -6
  136. mage_ai/server/frontend_dist/v2/canvas.html +2 -2
  137. mage_ai/server/frontend_dist/v2.html +2 -2
  138. mage_ai/server/frontend_dist/version-control.html +6 -6
  139. mage_ai/server/frontend_dist_base_path_template/404.html +3 -3
  140. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{1187-839336d276186105.js → 1187-4560c3895e1d7099.js} +1 -1
  141. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{1598-0adca9dce3ba4c60.js → 1598-cbf3f5a6078fc3f5.js} +1 -1
  142. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/2717-638a944d24d5abde.js +1 -0
  143. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3548-36f746b1824004f2.js +1 -0
  144. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{3763-39a5174f6a3924db.js → 3763-aabe2703076636b0.js} +1 -1
  145. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3782-3e2acb5ed45b582b.js +1 -0
  146. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/449-f689774546860ca4.js +1 -0
  147. mage_ai/server/{frontend_dist/_next/static/chunks/5627-d5e559859dd0e1e0.js → frontend_dist_base_path_template/_next/static/chunks/5627-10e76bafa5a26f5f.js} +1 -1
  148. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{5699-e49718dfc9eb2854.js → 5699-e99379e332bd0b41.js} +1 -1
  149. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{7966-163da2621b8c987c.js → 7966-a5a7db345ce81263.js} +1 -1
  150. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{_app-90de19bc03f1484b.js → _app-0392ef723ea2c6f8.js} +2 -2
  151. mage_ai/server/{frontend_dist/_next/static/chunks/pages/compute-ed67fa8e81662e8b.js → frontend_dist_base_path_template/_next/static/chunks/pages/compute-9e2dea78024e3bb4.js} +1 -1
  152. mage_ai/server/{frontend_dist/_next/static/chunks/pages/files-e0ecd7ced09a63b2.js → frontend_dist_base_path_template/_next/static/chunks/pages/files-e08c7fe76f968f9c.js} +1 -1
  153. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/global-data-products/{[...slug]-c7a729477ecda50e.js → [...slug]-30c3807057a4e65b.js} +1 -1
  154. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{global-data-products-fd6ae6a358a60a0c.js → global-data-products-8dcb3b31af9e0e39.js} +1 -1
  155. mage_ai/server/{frontend_dist/_next/static/chunks/pages/global-hooks/[...slug]-8e50243797a7fe59.js → frontend_dist_base_path_template/_next/static/chunks/pages/global-hooks/[...slug]-85a64b64d27214b6.js} +1 -1
  156. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{global-hooks-d0c003446332dc0d.js → global-hooks-4ff959d51b8a9502.js} +1 -1
  157. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{files-a69ed8e9f814490c.js → files-d08a460641d0efaa.js} +1 -1
  158. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{overview-1aad7093c6d39257.js → overview-aae747f487e08d51.js} +1 -1
  159. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{pipeline-runs-528d30e0d13b0cc7.js → pipeline-runs-09a842d64a6ada62.js} +1 -1
  160. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{settings-fb9201d9cf63031d.js → settings-2e98e57d9376a458.js} +1 -1
  161. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/{[user]-000f5a980a07da39.js → [user]-7be6e41ad66089bb.js} +1 -1
  162. mage_ai/server/{frontend_dist/_next/static/chunks/pages/manage/users/new-e4e613f6e817a733.js → frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/new-4c088833063bfa07.js} +1 -1
  163. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{users-a5e9d77ed5b50205.js → users-b99379d0aa6a8c25.js} +1 -1
  164. mage_ai/server/{frontend_dist/_next/static/chunks/pages/manage-34d718b8a4066c23.js → frontend_dist_base_path_template/_next/static/chunks/pages/manage-868fcd8cbeb265f0.js} +1 -1
  165. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{oauth-3bfd1b8d7f036726.js → oauth-6ceceb62191dfe8a.js} +1 -1
  166. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/overview-e51cd04bd4d1fffe.js +1 -0
  167. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{pipeline-runs-5f8c100e648efa8a.js → pipeline-runs-2d0136b51b57de93.js} +1 -1
  168. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills/{[...slug]-688c652f3296bb9c.js → [...slug]-1ad5238742e25b4c.js} +1 -1
  169. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills-bd11e87d026bfbf9.js +1 -0
  170. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/{dashboard-1236e36d39b1637d.js → dashboard-0f4f47f721b0723f.js} +1 -1
  171. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-432da20df91511fb.js +1 -0
  172. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/logs-fe91dfb0091f6bc6.js +1 -0
  173. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-0d68d4bf6290fefb.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-cf794b2d22a80f31.js} +1 -1
  174. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-9254358d58f07714.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-a964caef91bed9e1.js} +1 -1
  175. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/{monitors-821001e690caebe2.js → monitors-80bebb4401eefe25.js} +1 -1
  176. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-2eae7cb017027682.js +1 -0
  177. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-2d4b2a0800a66b33.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-e3d7fdfeb7f806f7.js} +1 -1
  178. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/{settings-cd49372ae1702963.js → settings-0abf8a1b7243f93b.js} +1 -1
  179. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/syncs-707ed8ca942ca802.js +1 -0
  180. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers/{[...slug]-259143ed3cf59e31.js → [...slug]-8429f17d4146e1ec.js} +1 -1
  181. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-193045d9836d8d80.js +1 -0
  182. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines-b578b075a8d857e3.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines-38187954b6ec4b40.js} +1 -1
  183. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/platform/global-hooks/{[...slug]-5eeec927e4202b63.js → [...slug]-6834ae87bd668cb2.js} +1 -1
  184. mage_ai/server/{frontend_dist/_next/static/chunks/pages/platform/global-hooks-fbe9ad995d46d837.js → frontend_dist_base_path_template/_next/static/chunks/pages/platform/global-hooks-b3f7309a23e592b2.js} +1 -1
  185. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/account/{profile-fc659962d4015cb3.js → profile-f8b7374385e1f1bf.js} +1 -1
  186. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/platform/preferences-8de68502a9afa299.js +1 -0
  187. mage_ai/server/{frontend_dist/_next/static/chunks/pages/settings/platform/settings-a4f88c334414402b.js → frontend_dist_base_path_template/_next/static/chunks/pages/settings/platform/settings-50fb6a34f3913f1f.js} +1 -1
  188. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/permissions/{[...slug]-4deb9579ef99a3c6.js → [...slug]-2e5c098c21ea32b7.js} +1 -1
  189. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{permissions-e0cda2f2bfce8d61.js → permissions-54e4b15b9585bfc4.js} +1 -1
  190. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/preferences-040f83d75d0f6537.js +1 -0
  191. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/roles/{[...slug]-910257d16c604ebd.js → [...slug]-95088f43034e3c95.js} +1 -1
  192. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{roles-4f7a0756806cee34.js → roles-e9149e1fcf218f42.js} +1 -1
  193. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{sync-data-208d6f955204d704.js → sync-data-75b67ae4a00818ef.js} +1 -1
  194. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/users/{[...slug]-c89dc67e5a1706a8.js → [...slug]-557dda05ca6c6124.js} +1 -1
  195. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{users-5212c01a9dc558da.js → users-3ee783f5139c76a1.js} +1 -1
  196. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{sign-in-054b33312d3193c3.js → sign-in-593c40985d63fcf7.js} +1 -1
  197. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/templates/{[...slug]-b6ed6a5d818bfd20.js → [...slug]-252c4b6b818345d5.js} +1 -1
  198. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{templates-852357bc983af2ea.js → templates-ca528bc607753ab8.js} +1 -1
  199. mage_ai/server/{frontend_dist/_next/static/chunks/pages/terminal-1f9c56d671bbc67d.js → frontend_dist_base_path_template/_next/static/chunks/pages/terminal-287362c1defcc96b.js} +1 -1
  200. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/test-2f83af8c9f1378fe.js +1 -0
  201. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/triggers-d9de73fb799efed8.js +1 -0
  202. mage_ai/server/{frontend_dist/_next/static/chunks/pages/version-control-ae3469b992a341d6.js → frontend_dist_base_path_template/_next/static/chunks/pages/version-control-573f0225d7a703ed.js} +1 -1
  203. mage_ai/server/frontend_dist_base_path_template/_next/static/dxnSzgIvSG4Ke5LM-tPQX/_buildManifest.js +1 -0
  204. mage_ai/server/frontend_dist_base_path_template/block-layout.html +3 -3
  205. mage_ai/server/frontend_dist_base_path_template/compute.html +6 -6
  206. mage_ai/server/frontend_dist_base_path_template/files.html +6 -6
  207. mage_ai/server/frontend_dist_base_path_template/global-data-products/[...slug].html +6 -6
  208. mage_ai/server/frontend_dist_base_path_template/global-data-products.html +6 -6
  209. mage_ai/server/frontend_dist_base_path_template/global-hooks/[...slug].html +6 -6
  210. mage_ai/server/frontend_dist_base_path_template/global-hooks.html +6 -6
  211. mage_ai/server/frontend_dist_base_path_template/index.html +3 -3
  212. mage_ai/server/frontend_dist_base_path_template/manage/files.html +6 -6
  213. mage_ai/server/frontend_dist_base_path_template/manage/overview.html +6 -6
  214. mage_ai/server/frontend_dist_base_path_template/manage/pipeline-runs.html +6 -6
  215. mage_ai/server/frontend_dist_base_path_template/manage/settings.html +6 -6
  216. mage_ai/server/frontend_dist_base_path_template/manage/users/[user].html +6 -6
  217. mage_ai/server/frontend_dist_base_path_template/manage/users/new.html +6 -6
  218. mage_ai/server/frontend_dist_base_path_template/manage/users.html +6 -6
  219. mage_ai/server/frontend_dist_base_path_template/manage.html +6 -6
  220. mage_ai/server/frontend_dist_base_path_template/oauth.html +5 -5
  221. mage_ai/server/frontend_dist_base_path_template/overview.html +6 -6
  222. mage_ai/server/frontend_dist_base_path_template/pipeline-runs.html +6 -6
  223. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills/[...slug].html +6 -6
  224. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills.html +6 -6
  225. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/dashboard.html +6 -6
  226. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/edit.html +3 -3
  227. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/logs.html +6 -6
  228. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runs.html +6 -6
  229. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runtime.html +6 -6
  230. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors.html +6 -6
  231. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs/[run].html +6 -6
  232. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs.html +6 -6
  233. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/settings.html +6 -6
  234. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/syncs.html +6 -6
  235. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers/[...slug].html +6 -6
  236. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers.html +6 -6
  237. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline].html +3 -3
  238. mage_ai/server/frontend_dist_base_path_template/pipelines.html +6 -6
  239. mage_ai/server/frontend_dist_base_path_template/platform/global-hooks/[...slug].html +6 -6
  240. mage_ai/server/frontend_dist_base_path_template/platform/global-hooks.html +6 -6
  241. mage_ai/server/frontend_dist_base_path_template/settings/account/profile.html +6 -6
  242. mage_ai/server/frontend_dist_base_path_template/settings/platform/preferences.html +6 -6
  243. mage_ai/server/frontend_dist_base_path_template/settings/platform/settings.html +6 -6
  244. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions/[...slug].html +6 -6
  245. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions.html +6 -6
  246. mage_ai/server/frontend_dist_base_path_template/settings/workspace/preferences.html +6 -6
  247. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles/[...slug].html +6 -6
  248. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles.html +6 -6
  249. mage_ai/server/frontend_dist_base_path_template/settings/workspace/sync-data.html +6 -6
  250. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users/[...slug].html +6 -6
  251. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users.html +6 -6
  252. mage_ai/server/frontend_dist_base_path_template/settings.html +3 -3
  253. mage_ai/server/frontend_dist_base_path_template/sign-in.html +7 -7
  254. mage_ai/server/frontend_dist_base_path_template/templates/[...slug].html +6 -6
  255. mage_ai/server/frontend_dist_base_path_template/templates.html +6 -6
  256. mage_ai/server/frontend_dist_base_path_template/terminal.html +6 -6
  257. mage_ai/server/frontend_dist_base_path_template/test.html +3 -3
  258. mage_ai/server/frontend_dist_base_path_template/triggers.html +6 -6
  259. mage_ai/server/frontend_dist_base_path_template/v2/canvas.html +2 -2
  260. mage_ai/server/frontend_dist_base_path_template/v2.html +2 -2
  261. mage_ai/server/frontend_dist_base_path_template/version-control.html +6 -6
  262. mage_ai/server/server.py +9 -0
  263. mage_ai/services/teams/config.py +13 -2
  264. mage_ai/services/teams/teams.py +13 -11
  265. mage_ai/shared/constants.py +3 -1
  266. mage_ai/shared/enum.py +2 -5
  267. mage_ai/shared/environments.py +27 -3
  268. mage_ai/streaming/sinks/elasticsearch.py +15 -5
  269. mage_ai/tests/api/endpoints/test_blocks.py +1 -101
  270. mage_ai/tests/api/endpoints/test_configuration_options.py +1 -48
  271. mage_ai/tests/api/endpoints/test_configuration_options_project_platform.py +68 -0
  272. mage_ai/tests/api/endpoints/test_custom_designs.py +1 -106
  273. mage_ai/tests/api/endpoints/test_custom_designs_project_platform.py +132 -0
  274. mage_ai/tests/api/endpoints/test_dbt_blocks.py +111 -0
  275. mage_ai/tests/api/endpoints/test_file_contents.py +0 -48
  276. mage_ai/tests/api/endpoints/test_file_contents_with_project_platform.py +66 -0
  277. mage_ai/tests/api/endpoints/test_pipelines.py +0 -134
  278. mage_ai/tests/api/endpoints/test_pipelines_with_project_platform.py +143 -0
  279. mage_ai/tests/data_preparation/models/block/hook/test_hook_block.py +3 -3
  280. mage_ai/tests/data_preparation/models/test_block.py +7 -0
  281. mage_ai/tests/data_preparation/test_repo_manager.py +0 -63
  282. mage_ai/tests/data_preparation/test_repo_manager_project_platform.py +67 -0
  283. mage_ai/tests/data_preparation/test_variable_manager.py +0 -51
  284. mage_ai/tests/data_preparation/test_variable_manager_project_platform.py +41 -0
  285. mage_ai/tests/io/create_table/test_postgresql.py +27 -0
  286. mage_ai/tests/orchestration/notification/test_config.py +3 -3
  287. mage_ai-0.9.75.dist-info/METADATA +377 -0
  288. {mage_ai-0.9.74.dist-info → mage_ai-0.9.75.dist-info}/RECORD +296 -289
  289. {mage_ai-0.9.74.dist-info → mage_ai-0.9.75.dist-info}/WHEEL +1 -1
  290. mage_ai/server/frontend_dist/_next/static/chunks/1557-1ad0c64c2d08e569.js +0 -1
  291. mage_ai/server/frontend_dist/_next/static/chunks/2717-d65056b6b5e124eb.js +0 -1
  292. mage_ai/server/frontend_dist/_next/static/chunks/3548-b2c5edfb710886a6.js +0 -1
  293. mage_ai/server/frontend_dist/_next/static/chunks/3782-4b3091e550f809a2.js +0 -1
  294. mage_ai/server/frontend_dist/_next/static/chunks/pages/overview-69af3253ad0d0d89.js +0 -1
  295. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills-fe112809feb25b05.js +0 -1
  296. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-e9ca358209cdf93d.js +0 -1
  297. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/logs-a29f1615d2e7d330.js +0 -1
  298. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-6d382ae5bad9745c.js +0 -1
  299. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/syncs-135be8974f7f5f2b.js +0 -1
  300. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-3af13e89adff4d6c.js +0 -1
  301. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/platform/preferences-058d283ee178c038.js +0 -1
  302. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/preferences-7b02bb70462144cb.js +0 -1
  303. mage_ai/server/frontend_dist/_next/static/chunks/pages/test-86b12cc12d4a625f.js +0 -1
  304. mage_ai/server/frontend_dist/_next/static/chunks/pages/triggers-2481c40b18d5b6d4.js +0 -1
  305. mage_ai/server/frontend_dist/_next/static/pLWT6Sqd09xYpufCVIqnz/_buildManifest.js +0 -1
  306. mage_ai/server/frontend_dist_base_path_template/_next/static/JQewSAObpbhO0wrdAM6Ng/_buildManifest.js +0 -1
  307. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1557-1ad0c64c2d08e569.js +0 -1
  308. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/2717-d65056b6b5e124eb.js +0 -1
  309. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3548-b2c5edfb710886a6.js +0 -1
  310. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3782-4b3091e550f809a2.js +0 -1
  311. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/overview-69af3253ad0d0d89.js +0 -1
  312. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills-fe112809feb25b05.js +0 -1
  313. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-e9ca358209cdf93d.js +0 -1
  314. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/logs-a29f1615d2e7d330.js +0 -1
  315. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-6d382ae5bad9745c.js +0 -1
  316. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/syncs-135be8974f7f5f2b.js +0 -1
  317. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-3af13e89adff4d6c.js +0 -1
  318. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/platform/preferences-058d283ee178c038.js +0 -1
  319. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/preferences-7b02bb70462144cb.js +0 -1
  320. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/test-86b12cc12d4a625f.js +0 -1
  321. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/triggers-2481c40b18d5b6d4.js +0 -1
  322. mage_ai-0.9.74.dist-info/METADATA +0 -544
  323. /mage_ai/server/frontend_dist/_next/static/{pLWT6Sqd09xYpufCVIqnz → 38-PtskJFUTYUpRhT1qF_}/_ssgManifest.js +0 -0
  324. /mage_ai/server/frontend_dist/_next/static/chunks/pages/{_app-5bdff745074fb350.js.LICENSE.txt → _app-13bf3b7dcef50c29.js.LICENSE.txt} +0 -0
  325. /mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{_app-90de19bc03f1484b.js.LICENSE.txt → _app-0392ef723ea2c6f8.js.LICENSE.txt} +0 -0
  326. /mage_ai/server/frontend_dist_base_path_template/_next/static/{JQewSAObpbhO0wrdAM6Ng → dxnSzgIvSG4Ke5LM-tPQX}/_ssgManifest.js +0 -0
  327. {mage_ai-0.9.74.dist-info → mage_ai-0.9.75.dist-info}/LICENSE +0 -0
  328. {mage_ai-0.9.74.dist-info → mage_ai-0.9.75.dist-info}/entry_points.txt +0 -0
  329. {mage_ai-0.9.74.dist-info → mage_ai-0.9.75.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,66 @@
1
+ import urllib.parse
2
+ from unittest.mock import patch
3
+
4
+ from mage_ai.tests.api.endpoints.mixins import (
5
+ BaseAPIEndpointTest,
6
+ build_detail_endpoint_tests,
7
+ build_update_endpoint_tests,
8
+ )
9
+ from mage_ai.tests.shared.mixins import ProjectPlatformMixin
10
+
11
+
12
+ def get_resource_id(self) -> str:
13
+ block = list(self.pipeline.blocks_by_uuid.values())[0]
14
+ pk = block.file_path.replace(f'{self.pipeline.repo_path}/', '')
15
+ return urllib.parse.quote_plus(pk)
16
+
17
+
18
+ def get_model_before_update(self):
19
+ block = list(self.pipeline.blocks_by_uuid.values())[0]
20
+ return block.file.content()
21
+
22
+
23
+ class FileContentWithProjectPlatformAPIEndpointTest(BaseAPIEndpointTest, ProjectPlatformMixin):
24
+ def setUp(self):
25
+ with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
26
+ with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
27
+ super().setUp()
28
+ self.setup_final()
29
+
30
+ def tearDown(self):
31
+ with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
32
+ with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
33
+ self.teardown_final()
34
+ super().tearDown()
35
+
36
+
37
+ build_detail_endpoint_tests(
38
+ FileContentWithProjectPlatformAPIEndpointTest,
39
+ resource='file_content',
40
+ get_resource_id=get_resource_id,
41
+ result_keys_to_compare=[
42
+ 'content',
43
+ 'name',
44
+ 'path',
45
+ ],
46
+ patch_function_settings=[
47
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
48
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
49
+ ('mage_ai.api.resources.FileContentResource.ensure_file_is_in_project', lambda _x: True),
50
+ ],
51
+ )
52
+
53
+
54
+ build_update_endpoint_tests(
55
+ FileContentWithProjectPlatformAPIEndpointTest,
56
+ resource='file_content',
57
+ get_resource_id=get_resource_id,
58
+ build_payload=lambda self: dict(content=self.faker.unique.name()),
59
+ get_model_before_update=get_model_before_update,
60
+ assert_after_update=lambda self, result, model, mocks: model != result['content'],
61
+ patch_function_settings=[
62
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
63
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
64
+ ('mage_ai.api.resources.FileContentResource.ensure_file_is_in_project', lambda _x: True),
65
+ ],
66
+ )
@@ -1,5 +1,3 @@
1
- from unittest.mock import patch
2
-
3
1
  from mage_ai.data_preparation.models.pipeline import Pipeline
4
2
  from mage_ai.tests.api.endpoints.mixins import (
5
3
  BaseAPIEndpointTest,
@@ -9,7 +7,6 @@ from mage_ai.tests.api.endpoints.mixins import (
9
7
  build_list_endpoint_tests,
10
8
  build_update_endpoint_tests,
11
9
  )
12
- from mage_ai.tests.shared.mixins import ProjectPlatformMixin
13
10
 
14
11
 
15
12
  class PipelineAPIEndpointTest(BaseAPIEndpointTest):
@@ -111,134 +108,3 @@ build_delete_endpoint_tests(
111
108
  Pipeline.get_all_pipelines(self.repo_path),
112
109
  ) == 1,
113
110
  )
114
-
115
-
116
- class PipelineWithProjectPlatformAPIEndpointTest(BaseAPIEndpointTest, ProjectPlatformMixin):
117
- def setUp(self):
118
- with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
119
- with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
120
- super().setUp()
121
- self.setup_final()
122
-
123
- def tearDown(self):
124
- with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
125
- with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
126
- self.teardown_final()
127
- super().tearDown()
128
-
129
-
130
- build_list_endpoint_tests(
131
- PipelineWithProjectPlatformAPIEndpointTest,
132
- resource='pipelines',
133
- list_count=3,
134
- result_keys_to_compare=[
135
- 'blocks',
136
- 'callbacks',
137
- 'concurrency_config',
138
- 'conditionals',
139
- 'created_at',
140
- 'data_integration',
141
- 'description',
142
- 'executor_config',
143
- 'executor_count',
144
- 'executor_type',
145
- 'name',
146
- 'notification_config',
147
- 'retry_config',
148
- 'run_pipeline_in_one_process',
149
- 'spark_config',
150
- 'tags',
151
- 'type',
152
- 'updated_at',
153
- 'uuid',
154
- 'widgets',
155
- ],
156
- patch_function_settings=[
157
- ('mage_ai.settings.repo.project_platform_activated', lambda: True),
158
- ('mage_ai.settings.platform.project_platform_activated', lambda: True),
159
- ],
160
- )
161
-
162
-
163
- build_create_endpoint_tests(
164
- PipelineWithProjectPlatformAPIEndpointTest,
165
- resource='pipelines',
166
- assert_after_create_count=lambda self, mocks: len(
167
- Pipeline.get_all_pipelines(self.repo_path),
168
- ) == 4,
169
- assert_before_create_count=lambda self, mocks: len(
170
- Pipeline.get_all_pipelines(self.repo_path),
171
- ) == 3,
172
- build_payload=lambda self: dict(
173
- name=self.faker.unique.name(),
174
- ),
175
- patch_function_settings=[
176
- ('mage_ai.settings.repo.project_platform_activated', lambda: True),
177
- ('mage_ai.settings.platform.project_platform_activated', lambda: True),
178
- ],
179
- )
180
-
181
-
182
- build_detail_endpoint_tests(
183
- PipelineWithProjectPlatformAPIEndpointTest,
184
- resource='pipelines',
185
- get_resource_id=lambda self: self.pipeline.uuid,
186
- result_keys_to_compare=[
187
- 'blocks',
188
- 'callbacks',
189
- 'concurrency_config',
190
- 'conditionals',
191
- 'created_at',
192
- 'data_integration',
193
- 'description',
194
- 'executor_config',
195
- 'executor_count',
196
- 'executor_type',
197
- 'extensions',
198
- 'name',
199
- 'notification_config',
200
- 'retry_config',
201
- 'run_pipeline_in_one_process',
202
- 'spark_config',
203
- 'tags',
204
- 'type',
205
- 'updated_at',
206
- 'uuid',
207
- 'widgets',
208
- ],
209
- patch_function_settings=[
210
- ('mage_ai.settings.repo.project_platform_activated', lambda: True),
211
- ('mage_ai.settings.platform.project_platform_activated', lambda: True),
212
- ],
213
- )
214
-
215
-
216
- build_update_endpoint_tests(
217
- PipelineWithProjectPlatformAPIEndpointTest,
218
- resource='pipelines',
219
- get_resource_id=lambda self: self.pipeline.uuid,
220
- build_payload=lambda self: dict(name=self.faker.unique.name()),
221
- get_model_before_update=lambda self: self.pipeline,
222
- assert_after_update=lambda self, result, model, mocks: model.name != result['name'],
223
- patch_function_settings=[
224
- ('mage_ai.settings.repo.project_platform_activated', lambda: True),
225
- ('mage_ai.settings.platform.project_platform_activated', lambda: True),
226
- ],
227
- )
228
-
229
-
230
- build_delete_endpoint_tests(
231
- PipelineWithProjectPlatformAPIEndpointTest,
232
- resource='pipelines',
233
- get_resource_id=lambda self: self.pipeline.uuid,
234
- assert_after_delete_count=lambda self, mocks: len(
235
- Pipeline.get_all_pipelines(self.repo_path),
236
- ) == 2,
237
- assert_before_delete_count=lambda self, mocks: len(
238
- Pipeline.get_all_pipelines(self.repo_path),
239
- ) == 3,
240
- patch_function_settings=[
241
- ('mage_ai.settings.repo.project_platform_activated', lambda: True),
242
- ('mage_ai.settings.platform.project_platform_activated', lambda: True),
243
- ],
244
- )
@@ -0,0 +1,143 @@
1
+ from unittest.mock import patch
2
+
3
+ from mage_ai.data_preparation.models.pipeline import Pipeline
4
+ from mage_ai.tests.api.endpoints.mixins import (
5
+ BaseAPIEndpointTest,
6
+ build_create_endpoint_tests,
7
+ build_delete_endpoint_tests,
8
+ build_detail_endpoint_tests,
9
+ build_list_endpoint_tests,
10
+ build_update_endpoint_tests,
11
+ )
12
+ from mage_ai.tests.shared.mixins import ProjectPlatformMixin
13
+
14
+
15
+ class PipelineWithProjectPlatformAPIEndpointTest(BaseAPIEndpointTest, ProjectPlatformMixin):
16
+ def setUp(self):
17
+ with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
18
+ with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
19
+ super().setUp()
20
+ self.setup_final()
21
+
22
+ def tearDown(self):
23
+ with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
24
+ with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
25
+ self.teardown_final()
26
+ super().tearDown()
27
+
28
+
29
+ build_list_endpoint_tests(
30
+ PipelineWithProjectPlatformAPIEndpointTest,
31
+ resource='pipelines',
32
+ list_count=3,
33
+ result_keys_to_compare=[
34
+ 'blocks',
35
+ 'callbacks',
36
+ 'concurrency_config',
37
+ 'conditionals',
38
+ 'created_at',
39
+ 'data_integration',
40
+ 'description',
41
+ 'executor_config',
42
+ 'executor_count',
43
+ 'executor_type',
44
+ 'name',
45
+ 'notification_config',
46
+ 'retry_config',
47
+ 'run_pipeline_in_one_process',
48
+ 'spark_config',
49
+ 'tags',
50
+ 'type',
51
+ 'updated_at',
52
+ 'uuid',
53
+ 'widgets',
54
+ ],
55
+ patch_function_settings=[
56
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
57
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
58
+ ],
59
+ )
60
+
61
+
62
+ build_create_endpoint_tests(
63
+ PipelineWithProjectPlatformAPIEndpointTest,
64
+ resource='pipelines',
65
+ assert_after_create_count=lambda self, mocks: len(
66
+ Pipeline.get_all_pipelines(self.repo_path),
67
+ ) == 4,
68
+ assert_before_create_count=lambda self, mocks: len(
69
+ Pipeline.get_all_pipelines(self.repo_path),
70
+ ) == 3,
71
+ build_payload=lambda self: dict(
72
+ name=self.faker.unique.name(),
73
+ ),
74
+ patch_function_settings=[
75
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
76
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
77
+ ],
78
+ )
79
+
80
+
81
+ build_detail_endpoint_tests(
82
+ PipelineWithProjectPlatformAPIEndpointTest,
83
+ resource='pipelines',
84
+ get_resource_id=lambda self: self.pipeline.uuid,
85
+ result_keys_to_compare=[
86
+ 'blocks',
87
+ 'callbacks',
88
+ 'concurrency_config',
89
+ 'conditionals',
90
+ 'created_at',
91
+ 'data_integration',
92
+ 'description',
93
+ 'executor_config',
94
+ 'executor_count',
95
+ 'executor_type',
96
+ 'extensions',
97
+ 'name',
98
+ 'notification_config',
99
+ 'retry_config',
100
+ 'run_pipeline_in_one_process',
101
+ 'spark_config',
102
+ 'tags',
103
+ 'type',
104
+ 'updated_at',
105
+ 'uuid',
106
+ 'widgets',
107
+ ],
108
+ patch_function_settings=[
109
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
110
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
111
+ ],
112
+ )
113
+
114
+
115
+ build_update_endpoint_tests(
116
+ PipelineWithProjectPlatformAPIEndpointTest,
117
+ resource='pipelines',
118
+ get_resource_id=lambda self: self.pipeline.uuid,
119
+ build_payload=lambda self: dict(name=self.faker.unique.name()),
120
+ get_model_before_update=lambda self: self.pipeline,
121
+ assert_after_update=lambda self, result, model, mocks: model.name != result['name'],
122
+ patch_function_settings=[
123
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
124
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
125
+ ],
126
+ )
127
+
128
+
129
+ build_delete_endpoint_tests(
130
+ PipelineWithProjectPlatformAPIEndpointTest,
131
+ resource='pipelines',
132
+ get_resource_id=lambda self: self.pipeline.uuid,
133
+ assert_after_delete_count=lambda self, mocks: len(
134
+ Pipeline.get_all_pipelines(self.repo_path),
135
+ ) == 2,
136
+ assert_before_delete_count=lambda self, mocks: len(
137
+ Pipeline.get_all_pipelines(self.repo_path),
138
+ ) == 3,
139
+ patch_function_settings=[
140
+ ('mage_ai.settings.repo.project_platform_activated', lambda: True),
141
+ ('mage_ai.settings.platform.project_platform_activated', lambda: True),
142
+ ],
143
+ )
@@ -5,7 +5,7 @@ from faker import Faker
5
5
  from mage_ai.data_preparation.models.block.hook.block import HookBlock
6
6
  from mage_ai.data_preparation.models.constants import BlockType
7
7
  from mage_ai.data_preparation.models.global_hooks.models import HookStatus, HookStrategy
8
- from mage_ai.shared.constants import ENV_TEST
8
+ from mage_ai.shared.constants import ENV_TEST_MAGE
9
9
  from mage_ai.tests.api.operations.test_base import BaseApiTestCase
10
10
  from mage_ai.tests.factory import create_pipeline_with_blocks
11
11
  from mage_ai.tests.shared.mixins import build_hooks
@@ -46,7 +46,7 @@ class HookBlockTest(BaseApiTestCase):
46
46
  check_status=True,
47
47
  configuration={},
48
48
  context={},
49
- env=ENV_TEST,
49
+ env=ENV_TEST_MAGE,
50
50
  error_on_failure=True,
51
51
  mage=1,
52
52
  pipeline_uuid='',
@@ -85,7 +85,7 @@ class HookBlockTest(BaseApiTestCase):
85
85
  check_status=True,
86
86
  configuration={},
87
87
  context={},
88
- env=ENV_TEST,
88
+ env=ENV_TEST_MAGE,
89
89
  error_on_failure=True,
90
90
  mage=1,
91
91
  pipeline_uuid='',
@@ -746,6 +746,13 @@ def test_output(output, *args) -> None:
746
746
  ),
747
747
  )
748
748
 
749
+ def test_none_block_type(self):
750
+ block = Block(name='test block', uuid='test uuid', block_type=None)
751
+ self.assertEqual(
752
+ block.get_typed_content('test content'),
753
+ '@None\ntest content',
754
+ )
755
+
749
756
 
750
757
  class BlockProjectPlatformTests(ProjectPlatformMixin):
751
758
  def test_file_path(self):
@@ -10,13 +10,10 @@ from mage_ai.data_preparation.repo_manager import (
10
10
  get_project_uuid,
11
11
  get_repo_config,
12
12
  init_project_uuid,
13
- init_repo,
14
13
  set_project_uuid_from_metadata,
15
14
  )
16
15
  from mage_ai.settings.repo import MAGE_DATA_DIR_ENV_VAR
17
- from mage_ai.settings.utils import base_repo_path
18
16
  from mage_ai.tests.base_test import DBTestCase
19
- from mage_ai.tests.shared.mixins import ProjectPlatformMixin
20
17
 
21
18
 
22
19
  def mock_uuid_value():
@@ -134,63 +131,3 @@ class RepoManagerTest(DBTestCase):
134
131
  self.assertEqual(get_project_uuid(), '000000')
135
132
 
136
133
  os.remove(metadata_path)
137
-
138
-
139
- @patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
140
- @patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
141
- class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
142
- def test_init(self):
143
- repo = RepoConfig(root_project=False)
144
- self.assertFalse(repo.root_project)
145
- self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
146
- self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
147
-
148
- repo = RepoConfig(root_project=True)
149
- self.assertTrue(repo.root_project)
150
- self.assertEqual(repo.repo_path, base_repo_path())
151
- self.assertEqual(repo.variables_dir, base_repo_path())
152
-
153
- def test_from_dict(self):
154
- repo = RepoConfig.from_dict(dict(), root_project=False)
155
- self.assertFalse(repo.root_project)
156
- self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
157
- self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
158
-
159
- repo = RepoConfig.from_dict(dict(), root_project=True)
160
- self.assertTrue(repo.root_project)
161
- self.assertEqual(repo.repo_path, base_repo_path())
162
- self.assertEqual(repo.variables_dir, base_repo_path())
163
-
164
- def test_metadata_path(self):
165
- repo = RepoConfig(root_project=False)
166
- self.assertEqual(
167
- repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
168
- )
169
-
170
- repo = RepoConfig(root_project=True)
171
- self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
172
-
173
- def test_get_repo_config(self):
174
- repo = get_repo_config(root_project=False)
175
- self.assertFalse(repo.root_project)
176
- self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
177
- self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
178
-
179
- repo = get_repo_config(root_project=True)
180
- self.assertTrue(repo.root_project)
181
- self.assertEqual(repo.repo_path, base_repo_path())
182
- self.assertEqual(repo.variables_dir, base_repo_path())
183
-
184
- def test_init_repo(self):
185
- path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
186
- try:
187
- shutil.rmtree(path)
188
- except Exception:
189
- pass
190
- with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
191
- init_repo(path, root_project=True)
192
- mock_get_repo_config.assert_called_once_with(path, root_project=True)
193
- try:
194
- shutil.rmtree(path)
195
- except Exception:
196
- pass
@@ -0,0 +1,67 @@
1
+ import os
2
+ import shutil
3
+ from unittest.mock import patch
4
+
5
+ from mage_ai.data_preparation.repo_manager import RepoConfig, get_repo_config, init_repo
6
+ from mage_ai.settings.utils import base_repo_path
7
+ from mage_ai.tests.shared.mixins import ProjectPlatformMixin
8
+
9
+
10
+ @patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
11
+ @patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
12
+ class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
13
+ def test_init(self):
14
+ repo = RepoConfig(root_project=False)
15
+ self.assertFalse(repo.root_project)
16
+ self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
17
+ self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
18
+
19
+ repo = RepoConfig(root_project=True)
20
+ self.assertTrue(repo.root_project)
21
+ self.assertEqual(repo.repo_path, base_repo_path())
22
+ self.assertEqual(repo.variables_dir, base_repo_path())
23
+
24
+ def test_from_dict(self):
25
+ repo = RepoConfig.from_dict(dict(), root_project=False)
26
+ self.assertFalse(repo.root_project)
27
+ self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
28
+ self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
29
+
30
+ repo = RepoConfig.from_dict(dict(), root_project=True)
31
+ self.assertTrue(repo.root_project)
32
+ self.assertEqual(repo.repo_path, base_repo_path())
33
+ self.assertEqual(repo.variables_dir, base_repo_path())
34
+
35
+ def test_metadata_path(self):
36
+ repo = RepoConfig(root_project=False)
37
+ self.assertEqual(
38
+ repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
39
+ )
40
+
41
+ repo = RepoConfig(root_project=True)
42
+ self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
43
+
44
+ def test_get_repo_config(self):
45
+ repo = get_repo_config(root_project=False)
46
+ self.assertFalse(repo.root_project)
47
+ self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
48
+ self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
49
+
50
+ repo = get_repo_config(root_project=True)
51
+ self.assertTrue(repo.root_project)
52
+ self.assertEqual(repo.repo_path, base_repo_path())
53
+ self.assertEqual(repo.variables_dir, base_repo_path())
54
+
55
+ def test_init_repo(self):
56
+ path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
57
+ try:
58
+ shutil.rmtree(path)
59
+ except Exception:
60
+ pass
61
+ with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
62
+ init_repo(path, root_project=True)
63
+ mock_get_repo_config.assert_called_once_with(path, root_project=True)
64
+ try:
65
+ shutil.rmtree(path)
66
+ except Exception:
67
+ pass
@@ -1,5 +1,3 @@
1
- from unittest.mock import patch
2
-
3
1
  import pandas as pd
4
2
  from pandas.testing import assert_frame_equal
5
3
 
@@ -10,12 +8,10 @@ from mage_ai.data_preparation.repo_manager import get_repo_config
10
8
  from mage_ai.data_preparation.variable_manager import (
11
9
  VariableManager,
12
10
  get_global_variable,
13
- get_global_variables,
14
11
  set_global_variable,
15
12
  )
16
13
  from mage_ai.settings.repo import set_repo_path
17
14
  from mage_ai.tests.base_test import DBTestCase
18
- from mage_ai.tests.shared.mixins import ProjectPlatformMixin
19
15
 
20
16
 
21
17
  class VariableManagerTest(DBTestCase):
@@ -128,50 +124,3 @@ class VariableManagerTest(DBTestCase):
128
124
  # get_global_variables(None, pipeline=pipeline), pipeline.variables
129
125
  # )
130
126
  # self.assertEqual(get_global_variables(pipeline.uuid), pipeline.variables)
131
-
132
-
133
- class VariableManagerProjectPlatformTests(ProjectPlatformMixin):
134
- def test_get_global_variable(self):
135
- with patch(
136
- "mage_ai.data_preparation.variable_manager.project_platform_activated",
137
- lambda: True,
138
- ):
139
- with patch(
140
- "mage_ai.data_preparation.models.pipeline.project_platform_activated",
141
- lambda: True,
142
- ):
143
- for settings in self.repo_paths.values():
144
- pipeline = Pipeline.create(
145
- self.faker.unique.name(),
146
- repo_path=settings["full_path"],
147
- )
148
- value = self.faker.unique.name()
149
- pipeline.variables = dict(mage=value)
150
- pipeline.save()
151
-
152
- self.assertEqual(get_global_variable(pipeline.uuid, "mage"), value)
153
-
154
- def test_get_global_variables(self):
155
- with patch(
156
- "mage_ai.data_preparation.variable_manager.project_platform_activated",
157
- lambda: True,
158
- ):
159
- with patch(
160
- "mage_ai.data_preparation.models.pipeline.project_platform_activated",
161
- lambda: True,
162
- ):
163
- for settings in self.repo_paths.values():
164
- pipeline = Pipeline.create(
165
- self.faker.unique.name(),
166
- repo_path=settings["full_path"],
167
- )
168
- pipeline.variables = self.faker.unique.name()
169
- pipeline.save()
170
-
171
- self.assertEqual(
172
- get_global_variables(None, pipeline=pipeline),
173
- pipeline.variables,
174
- )
175
- self.assertEqual(
176
- get_global_variables(pipeline.uuid), pipeline.variables
177
- )
@@ -0,0 +1,41 @@
1
+ from unittest.mock import patch
2
+
3
+ from mage_ai.data_preparation.models.pipeline import Pipeline
4
+ from mage_ai.data_preparation.variable_manager import (
5
+ get_global_variable,
6
+ get_global_variables,
7
+ )
8
+ from mage_ai.tests.shared.mixins import ProjectPlatformMixin
9
+
10
+
11
+ @patch('mage_ai.data_preparation.models.pipeline.project_platform_activated', lambda: True)
12
+ @patch('mage_ai.data_preparation.variable_manager.project_platform_activated', lambda: True)
13
+ class VariableManagerProjectPlatformTests(ProjectPlatformMixin):
14
+ def test_get_global_variable(self):
15
+ for settings in self.repo_paths.values():
16
+ pipeline = Pipeline.create(
17
+ self.faker.unique.name(),
18
+ repo_path=settings["full_path"],
19
+ )
20
+ value = self.faker.unique.name()
21
+ pipeline.variables = dict(mage=value)
22
+ pipeline.save()
23
+
24
+ self.assertEqual(get_global_variable(pipeline.uuid, "mage"), value)
25
+
26
+ def test_get_global_variables(self):
27
+ for settings in self.repo_paths.values():
28
+ pipeline = Pipeline.create(
29
+ self.faker.unique.name(),
30
+ repo_path=settings["full_path"],
31
+ )
32
+ pipeline.variables = self.faker.unique.name()
33
+ pipeline.save()
34
+
35
+ self.assertEqual(
36
+ get_global_variables(None, pipeline=pipeline),
37
+ pipeline.variables,
38
+ )
39
+ self.assertEqual(
40
+ get_global_variables(pipeline.uuid), pipeline.variables
41
+ )