mage-ai 0.9.49__py3-none-any.whl → 0.9.51__py3-none-any.whl

mage_ai/api/operations/base.py

@@ -42,6 +42,7 @@ from mage_ai.orchestration.db import db_connection
 from mage_ai.orchestration.db.errors import DoesNotExistError
 from mage_ai.settings import REQUIRE_USER_PERMISSIONS
 from mage_ai.shared.array import flatten
+from mage_ai.shared.environments import is_debug
 from mage_ai.shared.hash import ignore_keys, merge_dict
 from mage_ai.shared.strings import classify
 
@@ -325,9 +326,17 @@ class BaseOperation():
                 resource_parent_id=self.resource_parent_id,
                 resource_parent_type=self.__resource_parent_entity_name(),
                 resources=resources,
-                user=dict(id=self.user.id) if self.user else None,
+                user=dict(
+                    avatar=self.user.avatar,
+                    first_name=self.user.first_name,
+                    id=self.user.id,
+                    last_name=self.user.last_name,
+                    username=self.user.username,
+                ) if self.user else None,
             )
-        except Exception:
+        except Exception as err:
+            if is_debug():
+                raise err
             hooks = []
 
         try:

mage_ai/api/presenters/WorkspacePresenter.py

@@ -19,6 +19,7 @@ class WorkspacePresenter(BasePresenter):
                 'name',
                 'repo_path',
                 'success',
+                'url',
                 *[f.name for f in fields(KubernetesWorkspaceConfig)],
                 *[f.name for f in fields(CloudRunWorkspaceConfig)],
                 *[f.name for f in fields(EcsWorkspaceConfig)],

mage_ai/api/resources/PipelineResource.py

@@ -207,25 +207,6 @@ class PipelineResource(BaseResource):
             )
 
         pipelines = [p for p in pipelines if p is not None]
-        if sorts:
-            def _sort_key(p, sorts=sorts, reverse_sort=reverse_sort):
-                bools = []
-                vals = []
-                for k in sorts:
-                    if hasattr(p, k):
-                        val = getattr(p, k)
-                        vals.append(val)
-                        bools.append(val is None if not reverse_sort else val is not None)
-                    else:
-                        bools.append(False)
-
-                return tuple(bools + vals)
-
-            pipelines = sorted(
-                pipelines,
-                key=_sort_key,
-                reverse=reverse_sort,
-            )
 
         @safe_db_query
         def query_pipeline_schedules(pipeline_uuids):
@@ -287,6 +268,34 @@ class PipelineResource(BaseResource):
                 if pipeline.uuid in history_by_pipeline_uuid:
                     pipeline.history = history_by_pipeline_uuid.get(pipeline.uuid)
 
+        if sorts:
+            def _sort_key(p, sorts=sorts, reverse_sort=reverse_sort):
+                bools = []
+                vals = []
+                for k in sorts:
+                    if 'blocks' == k.lower():
+                        val = len(p.blocks_by_uuid)
+                        vals.append(val)
+                        bools.append(val is None if not reverse_sort else val is not None)
+                    elif 'triggers' == k.lower():
+                        val = len(p.blocks_by_uuid)
+                        vals.append(val)
+                        bools.append(val is None if not reverse_sort else val is not None)
+                    elif hasattr(p, k):
+                        val = getattr(p, k)
+                        vals.append(val)
+                        bools.append(val is None if not reverse_sort else val is not None)
+                    else:
+                        bools.append(False)
+
+                return tuple(bools + vals)
+
+            pipelines = sorted(
+                pipelines,
+                key=_sort_key,
+                reverse=reverse_sort,
+            )
+
         if offset_limit_applied:
             results = pipelines
         else:

mage_ai/cache/utils.py

@@ -6,6 +6,7 @@ from mage_ai.shared.hash import extract, merge_dict
 PIPELINE_KEYS = [
     'description',
     'name',
+    'tags',
     'type',
     'updated_at',
     'uuid',
@@ -28,6 +29,7 @@ def build_pipeline_dict(
     pipeline_output_dict = dict(
         description=None,
         name=None,
+        tags=None,
         type=None,
         updated_at=None,
         uuid=None,