semantic-link-labs 0.8.10__py3-none-any.whl → 0.9.0__py3-none-any.whl

sempy_labs/report/_report_list_functions.py

@@ -3,13 +3,16 @@ from typing import Optional
 import pandas as pd
 from sempy_labs._helper_functions import (
     format_dax_object_name,
+    resolve_workspace_name_and_id,
+    resolve_dataset_name_and_id,
 )
 from sempy_labs.report._reportwrapper import ReportWrapper
 from sempy_labs._list_functions import list_reports_using_semantic_model
+from uuid import UUID
 
 
 def list_unused_objects_in_reports(
-    dataset: str, workspace: Optional[str] = None
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
     """
     Shows a list of all columns in the semantic model which are not used in any related Power BI reports (including dependencies).
@@ -17,10 +20,10 @@ def list_unused_objects_in_reports(
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    dataset : str | uuid.UUID
+        Name or ID of the semantic model.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -32,7 +35,12 @@ def list_unused_objects_in_reports(
 
     # TODO: what about relationships/RLS?
 
-    dfR = _list_all_report_semantic_model_objects(dataset=dataset, workspace=workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
+
+    dfR = _list_all_report_semantic_model_objects(
+        dataset=dataset_id, workspace=workspace_id
+    )
     dfR_filt = (
         dfR[dfR["Object Type"] == "Column"][["Table Name", "Object Name"]]
         .drop_duplicates()
@@ -42,7 +50,7 @@ def list_unused_objects_in_reports(
         dfR_filt["Table Name"], dfR_filt["Object Name"]
     )
 
-    dfC = fabric.list_columns(dataset=dataset, workspace=workspace)
+    dfC = fabric.list_columns(dataset=dataset_id, workspace=workspace_id)
     dfC["Column Object"] = format_dax_object_name(dfC["Table Name"], dfC["Column Name"])
 
     df = dfC[~(dfC["Column Object"].isin(dfR_filt["Column Object"].values))]
@@ -52,7 +60,7 @@ def list_unused_objects_in_reports(
 
 
 def _list_all_report_semantic_model_objects(
-    dataset: str, workspace: Optional[str] = None
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
     """
     Shows a unique list of all semantic model objects (columns, measures, hierarchies) which are used in all reports which leverage the semantic model.
@@ -60,10 +68,10 @@ def _list_all_report_semantic_model_objects(
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    dataset : str | uuid.UUID
+        Name or ID of the semantic model.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -73,7 +81,10 @@ def _list_all_report_semantic_model_objects(
         A pandas dataframe.
     """
 
-    dfR = list_reports_using_semantic_model(dataset=dataset, workspace=workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
+
+    dfR = list_reports_using_semantic_model(dataset=dataset_id, workspace=workspace_id)
     dfs = []
 
     for _, r in dfR.iterrows():

sempy_labs/report/_report_rebind.py

@@ -1,20 +1,22 @@
 import sempy.fabric as fabric
 from sempy_labs._helper_functions import (
     resolve_dataset_id,
+    resolve_workspace_name_and_id,
     resolve_report_id,
 )
 from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
 @log
 def report_rebind(
     report: str | List[str],
     dataset: str,
-    report_workspace: Optional[str] = None,
-    dataset_workspace: Optional[str] = None,
+    report_workspace: Optional[str | UUID] = None,
+    dataset_workspace: Optional[str | UUID] = None,
 ):
     """
     Rebinds a report to a semantic model.
@@ -27,23 +29,22 @@ def report_rebind(
         Name(s) of the Power BI report(s).
     dataset : str
         Name of the semantic model.
-    report_workspace : str, default=None
-        The name of the Fabric workspace in which the report resides.
+    report_workspace : str | uuid.UUID, default=None
+        The name or ID of the Fabric workspace in which the report resides.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-    dataset_workspace : str, default=None
-        The name of the Fabric workspace in which the semantic model resides.
+    dataset_workspace : str | uuid.UUID, default=None
+        The name or ID of the Fabric workspace in which the semantic model resides.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    if report_workspace is None:
-        report_workspace_id = fabric.get_workspace_id()
-        report_workspace = fabric.resolve_workspace_name(report_workspace_id)
-    else:
-        report_workspace_id = fabric.resolve_workspace_id(report_workspace)
+    (report_workspace_name, report_workspace_id) = resolve_workspace_name_and_id(
+        report_workspace
+    )
+
     if dataset_workspace is None:
-        dataset_workspace = report_workspace
+        dataset_workspace = report_workspace_name
 
     client = fabric.PowerBIRestClient()
 
@@ -51,14 +52,14 @@ def report_rebind(
         report = [report]
 
     for rpt in report:
-        reportId = resolve_report_id(report=rpt, workspace=report_workspace)
-        datasetId = resolve_dataset_id(dataset=dataset, workspace=dataset_workspace)
+        report_id = resolve_report_id(report=rpt, workspace=report_workspace_id)
+        dataset_id = resolve_dataset_id(dataset=dataset, workspace=dataset_workspace)
 
         # Prepare API
-        request_body = {"datasetId": datasetId}
+        request_body = {"datasetId": dataset_id}
 
         response = client.post(
-            f"/v1.0/myorg/groups/{report_workspace_id}/reports/{reportId}/Rebind",
+            f"/v1.0/myorg/groups/{report_workspace_id}/reports/{report_id}/Rebind",
             json=request_body,
         )
 

sempy_labs/report/_reportwrapper.py

@@ -8,6 +8,7 @@ from sempy_labs._helper_functions import (
     _add_part,
     lro,
     _decode_b64,
+    resolve_workspace_name_and_id,
 )
 from typing import Optional, List
 import pandas as pd
@@ -32,8 +33,8 @@ class ReportWrapper:
     ----------
     report : str
         The name of the report.
-    workspace : str
-        The name of the workspace in which the report resides.
+    workspace : str | uuid.UUID
+        The name or ID of the workspace in which the report resides.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -50,7 +51,7 @@ class ReportWrapper:
     def __init__(
         self,
         report: str,
-        workspace: Optional[str] = None,
+        workspace: Optional[str | UUID] = None,
     ):
         """
         Connects to a Power BI report and retrieves its definition.
@@ -61,8 +62,8 @@ class ReportWrapper:
         ----------
         report : str
             The name of the report.
-        workspace : str
-            The name of the workspace in which the report resides.
+        workspace : str | UUID
+            The name or ID of the workspace in which the report resides.
             Defaults to None which resolves to the workspace of the attached lakehouse
             or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -77,11 +78,12 @@ class ReportWrapper:
         warnings.simplefilter(action="ignore", category=FutureWarning)
 
         self._report = report
-        self._workspace = workspace
-        self._workspace_id = fabric.resolve_workspace_id(workspace)
-        self._report_id = resolve_report_id(report, workspace)
+        (self._workspace_name, self._workspace_id) = resolve_workspace_name_and_id(
+            workspace
+        )
+        self._report_id = resolve_report_id(report, self._workspace_id)
         self.rdef = get_report_definition(
-            report=self._report, workspace=self._workspace
+            report=self._report, workspace=self._workspace_id
         )
 
         if len(self.rdef[self.rdef["path"] == "definition/report.json"]) == 0:
@@ -95,12 +97,14 @@ class ReportWrapper:
 
         from sempy_labs.tom import connect_semantic_model
 
-        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace = (
-            resolve_dataset_from_report(report=self._report, workspace=self._workspace)
+        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace_name = (
+            resolve_dataset_from_report(
+                report=self._report, workspace=self._workspace_id
+            )
         )
 
         with connect_semantic_model(
-            dataset=dataset_name, readonly=True, workspace=dataset_workspace
+            dataset=dataset_id, readonly=True, workspace=dataset_workspace_id
         ) as tom:
             for index, row in dataframe.iterrows():
                 obj_type = row["Object Type"]
@@ -218,7 +222,7 @@ class ReportWrapper:
         theme_collection = rptJson.get("themeCollection", {})
         if theme_type not in theme_collection:
             raise ValueError(
-                f"{icons.red_dot} The {self._report} report within the '{self._workspace} workspace has no custom theme."
+                f"{icons.red_dot} The {self._report} report within the '{self._workspace_name} workspace has no custom theme."
             )
         ct = theme_collection.get(theme_type)
         theme_name = ct["name"]
@@ -413,7 +417,7 @@ class ReportWrapper:
                             )
 
         df["Page URL"] = df["Page Name"].apply(
-            lambda page_name: f"{helper.get_web_url(report=self._report, workspace=self._workspace)}/{page_name}"
+            lambda page_name: f"{helper.get_web_url(report=self._report, workspace=self._workspace_id)}/{page_name}"
         )
 
         bool_cols = ["Hidden", "Locked", "Used"]
@@ -691,7 +695,7 @@ class ReportWrapper:
         df[bool_cols] = df[bool_cols].astype(bool)
 
         df["Page URL"] = df["Page Name"].apply(
-            lambda page_name: f"{helper.get_web_url(report=self._report, workspace=self._workspace)}/{page_name}"
+            lambda page_name: f"{helper.get_web_url(report=self._report, workspace=self._workspace_id)}/{page_name}"
         )
 
         return df
@@ -1170,9 +1174,9 @@ class ReportWrapper:
         )
 
         if extended:
-            dataset_id, dataset_name, dataset_workspace_id, dataset_workspace = (
+            dataset_id, dataset_name, dataset_workspace_id, dataset_workspace_name = (
                 resolve_dataset_from_report(
-                    report=self._report, workspace=self._workspace
+                    report=self._report, workspace=self._workspace_id
                 )
             )
 
@@ -1195,7 +1199,7 @@ class ReportWrapper:
                 return object_validators.get(row["Object Type"], lambda: False)()
 
             with connect_semantic_model(
-                dataset=dataset_name, readonly=True, workspace=dataset_workspace
+                dataset=dataset_id, readonly=True, workspace=dataset_workspace_id
             ) as tom:
                 df["Valid Semantic Model Object"] = df.apply(
                     lambda row: check_validity(tom, row), axis=1
@@ -1214,11 +1218,13 @@ class ReportWrapper:
             .drop_duplicates()
             .reset_index(drop=True)
         )
-        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace = (
-            resolve_dataset_from_report(report=self._report, workspace=self._workspace)
+        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace_name = (
+            resolve_dataset_from_report(
+                report=self._report, workspace=self._workspace_id
+            )
         )
         dep = get_measure_dependencies(
-            dataset=dataset_name, workspace=dataset_workspace
+            dataset=dataset_id, workspace=dataset_workspace_id
         )
         rpt_measures = df[df["Object Type"] == "Measure"]["Object Name"].values
         new_rows = dep[dep["Object Name"].isin(rpt_measures)][
@@ -1232,7 +1238,7 @@ class ReportWrapper:
         )
 
         result_df["Dataset Name"] = dataset_name
-        result_df["Dataset Workspace Name"] = dataset_workspace
+        result_df["Dataset Workspace Name"] = dataset_workspace_name
         colName = "Dataset Name"
         result_df.insert(0, colName, result_df.pop(colName))
         colName = "Dataset Workspace Name"
@@ -1539,7 +1545,7 @@ class ReportWrapper:
 
         self.update_report(request_body=request_body)
         print(
-            f"{icons.green_dot} The '{theme_name}' theme has been set as the theme for the '{self._report}' report within the '{self._workspace}' workspace."
+            f"{icons.green_dot} The '{theme_name}' theme has been set as the theme for the '{self._report}' report within the '{self._workspace_name}' workspace."
         )
 
     def set_active_page(self, page_name: str):
@@ -1567,7 +1573,7 @@ class ReportWrapper:
         self._update_single_file(file_name=pages_file, new_payload=file_payload)
 
         print(
-            f"{icons.green_dot} The '{page_display_name}' page has been set as the active page in the '{self._report}' report within the '{self._workspace}' workspace."
+            f"{icons.green_dot} The '{page_display_name}' page has been set as the active page in the '{self._report}' report within the '{self._workspace_name}' workspace."
         )
 
     def set_page_type(self, page_name: str, page_type: str):
@@ -1640,7 +1646,7 @@ class ReportWrapper:
                 cv_remove_display.append(cv_display)
         if len(cv_remove) == 0:
             print(
-                f"{icons.green_dot} There are no unnecessary custom visuals in the '{self._report}' report within the '{self._workspace}' workspace."
+                f"{icons.green_dot} There are no unnecessary custom visuals in the '{self._report}' report within the '{self._workspace_name}' workspace."
             )
             return
 
@@ -1662,7 +1668,7 @@ class ReportWrapper:
 
         self.update_report(request_body=request_body)
         print(
-            f"{icons.green_dot} The {cv_remove_display} custom visuals have been removed from the '{self._report}' report within the '{self._workspace}' workspace."
+            f"{icons.green_dot} The {cv_remove_display} custom visuals have been removed from the '{self._report}' report within the '{self._workspace_name}' workspace."
         )
 
     def migrate_report_level_measures(self, measures: Optional[str | List[str]] = None):
@@ -1681,12 +1687,14 @@ class ReportWrapper:
         rlm = self.list_report_level_measures()
         if len(rlm) == 0:
             print(
-                f"{icons.green_dot} The '{self._report}' report within the '{self._workspace}' workspace has no report-level measures."
+                f"{icons.green_dot} The '{self._report}' report within the '{self._workspace_name}' workspace has no report-level measures."
             )
             return
 
-        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace = (
-            resolve_dataset_from_report(report=self._report, workspace=self._workspace)
+        dataset_id, dataset_name, dataset_workspace_id, dataset_workspace_name = (
+            resolve_dataset_from_report(
+                report=self._report, workspace=self._workspace_id
+            )
         )
 
         if isinstance(measures, str):
@@ -1703,7 +1711,7 @@ class ReportWrapper:
 
         mCount = 0
         with connect_semantic_model(
-            dataset=dataset_name, readonly=False, workspace=dataset_workspace
+            dataset=dataset_id, readonly=False, workspace=dataset_workspace_id
         ) as tom:
             for _, r in rlm.iterrows():
                 tableName = r["Table Name"]
@@ -1748,7 +1756,7 @@ class ReportWrapper:
 
         self.update_report(request_body=request_body)
         print(
-            f"{icons.green_dot} The report-level measures have been migrated to the '{dataset_name}' semantic model within the '{dataset_workspace}' workspace."
+            f"{icons.green_dot} The report-level measures have been migrated to the '{dataset_name}' semantic model within the '{dataset_workspace_name}' workspace."
         )
 
     def set_page_visibility(self, page_name: str, hidden: bool):
@@ -1798,7 +1806,7 @@ class ReportWrapper:
 
         if len(dfP_filt) == 0:
             print(
-                f"{icons.green_dot} There are no Tooltip or Drillthrough pages in the '{self._report}' report within the '{self._workspace}' workspace."
+                f"{icons.green_dot} There are no Tooltip or Drillthrough pages in the '{self._report}' report within the '{self._workspace_name}' workspace."
             )
             return
 
@@ -1837,7 +1845,7 @@ class ReportWrapper:
 
         self.update_report(request_body=request_body)
         print(
-            f"{icons.green_dot} Show items with data has been disabled for all visuals in the '{self._report}' report within the '{self._workspace}' workspace."
+            f"{icons.green_dot} Show items with data has been disabled for all visuals in the '{self._report}' report within the '{self._workspace_name}' workspace."
         )
 
     # Set Annotations

sempy_labs/tom/_model.py

@@ -20,6 +20,8 @@ import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
 import ast
 from uuid import UUID
+import sempy_labs._authentication as auth
+
 
 if TYPE_CHECKING:
     import Microsoft.AnalysisServices.Tabular
@@ -44,19 +46,91 @@ class TOMWrapper:
 
     def __init__(self, dataset, workspace, readonly):
 
-        (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-        (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
-        self._dataset_id = dataset_id
-        self._dataset_name = dataset_name
-        self._workspace_name = workspace_name
-        self._workspace_id = workspace_id
+        self._is_azure_as = False
+        prefix = "asazure"
+        prefix_full = f"{prefix}://"
+        read_write = ":rw"
+        self._token_provider = auth.token_provider.get()
+
+        # Azure AS workspace logic
+        if workspace is not None and workspace.startswith(prefix_full):
+            # Set read or read/write accordingly
+            if readonly is False and not workspace.endswith(read_write):
+                workspace += read_write
+            elif readonly is True and workspace.endswith(read_write):
+                workspace = workspace[: -len(read_write)]
+            self._workspace_name = workspace
+            self._workspace_id = workspace
+            self._dataset_id = dataset
+            self._dataset_name = dataset
+            self._is_azure_as = True
+            if self._token_provider is None:
+                raise ValueError(
+                    f"{icons.red_dot} A token provider must be provided when connecting to an Azure AS workspace."
+                )
+        else:
+            (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+            (dataset_name, dataset_id) = resolve_dataset_name_and_id(
+                dataset, workspace_id
+            )
+            self._dataset_id = dataset_id
+            self._dataset_name = dataset_name
+            self._workspace_name = workspace_name
+            self._workspace_id = workspace_id
         self._readonly = readonly
         self._tables_added = []
 
-        self._tom_server = fabric.create_tom_server(
-            readonly=readonly, workspace=workspace_id
-        )
-        self.model = self._tom_server.Databases[dataset_id].Model
+        # No token provider (standard authentication)
+        if self._token_provider is None:
+            self._tom_server = fabric.create_tom_server(
+                readonly=readonly, workspace=workspace_id
+            )
+        # Service Principal Authentication for Azure AS via token provider
+        elif self._is_azure_as:
+            import Microsoft.AnalysisServices.Tabular as TOM
+
+            # Extract region from the workspace
+            match = re.search(rf"{prefix_full}(.*?).{prefix}", self._workspace_name)
+            if match:
+                region = match.group(1)
+            if self._token_provider is None:
+                raise ValueError(
+                    f"{icons.red_dot} A token provider must be provided when connecting to Azure Analysis Services."
+                )
+            token = self._token_provider(audience="asazure", region=region)
+            connection_str = f'Provider=MSOLAP;Data Source={self._workspace_name};Password="{token}";Persist Security Info=True;Impersonation Level=Impersonate'
+            self._tom_server = TOM.Server()
+            self._tom_server.Connect(connection_str)
+        # Service Principal Authentication for Power BI via token provider
+        else:
+            from sempy.fabric._client._utils import _build_adomd_connection_string
+            import Microsoft.AnalysisServices.Tabular as TOM
+            from Microsoft.AnalysisServices import AccessToken
+            from sempy.fabric._token_provider import (
+                create_on_access_token_expired_callback,
+                ConstantTokenProvider,
+            )
+            from System import Func
+
+            token = self._token_provider(audience="pbi")
+            self._tom_server = TOM.Server()
+            get_access_token = create_on_access_token_expired_callback(
+                ConstantTokenProvider(token)
+            )
+            self._tom_server.AccessToken = get_access_token(None)
+            self._tom_server.OnAccessTokenExpired = Func[AccessToken, AccessToken](
+                get_access_token
+            )
+            workspace_url = f"powerbi://api.powerbi.com/v1.0/myorg/{workspace}"
+            connection_str = _build_adomd_connection_string(
+                workspace_url, readonly=readonly
+            )
+            self._tom_server.Connect(connection_str)
+
+        if self._is_azure_as:
+            self.model = self._tom_server.Databases.GetByName(self._dataset_name).Model
+        else:
+            self.model = self._tom_server.Databases[dataset_id].Model
 
         self._table_map = {}
         self._column_map = {}
@@ -3287,6 +3361,10 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
+        dependencies = dependencies[
+            dependencies["Object Name"] == dependencies["Parent Node"]
+        ]
+
         for obj in self.depends_on(object=object, dependencies=dependencies):
             if obj.ObjectType == TOM.ObjectType.Measure:
                 if (f"{obj.Parent.Name}[{obj.Name}]" in object.Expression) or (
@@ -3313,12 +3391,16 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
+        dependencies = dependencies[
+            dependencies["Object Name"] == dependencies["Parent Node"]
+        ]
+
         def create_pattern(tableList, b):
             patterns = [
-                r"(?<!" + re.escape(table) + r"\[)(?<!" + re.escape(table) + r"'\[)"
+                r"(?<!" + re.escape(table) + r")(?<!'" + re.escape(table) + r"')"
                 for table in tableList
             ]
-            combined_pattern = "".join(patterns) + re.escape(b)
+            combined_pattern = "".join(patterns) + re.escape(f"[{b}]")
             return combined_pattern
 
         for obj in self.depends_on(object=object, dependencies=dependencies):
@@ -4396,8 +4478,6 @@ class TOMWrapper:
         if isinstance(measure_name, str):
             measure_name = [measure_name]
 
-        client = fabric.FabricRestClient()
-
         if len(measure_name) > max_batch_size:
             measure_lists = [
                 measure_name[i : i + max_batch_size]
@@ -4438,6 +4518,7 @@ class TOMWrapper:
                     "modelItems"
                 ].append(new_item)
 
+            client = fabric.FabricRestClient()
             response = client.post("/explore/v202304/nl2nl/completions", json=payload)
             if response.status_code != 200:
                 raise FabricHTTPException(response)
@@ -4529,7 +4610,7 @@ class TOMWrapper:
 
     def add_role_member(self, role_name: str, member: str | List[str]):
         """
-        Adds a external model role member (AzureAD) to a role.
+        Adds an external model role member (AzureAD) to a role.
 
         Parameters
         ----------
@@ -4561,6 +4642,35 @@ class TOMWrapper:
                     f"{icons.yellow_dot} '{m}' is already a member in the '{role_name}' role."
                 )
 
+    def remove_role_member(self, role_name: str, member: str | List[str]):
+        """
+        Removes an external model role member (AzureAD) from a role.
+
+        Parameters
+        ----------
+        role_name : str
+            The role name.
+        member : str | List[str]
+            The email address(es) of the member(s) to remove.
+        """
+
+        if isinstance(member, str):
+            member = [member]
+
+        role = self.model.Roles[role_name]
+        current_members = {m.MemberName: m.Name for m in role.Members}
+        for m in member:
+            name = current_members.get(m)
+            if name is not None:
+                role.Members.Remove(role.Members[name])
+                print(
+                    f"{icons.green_dot} The '{m}' member has been removed from the '{role_name}' role."
+                )
+            else:
+                print(
+                    f"{icons.yellow_dot} '{m}' is not a member of the '{role_name}' role."
+                )
+
     def close(self):
 
         if not self._readonly and self.model is not None:
@@ -4630,22 +4740,26 @@ class TOMWrapper:
 @log
 @contextmanager
 def connect_semantic_model(
-    dataset: str | UUID, readonly: bool = True, workspace: Optional[str] = None
+    dataset: str | UUID,
+    readonly: bool = True,
+    workspace: Optional[str | UUID] = None,
 ) -> Iterator[TOMWrapper]:
     """
     Connects to the Tabular Object Model (TOM) within a semantic model.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
-    dataset : str | UUID
+    dataset : str | uuid.UUID
         Name or ID of the semantic model.
     readonly: bool, default=True
         Whether the connection is read-only or read/write. Setting this to False enables read/write which saves the changes made back to the server.
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID. Also supports Azure Analysis Services (Service Principal Authentication required).
+        If connecting to Azure Analysis Services, enter the workspace parameter in the following format: 'asazure://<region>.asazure.windows.net/<server_name>'.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-
     Returns
     -------
     typing.Iterator[TOMWrapper]
@@ -4655,7 +4769,11 @@ def connect_semantic_model(
     # initialize .NET to make sure System and Microsoft.AnalysisServices.Tabular is defined
     sempy.fabric._client._utils._init_analysis_services()
 
-    tw = TOMWrapper(dataset=dataset, workspace=workspace, readonly=readonly)
+    tw = TOMWrapper(
+        dataset=dataset,
+        workspace=workspace,
+        readonly=readonly,
+    )
     try:
         yield tw
     finally: