openmetadata-ingestion 1.3.3.0rc2__py3-none-any.whl → 1.4.0.0rc1__py3-none-any.whl

metadata/ingestion/source/dashboard/dashboard_service.py

@@ -40,7 +40,11 @@ from metadata.generated.schema.metadataIngestion.dashboardServiceMetadataPipelin
 from metadata.generated.schema.metadataIngestion.workflow import (
     Source as WorkflowSource,
 )
-from metadata.generated.schema.type.entityLineage import EntitiesEdge, LineageDetails
+from metadata.generated.schema.type.entityLineage import (
+    ColumnLineage,
+    EntitiesEdge,
+    LineageDetails,
+)
 from metadata.generated.schema.type.entityLineage import Source as LineageSource
 from metadata.generated.schema.type.entityReference import EntityReference
 from metadata.generated.schema.type.usageRequest import UsageRequest
@@ -48,13 +52,14 @@ from metadata.ingestion.api.delete import delete_entity_from_source
 from metadata.ingestion.api.models import Either, Entity
 from metadata.ingestion.api.steps import Source
 from metadata.ingestion.api.topology_runner import C, TopologyRunnerMixin
+from metadata.ingestion.lineage.sql_lineage import get_column_fqn
 from metadata.ingestion.models.delete_entity import DeleteEntity
 from metadata.ingestion.models.ometa_classification import OMetaTagAndClassification
 from metadata.ingestion.models.patch_request import PatchRequest
 from metadata.ingestion.models.topology import (
     NodeStage,
     ServiceTopology,
-    TopologyContext,
+    TopologyContextManager,
     TopologyNode,
 )
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
@@ -194,7 +199,7 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
     service_connection: DashboardConnection.__fields__["config"].type_
 
     topology = DashboardServiceTopology()
-    context = TopologyContext.create(topology)
+    context = TopologyContextManager(topology)
     dashboard_source_state: Set = set()
     datamodel_source_state: Set = set()
 
@@ -288,13 +293,13 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
         Returns:
             Lineage request between Data Models and Dashboards
         """
-        if hasattr(self.context, "dataModels") and self.context.dataModels:
-            for datamodel in self.context.dataModels:
+        if hasattr(self.context.get(), "dataModels") and self.context.get().dataModels:
+            for datamodel in self.context.get().dataModels:
                 try:
                     datamodel_fqn = fqn.build(
                         metadata=self.metadata,
                         entity_type=DashboardDataModel,
-                        service_name=self.context.dashboard_service,
+                        service_name=self.context.get().dashboard_service,
                         data_model_name=datamodel,
                     )
                     datamodel_entity = self.metadata.get_by_name(
@@ -304,8 +309,8 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                     dashboard_fqn = fqn.build(
                         self.metadata,
                         entity_type=Dashboard,
-                        service_name=self.context.dashboard_service,
-                        dashboard_name=self.context.dashboard,
+                        service_name=self.context.get().dashboard_service,
+                        dashboard_name=self.context.get().dashboard,
                     )
                     dashboard_entity = self.metadata.get_by_name(
                         entity=Dashboard, fqn=dashboard_fqn
@@ -319,6 +324,16 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                         f"Error to yield dashboard lineage details for data model name [{datamodel.name}]: {err}"
                     )
 
+    def get_db_service_names(self) -> List[str]:
+        """
+        Get the list of db service names
+        """
+        return (
+            self.source_config.lineageInformation.dbServiceNames or []
+            if self.source_config.lineageInformation
+            else []
+        )
+
     def yield_dashboard_lineage(
         self, dashboard_details: Any
     ) -> Iterable[Either[AddLineageRequest]]:
@@ -330,7 +345,9 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
         """
         yield from self.yield_datamodel_dashboard_lineage() or []
 
-        for db_service_name in self.source_config.dbServiceNames or []:
+        db_service_names = self.get_db_service_names()
+
+        for db_service_name in db_service_names or []:
             yield from self.yield_dashboard_lineage_details(
                 dashboard_details, db_service_name
             ) or []
@@ -380,7 +397,7 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                 entity_type=Dashboard,
                 entity_source_state=self.dashboard_source_state,
                 mark_deleted_entity=self.source_config.markDeletedDashboards,
-                params={"service": self.context.dashboard_service},
+                params={"service": self.context.get().dashboard_service},
             )
 
     def mark_datamodels_as_deleted(self) -> Iterable[Either[DeleteEntity]]:
@@ -394,7 +411,7 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                 entity_type=DashboardDataModel,
                 entity_source_state=self.datamodel_source_state,
                 mark_deleted_entity=self.source_config.markDeletedDataModels,
-                params={"service": self.context.dashboard_service},
+                params={"service": self.context.get().dashboard_service},
             )
 
     def get_owner_ref(  # pylint: disable=unused-argument, useless-return
@@ -440,6 +457,7 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
     def _get_add_lineage_request(
         to_entity: Union[Dashboard, DashboardDataModel],
         from_entity: Union[Table, DashboardDataModel, Dashboard],
+        column_lineage: List[ColumnLineage] = None,
     ) -> Optional[Either[AddLineageRequest]]:
         if from_entity and to_entity:
             return Either(
@@ -454,7 +472,8 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                             type=LINEAGE_MAP[type(to_entity)],
                         ),
                         lineageDetails=LineageDetails(
-                            source=LineageSource.DashboardLineage
+                            source=LineageSource.DashboardLineage,
+                            columnsLineage=column_lineage,
                         ),
                     )
                 )
@@ -462,6 +481,20 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
 
         return None
 
+    @staticmethod
+    def _get_data_model_column_fqn(
+        data_model_entity: DashboardDataModel, column: str
+    ) -> Optional[str]:
+        """
+        Get fqn of column if exist in table entity
+        """
+        if not data_model_entity:
+            return None
+        for tbl_column in data_model_entity.columns:
+            if tbl_column.displayName.lower() == column.lower():
+                return tbl_column.fullyQualifiedName.__root__
+        return None
+
     def get_dashboard(self) -> Any:
         """
         Method to iterate through dashboard lists filter dashboards & yield dashboard details
@@ -480,15 +513,15 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
 
             try:
                 dashboard_details = self.get_dashboard_details(dashboard)
-                self.context.project_name = (  # pylint: disable=assignment-from-none
+                self.context.get().project_name = (  # pylint: disable=E1128
                     self.get_project_name(dashboard_details=dashboard_details)
                 )
                 if filter_by_project(
                     self.source_config.projectFilterPattern,
-                    self.context.project_name,
+                    self.context.get().project_name,
                 ):
                     self.status.filter(
-                        self.context.project_name,
+                        self.context.get().project_name,
                         "Project / Workspace Filtered Out",
                     )
                     continue
@@ -519,7 +552,7 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
         :return: Entity FQN derived from context
         """
         context_names = [
-            self.context.__dict__[dependency]
+            self.context.get().__dict__[dependency]
             for dependency in stage.consumer or []  # root nodes do not have consumers
         ]
 
@@ -597,3 +630,29 @@ class DashboardServiceSource(TopologyRunnerMixin, Source, ABC):
                     )
             patch_request.new_entity.dataModels = datamodel_entity_ref_list
         return patch_request
+
+    def _get_column_lineage(
+        self,
+        om_table: Table,
+        data_model_entity: DashboardDataModel,
+        columns_list: List[str],
+    ) -> List[ColumnLineage]:
+        """
+        Get the column lineage from the fields
+        """
+        try:
+            column_lineage = []
+            for field in columns_list or []:
+                from_column = get_column_fqn(table_entity=om_table, column=field)
+                to_column = self._get_data_model_column_fqn(
+                    data_model_entity=data_model_entity,
+                    column=field,
+                )
+                if from_column and to_column:
+                    column_lineage.append(
+                        ColumnLineage(fromColumns=[from_column], toColumn=to_column)
+                    )
+            return column_lineage
+        except Exception as exc:
+            logger.debug(f"Error to get column lineage: {exc}")
+            logger.debug(traceback.format_exc())

metadata/ingestion/source/dashboard/domodashboard/metadata.py

@@ -61,7 +61,9 @@ class DomodashboardSource(DashboardServiceSource):
     metadata_config: OpenMetadataConnection
 
     @classmethod
-    def create(cls, config_dict, metadata: OpenMetadata):
+    def create(
+        cls, config_dict, metadata: OpenMetadata, pipeline_name: Optional[str] = None
+    ):
         config = WorkflowSource.parse_obj(config_dict)
         connection: DomoDashboardConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, DomoDashboardConnection):
@@ -124,12 +126,12 @@ class DomodashboardSource(DashboardServiceSource):
                     fqn.build(
                         self.metadata,
                         entity_type=Chart,
-                        service_name=self.context.dashboard_service,
+                        service_name=self.context.get().dashboard_service,
                         chart_name=chart,
                     )
-                    for chart in self.context.charts or []
+                    for chart in self.context.get().charts or []
                 ],
-                service=self.context.dashboard_service,
+                service=self.context.get().dashboard_service,
                 owner=self.get_owner_ref(dashboard_details=dashboard_details),
             )
             yield Either(right=dashboard_request)
@@ -219,7 +221,7 @@ class DomodashboardSource(DashboardServiceSource):
                             description=chart.description,
                             displayName=chart.name,
                             sourceUrl=chart_url,
-                            service=self.context.dashboard_service,
+                            service=self.context.get().dashboard_service,
                             chartType=get_standard_chart_type(chart.metadata.chartType),
                         )
                     )

metadata/ingestion/source/dashboard/lightdash/metadata.py

@@ -50,7 +50,9 @@ class LightdashSource(DashboardServiceSource):
     metadata_config: OpenMetadataConnection
 
     @classmethod
-    def create(cls, config_dict, metadata: OpenMetadata):
+    def create(
+        cls, config_dict, metadata: OpenMetadata, pipeline_name: Optional[str] = None
+    ):
         config = WorkflowSource.parse_obj(config_dict)
         connection: LightdashConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, LightdashConnection):
@@ -111,12 +113,12 @@ class LightdashSource(DashboardServiceSource):
                     fqn.build(
                         self.metadata,
                         entity_type=Chart,
-                        service_name=self.context.dashboard_service,
+                        service_name=self.context.get().dashboard_service,
                         chart_name=chart,
                     )
-                    for chart in self.context.charts or []
+                    for chart in self.context.get().charts or []
                 ],
-                service=self.context.dashboard_service,
+                service=self.context.get().dashboard_service,
                 owner=self.get_owner_ref(dashboard_details=dashboard_details),
             )
             yield dashboard_request
@@ -152,7 +154,7 @@ class LightdashSource(DashboardServiceSource):
                     displayName=chart.name,
                     description=chart.description,
                     sourceUrl=chart_url,
-                    service=self.context.dashboard_service,
+                    service=self.context.get().dashboard_service,
                 )
                 self.status.scanned(chart.name)
             except Exception as exc:  # pylint: disable=broad-except

metadata/ingestion/source/dashboard/looker/metadata.py

@@ -174,7 +174,12 @@ class LookerSource(DashboardServiceSource):
         self._added_lineage: Optional[Dict] = {}
 
     @classmethod
-    def create(cls, config_dict: dict, metadata: OpenMetadata) -> "LookerSource":
+    def create(
+        cls,
+        config_dict: dict,
+        metadata: OpenMetadata,
+        pipeline_name: Optional[str] = None,
+    ) -> "LookerSource":
         config = WorkflowSource.parse_obj(config_dict)
         connection: LookerConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, LookerConnection):
@@ -367,7 +372,7 @@ class LookerSource(DashboardServiceSource):
         fqn_datamodel = fqn.build(
             self.metadata,
             DashboardDataModel,
-            service_name=self.context.dashboard_service,
+            service_name=self.context.get().dashboard_service,
             data_model_name=data_model_name,
         )
 
@@ -396,7 +401,7 @@ class LookerSource(DashboardServiceSource):
                     name=datamodel_name,
                     displayName=model.name,
                     description=model.description,
-                    service=self.context.dashboard_service,
+                    service=self.context.get().dashboard_service,
                     dataModelType=DataModelType.LookMlExplore.value,
                     serviceType=DashboardServiceType.Looker.value,
                     columns=get_columns_from_model(model),
@@ -408,14 +413,14 @@ class LookerSource(DashboardServiceSource):
                 self.register_record_datamodel(datamodel_request=explore_datamodel)
 
                 # build datamodel by our hand since ack_sink=False
