[jk] Bugfixes - table render / rename destination block (#4797)

* [jk] Prevent table render error

* [jk] Fix integration pipeline data exporter rename error

* [jk] Include block pipelines in integration destination block

mage_ai/data_preparation/models/block/__init__.py

@@ -2497,6 +2497,7 @@ def to_dict_base(
     def to_dict(
         self,
         include_block_catalog: bool = False,
+        include_block_pipelines: bool = False,
         include_callback_blocks: bool = False,
         include_content: bool = False,
         include_outputs: bool = False,
@@ -2515,6 +2516,9 @@ def to_dict(
         if include_block_catalog and self.is_data_integration() and self.pipeline:
             data['catalog'] = self.get_catalog_from_file()
 
+        if include_block_pipelines:
+            data['pipelines'] = self.get_pipelines_from_cache()
+
         if include_outputs:
             include_outputs_use = include_outputs
             if self.is_using_spark() and self.compute_management_enabled():

mage_ai/data_preparation/models/block/integration/__init__.py

@@ -416,9 +416,10 @@ class SourceBlock(IntegrationBlock):
 class DestinationBlock(IntegrationBlock):
     def to_dict(
         self,
-        include_content=False,
-        include_outputs=False,
-        sample_count=None,
+        include_content: bool = False,
+        include_outputs: bool = False,
+        include_block_pipelines: bool = False,
+        sample_count: int = None,
         check_if_file_exists: bool = False,
         destination_table: str = None,
         state_stream: str = None,
@@ -444,6 +445,7 @@ def to_dict(
             super().to_dict(
                 include_content=include_content,
                 include_outputs=include_outputs,
+                include_block_pipelines=include_block_pipelines,
                 sample_count=sample_count,
                 check_if_file_exists=check_if_file_exists,
             ),
@@ -452,9 +454,10 @@ def to_dict(
 
     async def to_dict_async(
         self,
-        include_content=False,
-        include_outputs=False,
-        sample_count=None,
+        include_content: bool = False,
+        include_outputs: bool = False,
+        include_block_pipelines: bool = False,
+        sample_count: int = None,
         check_if_file_exists: bool = False,
         destination_table: str = None,
         state_stream: str = None,
@@ -463,13 +466,14 @@ async def to_dict_async(
         return self.to_dict(
             include_content=include_content,
             include_outputs=include_outputs,
+            include_block_pipelines=include_block_pipelines,
             sample_count=sample_count,
             check_if_file_exists=check_if_file_exists,
             destination_table=destination_table,
             state_stream=state_stream,
         )
 
-    def update(self, data, update_state=False):
+    def update(self, data, update_state=False, **kwargs):
         if update_state:
             from mage_ai.data_preparation.models.pipelines.integration_pipeline import (
                 IntegrationPipeline,
@@ -493,7 +497,7 @@ def update(self, data, update_state=False):
                     bookmark_values=bookmark_values
                 )
 
-        return super().update(data)
+        return super().update(data, **kwargs)
 
     def output_variables(self, execution_partition: str = None) -> List[str]:
         return []

mage_ai/frontend/components/CodeBlockV2/Output/index.tsx

@@ -272,7 +272,7 @@ function CodeBlockOutput({
             width,
           }) => (
             <DataTable
-              columns={data?.columns}
+              columns={data?.columns || []}
               disableScrolling={!selected}
               index={data?.index}
               key={`data-table-${data?.index}`}
@@ -284,7 +284,7 @@ function CodeBlockOutput({
               noBorderLeft
               noBorderRight
               noBorderTop={!data?.borderTop}
-              rows={data?.rows}
+              rows={data?.rows || []}
               width={columnWidth}
             />
           ),