mage-ai 0.9.69__py3-none-any.whl → 0.9.71__py3-none-any.whl

mage_ai/api/presenters/PipelinePresenter.py

@@ -2,6 +2,7 @@ from mage_ai.api.operations import constants
 from mage_ai.api.presenters.BasePresenter import BasePresenter
 from mage_ai.data_preparation.models.constants import (
     DATAFRAME_SAMPLE_COUNT_PREVIEW,
+    MAX_RESULTS_FOR_BLOCK_OUTPUTS_PREVIEW,
     PipelineType,
 )
 from mage_ai.data_preparation.models.project import Project
@@ -68,10 +69,18 @@ class PipelinePresenter(BasePresenter):
             if include_block_metadata:
                 include_block_metadata = include_block_metadata[0]
 
-            include_block_catalog = PipelineType.PYTHON == self.model.type and \
-                Project(self.model.repo_config).is_feature_enabled(
-                    FeatureUUID.DATA_INTEGRATION_IN_BATCH_PIPELINE,
-                )
+            max_results_for_block_outputs = query.get(
+                'max_results_for_block_outputs', [MAX_RESULTS_FOR_BLOCK_OUTPUTS_PREVIEW]
+            )
+            if max_results_for_block_outputs:
+                max_results_for_block_outputs = max_results_for_block_outputs[0]
+
+            include_block_catalog = PipelineType.PYTHON == self.model.type and Project(
+                context_data=kwargs.get('context_data'),
+                repo_config=self.model.repo_config,
+            ).is_feature_enabled(
+                FeatureUUID.DATA_INTEGRATION_IN_BATCH_PIPELINE,
+            )
 
             data = await self.model.to_dict_async(
                 include_block_catalog=include_block_catalog,
@@ -85,6 +94,9 @@ class PipelinePresenter(BasePresenter):
                 include_outputs=include_outputs,
                 include_outputs_spark=include_outputs_spark,
                 sample_count=DATAFRAME_SAMPLE_COUNT_PREVIEW,
+                max_results=max_results_for_block_outputs,
+                disable_block_output_previews=True,
+                exclude_blank_variable_uuids=True,
             )
         elif constants.UPDATE == display_format:
             data = self.model.to_dict(include_extensions=include_extensions)
@@ -103,7 +115,8 @@ class PipelinePresenter(BasePresenter):
 
 PipelinePresenter.register_format(
     constants.LIST,
-    PipelinePresenter.default_attributes + [
+    PipelinePresenter.default_attributes
+    + [
         'schedules',
     ],
 )

mage_ai/api/presenters/StatusPresenter.py

@@ -9,6 +9,8 @@ class StatusPresenter(BasePresenter):
         'repo_path_relative',
         'repo_path_relative_root',
         'repo_path_root',
+        'server_process_id',
+        'scheduler_process_id',
         'scheduler_status',
         'instance_type',
         'disable_pipeline_edit_access',

mage_ai/api/presenters/SyncPresenter.py

@@ -1,16 +1,21 @@
 from mage_ai.api.presenters.BasePresenter import BasePresenter
+from mage_ai.data_preparation.shared.secrets import get_secret_value
+from mage_ai.settings.repo import get_repo_path
 
 
 class SyncPresenter(BasePresenter):
     default_attributes = [
         'access_token',
+        'access_token_secret_name',
         'auth_type',
         'branch',
         'email',
         'remote_repo_link',
         'repo_path',
         'ssh_private_key',
+        'ssh_private_key_secret_name',
         'ssh_public_key',
+        'ssh_public_key_secret_name',
         'sync_on_pipeline_run',
         'sync_on_start',
         'sync_submodules',
@@ -18,3 +23,23 @@ class SyncPresenter(BasePresenter):
         'user_git_settings',
         'username',
     ]
+
+    def present(self, **kwargs):
+        data = self.model
+
+        def filter_invalid_secret_values(data):
+            for attribute in self.default_attributes:
+                if attribute.endswith('_secret_name'):
+                    secret_name = data.get(attribute)
+                    if secret_name and not get_secret_value(
+                        secret_name,
+                        repo_name=get_repo_path(user=self.current_user),
+                        suppress_warning=False,
+                    ):
+                        data[attribute] = None
+
+        filter_invalid_secret_values(self.model)
+        if 'user_git_settings' in data:
+            filter_invalid_secret_values(data['user_git_settings'])
+
+        return self.model

mage_ai/api/resources/AutocompleteItemResource.py

@@ -13,7 +13,7 @@ class AutocompleteItemResource(GenericResource):
     @classmethod
     @safe_db_query
     async def collection(self, query, meta, user, **kwargs):
-        repo_path = get_repo_path()
+        repo_path = get_repo_path(user=user)
 
         collection = []
         for file_group, mapping in [

mage_ai/api/resources/BlockLayoutItemResource.py

@@ -21,6 +21,9 @@ from mage_ai.presenters.charts.data_sources.pipeline_schedules import (
     ChartDataSourcePipelineSchedules,
 )
 from mage_ai.presenters.charts.data_sources.pipelines import ChartDataSourcePipelines
+from mage_ai.presenters.charts.data_sources.system_metrics import (
+    ChartDataSourceSystemMetrics,
+)
 from mage_ai.shared.hash import extract, merge_dict
 from mage_ai.usage_statistics.logger import UsageStatisticLogger
 
@@ -38,6 +41,13 @@ class BlockLayoutItemResource(GenericResource):
             else:
                 variables[k] = v
 
+        content_override = query.get('content_override', [None])
+        if content_override:
+            content_override = content_override[0]
+        skip_render = query.get('skip_render', [False])
+        if skip_render:
+            skip_render = skip_render[0]
+
         configuration_override = variables.pop('configuration_override', None)
         if configuration_override:
             configuration_override = json.loads(urllib.parse.unquote(configuration_override))
@@ -57,6 +67,7 @@ class BlockLayoutItemResource(GenericResource):
 
         content = None
         data = None
+        error = None
         block_type = block_config.get('type')
 
         if BlockType.CHART == block_type:
@@ -67,39 +78,58 @@ class BlockLayoutItemResource(GenericResource):
                 else:
                     data_source_config = data_source_override
 
-            configuration_to_use = configuration_override or block_config.get('configuration')
-
-            if data_source_config or ChartType.CUSTOM == configuration_to_use.get('chart_type'):
-                data_source_type = data_source_config.get('type')
-                pipeline_uuid = data_source_config.get('pipeline_uuid')
-
-                block = BlockFactory.get_block(
-                    block_config.get('name') or file_path or uuid,
-                    block_uuid,
-                    block_type,
-                    configuration=configuration_to_use,
-                    **extract(block_config, [
+            configuration_to_use = (
+                configuration_override or block_config.get('configuration') or {}
+            )
+
+            block = BlockFactory.get_block(
+                block_config.get('name') or file_path or uuid,
+                block_uuid,
+                block_type,
+                configuration=configuration_to_use,
+                **extract(
+                    block_config,
+                    [
                         'language',
-                    ]),
-                )
+                    ],
+                ),
+            )
 
+            content = content_override
+            if not content and block.file and block.file.exists():
                 content = block.content
 
+            if data_source_config or (
+                configuration_to_use and ChartType.CUSTOM == configuration_to_use.get('chart_type')
+            ):
+                data_source_type = data_source_config.get('type')
+                pipeline_uuid = data_source_config.get('pipeline_uuid')
+
                 if ChartDataSourceType.CHART_CODE == data_source_type:
                     data_source = ChartDataSourceChartCode(
                         block_uuid=block.uuid,
                         pipeline_uuid=pipeline_uuid,
                     )
-                    data = data_source.load_data(block=block)
+                    data = data_source.load_data(
+                        block=block,
+                        configuration=configuration_to_use,
+                        custom_code=content_override,
+                    )
                 else:
                     data_source_block_uuid = data_source_config.get('block_uuid')
 
-                    data_source_class_options = merge_dict(extract(data_source_config, [
-                        'pipeline_schedule_id',
-                    ]), dict(
-                        block_uuid=data_source_block_uuid,
-                        pipeline_uuid=pipeline_uuid,
-                    ))
+                    data_source_class_options = merge_dict(
+                        extract(
+                            data_source_config,
+                            [
+                                'pipeline_schedule_id',
+                            ],
+                        ),
+                        dict(
+                            block_uuid=data_source_block_uuid,
+                            pipeline_uuid=pipeline_uuid,
+                        ),
+                    )
                     data_source_output = None
 
                     if ChartDataSourceType.BLOCK == data_source_type:
@@ -132,25 +162,32 @@ class BlockLayoutItemResource(GenericResource):
                             user=user,
                             **kwargs,
                         )
+                    elif ChartDataSourceType.SYSTEM_METRICS == data_source_type:
+                        data_source = ChartDataSourceSystemMetrics(**data_source_class_options)
+                        data_source_output = data_source.load_data(
+                            user=user,
+                            **kwargs,
+                        )
 
-                    try:
-                        input_args = None
-                        if data_source_output is not None:
-                            input_args = [data_source_output]
-
-                        data = block.execute_with_callback(
-                            disable_json_serialization=True,
-                            input_args=input_args,
-                            global_vars=merge_dict(
-                                get_global_variables(pipeline_uuid) if pipeline_uuid else {},
-                                variables or {},
-                            ),
-                        ).get('output', None)
-                    except Exception as err:
-                        error = ApiError(ApiError.RESOURCE_NOT_FOUND.copy())
-                        error.message = str(err)
-                        error.errors = traceback.format_exc()
-                        raise error
+                    if not skip_render:
+                        try:
+                            input_args = None
+                            if data_source_output is not None:
+                                input_args = [data_source_output]
+
+                            data = block.execute_with_callback(
+                                custom_code=content_override,
+                                disable_json_serialization=True,
+                                input_args=input_args,
+                                global_vars=merge_dict(
+                                    get_global_variables(pipeline_uuid) if pipeline_uuid else {},
+                                    variables or {},
+                                ),
+                            ).get('output', None)
+                        except Exception as err:
+                            error = ApiError(ApiError.RESOURCE_NOT_FOUND.copy())
+                            error.message = str(err)
+                            error.errors = traceback.format_exc()
 
         block_config_to_show = {}
         block_config_to_show.update(block_config)
@@ -162,8 +199,17 @@ class BlockLayoutItemResource(GenericResource):
 
         await UsageStatisticLogger().chart_impression(block_config_to_show)
 
-        return self(merge_dict(block_config_to_show, dict(
-            content=content,
-            data=data,
-            uuid=uuid,
-        )), user, **kwargs)
+        return self(
+            merge_dict(
+                block_config_to_show,
+                dict(
+                    content=content,
+                    data=data,
+                    error=error,
+                    skip_render=skip_render,
+                    uuid=uuid,
+                ),
+            ),
+            user,
+            **kwargs,
+        )

mage_ai/api/resources/BlockOutputResource.py

@@ -1,8 +1,10 @@
 from mage_ai.api.errors import ApiError
 from mage_ai.api.resources.GenericResource import GenericResource
+from mage_ai.data_preparation.models.constants import DATAFRAME_SAMPLE_COUNT
 from mage_ai.data_preparation.models.pipeline import Pipeline
-from mage_ai.data_preparation.models.variable import VariableType
+from mage_ai.data_preparation.models.variables.constants import VariableType
 from mage_ai.orchestration.db import safe_db_query
+from mage_ai.settings.repo import get_repo_path
 
 
 class BlockOutputResource(GenericResource):
@@ -10,29 +12,60 @@ class BlockOutputResource(GenericResource):
     Resource to fetch block output for the notebook. Created to support legacy
     endpoint /api/pipelines/<pipeline_uuid>/blocks/<block_uuid>/outputs
     """
+
     @classmethod
     @safe_db_query
-    def member(self, pk, user, **kwargs):
+    def member(cls, pk, user, **kwargs):
+        query = kwargs.get('query', {})
         block_uuid = pk
 
-        query = kwargs.get('query', {})
         pipeline_uuid = query.get('pipeline_uuid', [None])
         if pipeline_uuid:
             pipeline_uuid = pipeline_uuid[0]
+
+        outputs_query = {}
+        for key in [
+            'block_uuid',
+            'csv_lines_only',
+            'dynamic_block_index',
+            'exclude_blank_variable_uuids',
+            'execution_partition',
+            'include_print_outputs',
+            'sample',
+            'sample_count',
+            'variable_type',
+        ]:
+            value = query.get(key, [None])
+            if value is not None:
+                value = value[0]
+                if value is not None:
+                    outputs_query[key] = value
+
+        for key, value in [
+            ('exclude_blank_variable_uuids', True),
+            ('include_print_outputs', False),
+            ('sample_count', DATAFRAME_SAMPLE_COUNT),
+            ('variable_type', VariableType.DATAFRAME),
+        ]:
+            if key not in outputs_query:
+                outputs_query[key] = value
+
         outputs = []
         if pipeline_uuid is not None:
-            pipeline = Pipeline.get(pipeline_uuid)
+            repo_path = get_repo_path(user=user)
+            pipeline = Pipeline.get(pipeline_uuid, repo_path=repo_path)
             block = pipeline.get_block(block_uuid)
+
             error = ApiError.RESOURCE_ERROR.copy()
             if block is None:
                 error.update(
-                    message=f'Block {block_uuid} does not exist in pipeline {pipeline_uuid}')
+                    message=f'Block {block_uuid} does not exist in pipeline {pipeline_uuid}'
+                )
                 raise ApiError(error)
+
             # Only fetch dataframe variables by default
             outputs = block.get_outputs(
-                include_print_outputs=False,
-                sample_count=None,
-                variable_type=VariableType.DATAFRAME,
+                **outputs_query,
             )
 
-        return self(dict(outputs=outputs), user, **kwargs)
+        return cls(dict(outputs=outputs), user, **kwargs)

mage_ai/api/resources/BlockResource.py

@@ -110,6 +110,7 @@ class BlockResource(GenericResource):
         name = payload.get('name')
         block_name = name or payload.get('uuid')
         require_unique_name = payload.get('require_unique_name')
+        repo_path = get_repo_path(user=user)
 
         payload_config = payload.get('config') or {}
         replicated_block = None
@@ -208,7 +209,7 @@ class BlockResource(GenericResource):
 
             if payload_config and payload_config.get('custom_template_uuid'):
                 template_uuid = payload_config.get('custom_template_uuid')
-                custom_template = CustomBlockTemplate.load(template_uuid=template_uuid)
+                custom_template = CustomBlockTemplate.load(repo_path, template_uuid=template_uuid)
                 block = custom_template.create_block(
                     block_name,
                     pipeline,
@@ -221,7 +222,7 @@ class BlockResource(GenericResource):
                 block = Block.create(
                     block_name,
                     block_type,
-                    get_repo_path(),
+                    repo_path,
                     require_unique_name=require_unique_name,
                     **block_attributes,
                 )
@@ -234,7 +235,7 @@ class BlockResource(GenericResource):
 
         if pipeline:
             cache = await BlockCache.initialize_cache()
-            cache.add_pipeline(block.to_dict(), pipeline)
+            cache.add_pipeline(block.to_dict(), pipeline, repo_path)
 
         cache_block_action_object = await BlockActionObjectCache.initialize_cache()
         cache_block_action_object.update_block(block)

mage_ai/api/resources/BlockRunResource.py

@@ -27,12 +27,22 @@ class BlockRunResource(DatabaseResource):
                 started_at = tup.started_at if hasattr(tup, 'started_at') else None
                 status = tup.status if hasattr(tup, 'status') else None
                 updated_at = tup.updated_at if hasattr(tup, 'updated_at') else None
-                pipeline_schedule_id = tup.pipeline_schedule_id if hasattr(
-                    tup, 'pipeline_schedule_id',
-                ) else None
-                pipeline_schedule_name = tup.pipeline_schedule_name if hasattr(
-                    tup, 'pipeline_schedule_name',
-                ) else None
+                pipeline_schedule_id = (
+                    tup.pipeline_schedule_id
+                    if hasattr(
+                        tup,
+                        'pipeline_schedule_id',
+                    )
+                    else None
+                )
+                pipeline_schedule_name = (
+                    tup.pipeline_schedule_name
+                    if hasattr(
+                        tup,
+                        'pipeline_schedule_name',
+                    )
+                    else None
+                )
             else:
                 (
                     block_uuid,
@@ -98,29 +108,22 @@ class BlockRunResource(DatabaseResource):
         ]
 
         query = (
-            BlockRun.
-            select(*columns).
-            join(b, a.pipeline_run_id == b.id).
-            join(c, b.pipeline_schedule_id == c.id)
+            BlockRun.select(*columns)
+            .join(b, a.pipeline_run_id == b.id)
+            .join(c, b.pipeline_schedule_id == c.id)
         )
 
         pipeline_run_id = query_arg.get('pipeline_run_id', [None])
         if pipeline_run_id:
             pipeline_run_id = pipeline_run_id[0]
         if pipeline_run_id and is_number(pipeline_run_id):
-            query = (
-                query.
-                filter(a.pipeline_run_id == int(pipeline_run_id))
-            )
+            query = query.filter(a.pipeline_run_id == int(pipeline_run_id))
 
         pipeline_uuid = query_arg.get('pipeline_uuid', [None])
         if pipeline_uuid:
             pipeline_uuid = pipeline_uuid[0]
         if pipeline_uuid:
-            query = (
-                query.
-                filter(c.pipeline_uuid == pipeline_uuid)
-            )
+            query = query.filter(c.pipeline_uuid == pipeline_uuid)
 
         # The order_by value should be an attribute on the BlockRun model.
         order_by_arg = query_arg.get('order_by', [None])
@@ -138,10 +141,12 @@ class BlockRunResource(DatabaseResource):
             try:
                 br_col = getattr(a, col)
                 initial_results = query.order_by(getattr(br_col, asc_desc)())
-            except (AttributeError):
-                raise Exception('Block run sort column/query is invalid. The sort column ' +
-                                'must be an attribute of the BlockRun model. The sort direction ' +
-                                'is either "asc" (ascending order) or "desc" (descending order).')
+            except AttributeError:
+                raise Exception(
+                    'Block run sort column/query is invalid. The sort column '
+                    + 'must be an attribute of the BlockRun model. The sort direction '
+                    + 'is either "asc" (ascending order) or "desc" (descending order).'
+                )
         else:
             initial_results = query.order_by(
                 a.started_at.desc(),

mage_ai/api/resources/ClusterResource.py

@@ -4,6 +4,7 @@ from mage_ai.cluster_manager.constants import ClusterType
 from mage_ai.orchestration.db import safe_db_query
 from mage_ai.server.active_kernel import get_active_kernel_name
 from mage_ai.server.kernels import KernelName
+from mage_ai.settings.repo import get_repo_path
 
 
 class ClusterResource(GenericResource):
@@ -41,7 +42,9 @@ class ClusterResource(GenericResource):
                 error['message'] = 'Please include cluster_id in the request payload.'
                 raise ApiError(error)
 
-            emr_cluster_manager.set_active_cluster(cluster_id=cluster_id)
+            repo_path = get_repo_path(user=self.current_user)
+
+            emr_cluster_manager.set_active_cluster(repo_path=repo_path, cluster_id=cluster_id)
             success = True
 
             self.model.update(

mage_ai/api/resources/CustomTemplateResource.py

@@ -20,6 +20,7 @@ from mage_ai.data_preparation.models.custom_templates.utils import (
 )
 from mage_ai.data_preparation.models.pipeline import Pipeline
 from mage_ai.data_preparation.templates.template import fetch_template_source
+from mage_ai.settings.repo import get_repo_path
 from mage_ai.shared.hash import ignore_keys
 from mage_ai.shared.utils import clean_name
 from mage_ai.usage_statistics.logger import UsageStatisticLogger
@@ -64,14 +65,20 @@ class CustomTemplateResource(GenericResource):
             template_uuid = clean_name(template_uuid)
             payload['template_uuid'] = template_uuid
 
+        repo_path = get_repo_path(user=user)
         if DIRECTORY_FOR_BLOCK_TEMPLATES == object_type:
-            custom_template = CustomBlockTemplate.load(template_uuid=template_uuid)
-
+            custom_template = CustomBlockTemplate.load(repo_path, template_uuid=template_uuid)
             if not custom_template:
-                custom_template = CustomBlockTemplate(**ignore_keys(payload, [
-                    'uuid',
-                    OBJECT_TYPE_KEY,
-                ]))
+                custom_template = CustomBlockTemplate(
+                    repo_path=repo_path,
+                    **ignore_keys(
+                        payload,
+                        [
+                            'uuid',
+                            OBJECT_TYPE_KEY,
+                        ],
+                    )
+                )
                 if user:
                     custom_template.user = dict(
                         username=user.username,
@@ -88,10 +95,12 @@ class CustomTemplateResource(GenericResource):
                 cache = await BlockActionObjectCache.initialize_cache()
                 cache.update_custom_block_template(custom_template)
         elif DIRECTORY_FOR_PIPELINE_TEMPLATES == object_type:
-            custom_template = CustomPipelineTemplate.load(template_uuid=template_uuid)
+            custom_template = CustomPipelineTemplate.load(
+                template_uuid=template_uuid, repo_path=repo_path
+            )
 
             if not custom_template:
-                pipeline = Pipeline.get(payload.get('pipeline_uuid'))
+                pipeline = Pipeline.get(payload.get('pipeline_uuid'), repo_path=repo_path)
                 custom_template = CustomPipelineTemplate.create_from_pipeline(
                     pipeline,
                     template_uuid,
@@ -120,12 +129,20 @@ class CustomTemplateResource(GenericResource):
 
         template_uuid = urllib.parse.unquote(pk)
 
+        repo_path = get_repo_path(user=user)
         try:
             if DIRECTORY_FOR_BLOCK_TEMPLATES == object_type:
-                return self(CustomBlockTemplate.load(template_uuid=template_uuid), user, **kwargs)
+                return self(
+                    CustomBlockTemplate.load(repo_path, template_uuid=template_uuid),
+                    user,
+                    **kwargs,
+                )
             elif DIRECTORY_FOR_PIPELINE_TEMPLATES == object_type:
                 return self(
-                    CustomPipelineTemplate.load(template_uuid=template_uuid),
+                    CustomPipelineTemplate.load(
+                        repo_path,
+                        template_uuid=template_uuid,
+                    ),
                     user,
                     **kwargs,
                 )
@@ -152,10 +169,13 @@ class CustomTemplateResource(GenericResource):
             cache = await BlockActionObjectCache.initialize_cache()
             cache.update_custom_block_template(self.model, remove=True)
 
-        for key, value in ignore_keys(payload, [
-            'uuid',
-            OBJECT_TYPE_KEY,
-        ]).items():
+        for key, value in ignore_keys(
+            payload,
+            [
+                'uuid',
+                OBJECT_TYPE_KEY,
+            ],
+        ).items():
             setattr(self.model, key, value)
         self.model.save()
 

mage_ai/api/resources/DataProviderResource.py

@@ -43,7 +43,7 @@ class DataProviderResource(GenericResource):
     @classmethod
     @safe_db_query
     async def collection(self, query, meta, user, **kwargs):
-        file_path = f'{get_repo_path()}/io_config.yaml'
+        file_path = f'{get_repo_path(user=user)}/io_config.yaml'
         profiles = []
         if os.path.exists(file_path):
             async with aiofiles.open(file_path, mode='r') as file:

mage_ai/api/resources/ExecutionStateResource.py

@@ -1,6 +1,7 @@
 from mage_ai.api.resources.GenericResource import GenericResource
 from mage_ai.data_preparation.models.constants import PipelineType
 from mage_ai.data_preparation.models.pipeline import Pipeline
+from mage_ai.settings.repo import get_repo_path
 
 
 class ExecutionStateResource(GenericResource):
@@ -17,7 +18,8 @@ class ExecutionStateResource(GenericResource):
             block_uuid = block_uuid[0]
 
         if pipeline_uuid and block_uuid:
-            pipeline = await Pipeline.get_async(pipeline_uuid)
+            repo_path = get_repo_path(user=user)
+            pipeline = await Pipeline.get_async(pipeline_uuid, repo_path=repo_path)
             if pipeline and pipeline.type in [PipelineType.PYTHON, PipelineType.PYSPARK]:
                 block = pipeline.get_block(block_uuid)
                 if block and block.compute_management_enabled() and block.is_using_spark():

mage_ai/api/resources/FileContentResource.py

@@ -22,7 +22,10 @@ class FileContentResource(GenericResource):
         if 'payload' in kwargs and 'file_content' in kwargs['payload']:
             payload = kwargs['payload']['file_content']
             if 'block_uuid' in payload and 'pipeline_uuid' in payload:
-                pipeline = await Pipeline.get_async(payload.get('pipeline_uuid'))
+                repo_path = get_repo_path(user=user)
+                pipeline = await Pipeline.get_async(
+                    payload.get('pipeline_uuid'), repo_path=repo_path
+                )
                 if pipeline:
                     block = pipeline.get_block(payload.get('block_uuid'))
                     if block:
@@ -57,7 +60,10 @@ class FileContentResource(GenericResource):
 
         await self.model.update_content_async(content)
 
-        block_type = Block.block_type_from_path(self.model.dir_path)
+        block_type = Block.block_type_from_path(
+            self.model.dir_path,
+            repo_path=get_repo_path(user=self.current_user),
+        )
         if block_type:
             cache_block_action_object = await BlockActionObjectCache.initialize_cache()
             cache_block_action_object.update_block(block_file_absolute_path=self.model.file_path)