-                self.context.dataModel = self._build_data_model(datamodel_name)
-                self._view_data_model = copy.deepcopy(self.context.dataModel)
+                self.context.get().dataModel = self._build_data_model(datamodel_name)
+                self._view_data_model = copy.deepcopy(self.context.get().dataModel)
 
                 # Maybe use the project_name as key too?
                 # Save the explores for when we create the lineage with the dashboards and views
                 self._explores_cache[
                     explore_datamodel.name.__root__
-                ] = self.context.dataModel  # This is the newly created explore
+                ] = self.context.get().dataModel  # This is the newly created explore
 
                 # We can get VIEWs from the JOINs to know the dependencies
                 # We will only try and fetch if we have the credentials
@@ -494,7 +499,7 @@ class LookerSource(DashboardServiceSource):
                     name=build_datamodel_name(explore.model_name, view.name),
                     displayName=view.name,
                     description=view.description,
-                    service=self.context.dashboard_service,
+                    service=self.context.get().dashboard_service,
                     dataModelType=DataModelType.LookMlView.value,
                     serviceType=DashboardServiceType.Looker.value,
                     columns=get_columns_from_model(view),
@@ -541,11 +546,13 @@ class LookerSource(DashboardServiceSource):
                     " while processing view lineage."
                 )
 
+            db_service_names = self.get_db_service_names()
+
             if view.sql_table_name:
                 source_table_name = self._clean_table_name(view.sql_table_name)
 
                 # View to the source is only there if we are informing the dbServiceNames
-                for db_service_name in self.source_config.dbServiceNames or []:
+                for db_service_name in db_service_names or []:
                     yield self.build_lineage_request(
                         source=source_table_name,
                         db_service_name=db_service_name,
@@ -556,7 +563,7 @@ class LookerSource(DashboardServiceSource):
                 sql_query = view.derived_table.sql
                 if not sql_query:
                     return
-                for db_service_name in self.source_config.dbServiceNames or []:
+                for db_service_name in db_service_names or []:
                     db_service = self.metadata.get_by_name(
                         DatabaseService, db_service_name
                     )
@@ -654,16 +661,16 @@ class LookerSource(DashboardServiceSource):
                 fqn.build(
                     self.metadata,
                     entity_type=Chart,
-                    service_name=self.context.dashboard_service,
+                    service_name=self.context.get().dashboard_service,
                     chart_name=chart,
                 )
-                for chart in self.context.charts or []
+                for chart in self.context.get().charts or []
             ],
             # Dashboards are created from the UI directly. They are not linked to a project
             # like LookML assets, but rather just organised in folders.
             project=self.get_project_name(dashboard_details),
             sourceUrl=f"{clean_uri(self.service_connection.hostPort)}/dashboards/{dashboard_details.id}",
-            service=self.context.dashboard_service,
+            service=self.context.get().dashboard_service,
             owner=self.get_owner_ref(dashboard_details=dashboard_details),
         )
         yield Either(right=dashboard_request)
@@ -732,7 +739,7 @@ class LookerSource(DashboardServiceSource):
             fqn=fqn.build(
                 self.metadata,
                 entity_type=DashboardDataModel,
-                service_name=self.context.dashboard_service,
+                service_name=self.context.get().dashboard_service,
                 data_model_name=explore_name,
             ),
         )
@@ -757,8 +764,8 @@ class LookerSource(DashboardServiceSource):
                     dashboard_fqn = fqn.build(
                         self.metadata,
                         entity_type=Dashboard,
-                        service_name=self.context.dashboard_service,
-                        dashboard_name=self.context.dashboard,
+                        service_name=self.context.get().dashboard_service,
+                        dashboard_name=self.context.get().dashboard,
                     )
                     dashboard_entity = self.metadata.get_by_name(
                         entity=Dashboard, fqn=dashboard_fqn
@@ -869,7 +876,7 @@ class LookerSource(DashboardServiceSource):
                         sourceUrl=chart.query.share_url
                         if chart.query is not None
                         else f"{clean_uri(self.service_connection.hostPort)}/merge?mid={chart.merge_result_id}",
-                        service=self.context.dashboard_service,
+                        service=self.context.get().dashboard_service,
                     )
                 )
 
@@ -938,13 +945,13 @@ class LookerSource(DashboardServiceSource):
         :return: UsageRequest, if not computed
         """
 
-        dashboard_name = self.context.dashboard
+        dashboard_name = self.context.get().dashboard
 
         try:
             dashboard_fqn = fqn.build(
                 metadata=self.metadata,
                 entity_type=Dashboard,
-                service_name=self.context.dashboard_service,
+                service_name=self.context.get().dashboard_service,
                 dashboard_name=dashboard_name,
             )
 

metadata/ingestion/source/dashboard/metabase/client.py

@@ -85,18 +85,49 @@ class MetabaseClient:
         )
         self.client = REST(client_config)
 
-    def get_dashboards_list(self) -> List[MetabaseDashboard]:
+    def get_dashboards_list(
+        self, collections: List[MetabaseCollection]
+    ) -> List[MetabaseDashboard]:
         """
         Get List of all dashboards
         """
-        try:
-            resp_dashboards = self.client.get("/dashboard")
+        dashboards = []
+        for collection in collections or []:
+            try:
+                resp_dashboards = self.client.get(
+                    f"/collection/{collection.id}/items?models=dashboard"
+                )
+                if resp_dashboards:
+                    dashboard_list = MetabaseDashboardList(**resp_dashboards)
+                    dashboards.extend(dashboard_list.data)
+            except Exception:
+                logger.debug(traceback.format_exc())
+                logger.warning("Failed to fetch the dashboard list")
+        return dashboards
+
+    def get_dashboards_list_test_conn(
+        self, collections: List[MetabaseCollection]
+    ) -> List[MetabaseDashboard]:
+        """
+        Get List of all dashboards
+        """
+        for collection in collections or []:
+            resp_dashboards = self.client.get(
+                f"/collection/{collection.id}/items?models=dashboard"
+            )
             if resp_dashboards:
-                dashboard_list = MetabaseDashboardList(dashboards=resp_dashboards)
-                return dashboard_list.dashboards
-        except Exception:
-            logger.debug(traceback.format_exc())
-            logger.warning("Failed to fetch the dashboard list")
+                dashboard_list = MetabaseDashboardList(**resp_dashboards)
+                return dashboard_list.data
+        return []
+
+    def get_collections_list_test_conn(self) -> List[MetabaseCollection]:
+        """
+        Get List of all collections
+        """
+        resp_collections = self.client.get("/collection")
+        if resp_collections:
+            collection_list = MetabaseCollectionList(collections=resp_collections)
+            return collection_list.collections
         return []
 
     def get_collections_list(self) -> List[MetabaseCollection]:

metadata/ingestion/source/dashboard/metabase/connection.py

@@ -44,7 +44,8 @@ def test_connection(
     """
 
     def custom_executor():
-        return client.get_dashboards_list()
+        collections = client.get_collections_list_test_conn()
+        return client.get_dashboards_list_test_conn(collections)
 
     test_fn = {"GetDashboards": custom_executor}
 

metadata/ingestion/source/dashboard/metabase/metadata.py

@@ -67,7 +67,9 @@ class MetabaseSource(DashboardServiceSource):
     metadata_config: OpenMetadataConnection
 
     @classmethod
-    def create(cls, config_dict, metadata: OpenMetadata):
+    def create(
+        cls, config_dict, metadata: OpenMetadata, pipeline_name: Optional[str] = None
+    ):
         config = WorkflowSource.parse_obj(config_dict)
         connection: MetabaseConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, MetabaseConnection):
@@ -92,7 +94,7 @@ class MetabaseSource(DashboardServiceSource):
         """
         Get List of all dashboards
         """
-        return self.client.get_dashboards_list()
+        return self.client.get_dashboards_list(self.collections)
 
     def get_dashboard_name(self, dashboard: MetabaseDashboard) -> str:
         """
@@ -144,17 +146,17 @@ class MetabaseSource(DashboardServiceSource):
                 sourceUrl=dashboard_url,
                 displayName=dashboard_details.name,
                 description=dashboard_details.description,
-                project=self.context.project_name,
+                project=self.context.get().project_name,
                 charts=[
                     fqn.build(
                         self.metadata,
                         entity_type=Chart,
-                        service_name=self.context.dashboard_service,
+                        service_name=self.context.get().dashboard_service,
                         chart_name=chart,
                     )
-                    for chart in self.context.charts or []
+                    for chart in self.context.get().charts or []
                 ],
-                service=self.context.dashboard_service,
+                service=self.context.get().dashboard_service,
                 owner=self.get_owner_ref(dashboard_details=dashboard_details),
             )
             yield Either(right=dashboard_request)
@@ -200,7 +202,7 @@ class MetabaseSource(DashboardServiceSource):
                         description=chart_details.description,
                         chartType=get_standard_chart_type(chart_details.display).value,
                         sourceUrl=chart_url,
-                        service=self.context.dashboard_service,
+                        service=self.context.get().dashboard_service,
                     )
                 )
             except Exception as exc:  # pylint: disable=broad-except

metadata/ingestion/source/dashboard/metabase/models.py

@@ -37,7 +37,7 @@ class MetabaseCollection(BaseModel):
 
 
 class MetabaseDashboardList(BaseModel):
-    dashboards: Optional[List[MetabaseDashboard]]
+    data: Optional[List[MetabaseDashboard]]
 
 
 class MetabaseCollectionList(BaseModel):

metadata/ingestion/source/dashboard/mode/metadata.py

@@ -59,7 +59,9 @@ class ModeSource(DashboardServiceSource):
         self.data_sources = self.client.get_all_data_sources(self.workspace_name)
 
     @classmethod
-    def create(cls, config_dict, metadata: OpenMetadata):
+    def create(
+        cls, config_dict, metadata: OpenMetadata, pipeline_name: Optional[str] = None
+    ):
         config = WorkflowSource.parse_obj(config_dict)
         connection: ModeConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, ModeConnection):
@@ -103,12 +105,12 @@ class ModeSource(DashboardServiceSource):
                 fqn.build(
                     self.metadata,
                     entity_type=Chart,
-                    service_name=self.context.dashboard_service,
+                    service_name=self.context.get().dashboard_service,
                     chart_name=chart,
                 )
-                for chart in self.context.charts or []
+                for chart in self.context.get().charts or []
             ],
-            service=self.context.dashboard_service,
+            service=self.context.get().dashboard_service,
             owner=self.get_owner_ref(dashboard_details=dashboard_details),
         )
         yield Either(right=dashboard_request)
@@ -203,7 +205,7 @@ class ModeSource(DashboardServiceSource):
                             displayName=chart_name,
                             chartType=ChartType.Other,
                             sourceUrl=chart_url,
-                            service=self.context.dashboard_service,
+                            service=self.context.get().dashboard_service,
                         )
                     )
                 except Exception as exc:

metadata/ingestion/source/dashboard/mstr/metadata.py

@@ -48,7 +48,12 @@ class MstrSource(DashboardServiceSource):
     """
 
     @classmethod
-    def create(cls, config_dict: dict, metadata: OpenMetadata):
+    def create(
+        cls,
+        config_dict: dict,
+        metadata: OpenMetadata,
+        pipeline_name: Optional[str] = None,
+    ):
         config = WorkflowSource.parse_obj(config_dict)
         connection: MstrConnection = config.serviceConnection.__root__.config
         if not isinstance(connection, MstrConnection):
@@ -110,12 +115,12 @@ class MstrSource(DashboardServiceSource):
                     fqn.build(
                         self.metadata,
                         entity_type=Chart,
-                        service_name=self.context.dashboard_service,
+                        service_name=self.context.get().dashboard_service,
                         chart_name=chart,
                     )
-                    for chart in self.context.charts or []
+                    for chart in self.context.get().charts or []
                 ],
-                service=self.context.dashboard_service,
+                service=self.context.get().dashboard_service,
                 owner=self.get_owner_ref(dashboard_details=dashboard_details),
             )
             yield Either(right=dashboard_request)
@@ -169,7 +174,7 @@ class MstrSource(DashboardServiceSource):
                         chartType=get_standard_chart_type(
                             chart.visualizationType
                         ).value,
-                        service=self.context.dashboard_service,
+                        service=self.context.get().dashboard_service,
                     )
                 )
             except Exception as exc: