semantic-link-labs 0.11.3__py3-none-any.whl → 0.12.1__py3-none-any.whl

sempy_labs/_workspaces.py

@@ -1,7 +1,8 @@
 import pandas as pd
 import sempy_labs._icons as icons
-from typing import Optional
-from ._helper_functions import (
+from typing import Optional, Literal
+from sempy_labs._helper_functions import (
+    resolve_workspace_id,
     resolve_workspace_name_and_id,
     resolve_capacity_id,
     _base_api,
@@ -127,7 +128,7 @@ def list_workspace_users(workspace: Optional[str | UUID] = None) -> pd.DataFrame
         A pandas dataframe the users of a workspace and their properties.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     columns = {
         "User Name": "string",
@@ -176,6 +177,8 @@ def add_user_to_workspace(
 
     This is a wrapper function for the following API: `Groups - Add Group User <https://learn.microsoft.com/rest/api/power-bi/groups/add-group-user>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     email_address : str
@@ -217,6 +220,7 @@ def add_user_to_workspace(
         request=f"/v1.0/myorg/groups/{workspace_id}/users",
         method="post",
         payload=payload,
+        client="fabric_sp",
     )
     print(
         f"{icons.green_dot} The '{email_address}' user has been added as a{plural} '{role_name}' within the '{workspace_name}' workspace."
@@ -234,6 +238,8 @@ def assign_workspace_to_capacity(
 
     This is a wrapper function for the following API: `Workspaces - Assign To Capacity <https://learn.microsoft.com/rest/api/fabric/core/workspaces/assign-to-capacity>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity : str | uuid.UUID
@@ -260,6 +266,7 @@ def assign_workspace_to_capacity(
         method="post",
         payload=payload,
         status_codes=[200, 202],
+        client="fabric_sp",
     )
     print(
         f"{icons.green_dot} The '{workspace_name}' workspace has been assigned to the '{capacity}' capacity."
@@ -320,7 +327,7 @@ def list_workspace_role_assignments(
         A pandas dataframe showing the members of a given workspace and their roles.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
 
     columns = {
         "User Name": "string",
@@ -336,16 +343,23 @@ def list_workspace_role_assignments(
         client="fabric_sp",
     )
 
+    rows = []
     for r in responses:
         for i in r.get("value", []):
             principal = i.get("principal", {})
-            new_data = {
-                "User Name": principal.get("displayName"),
-                "Role Name": i.get("role"),
-                "Type": principal.get("type"),
-                "User Email": principal.get("userDetails", {}).get("userPrincipalName"),
-            }
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            rows.append(
+                {
+                    "User Name": principal.get("displayName"),
+                    "Role Name": i.get("role"),
+                    "Type": principal.get("type"),
+                    "User Email": principal.get("userDetails", {}).get(
+                        "userPrincipalName"
+                    ),
+                }
+            )
+
+    if rows:
+        df = pd.DataFrame(rows, columns=list(columns.keys()))
 
     return df
 
@@ -357,6 +371,8 @@ def delete_workspace(workspace: Optional[str | UUID] = None):
 
     This is a wrapper function for the following API: `Workspaces - Delete Workspace <https://learn.microsoft.com/rest/api/fabric/core/workspaces/delete-workspace>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     workspace : str | uuid.UUID, default=None
@@ -372,3 +388,121 @@ def delete_workspace(workspace: Optional[str | UUID] = None):
     )
 
     print(f"{icons.green_dot} The '{workspace_name}' workspace has been deleted.")
+
+
+@log
+def get_workspace_network_communication_policy(
+    workspace: Optional[str | UUID] = None,
+) -> pd.DataFrame:
+    """
+    Returns networking communication policy for the specified workspace. This feature is currently in preview.
+
+    This is a wrapper function for the following API: `Workspaces - Get Network Communication Policy <https://learn.microsoft.com/rest/api/fabric/core/workspaces/get-network-communication-policy>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    workspace : str | uuid.UUID, default=None
+        The 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.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing the networking communication policy for the specified workspace.
+    """
+
+    workspace_id = resolve_workspace_id(workspace)
+
+    columns = {
+        "Inbound Public Access Rules": "string",
+        "Outbound Public Access Rules": "string",
+    }
+
+    df = _create_dataframe(columns=columns)
+
+    data = _base_api(
+        request=f"/v1/workspaces/{workspace_id}/networking/communicationPolicy",
+        client="fabric_sp",
+    ).json()
+
+    if data:
+        df = pd.DataFrame(
+            [
+                {
+                    "Inbound Public Access Rules": data.get("inbound", {})
+                    .get("publicAccessRules", {})
+                    .get("defaultAction"),
+                    "Outbound Public Access Rules": data.get("outbound", {})
+                    .get("publicAccessRules", {})
+                    .get("defaultAction"),
+                }
+            ]
+        )
+
+    return df
+
+
+@log
+def set_workspace_network_communication_policy(
+    inbound_policy: Literal["Allow", "Deny"],
+    outbound_policy: Literal["Allow", "Deny"],
+    workspace: Optional[str | UUID] = None,
+):
+    """
+    Sets networking communication policy for the specified workspace. This API uses the PUT method and will overwrite all settings. Remaining policy will be set to default value if only partial policy is provided in the request body. Always run Get Network Communication Policy first and provide full policy in the request body. This feature is currently in preview.
+
+    This is a wrapper function for the following API: `Workspaces - Set Network Communication Policy <https://learn.microsoft.com/rest/api/fabric/core/workspaces/set-network-communication-policy>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    inbound_policy : Literal['Allow', 'Deny']
+        The policy for all inbound communications to a workspace.
+    outbound_policy : Literal['Allow', 'Deny']
+        The policy for all outbound communications to a workspace.
+    workspace : str | uuid.UUID, default=None
+        The 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.
+    """
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    inbound_policy = inbound_policy.capitalize()
+    outbound_policy = outbound_policy.capitalize()
+
+    if inbound_policy not in ["Allow", "Deny"]:
+        raise ValueError(
+            f"{icons.red_dot} The 'inbound_policy' must be either 'Allow' or 'Deny'."
+        )
+    if outbound_policy not in ["Allow", "Deny"]:
+        raise ValueError(
+            f"{icons.red_dot} The 'outbound_policy' must be either 'Allow' or 'Deny'."
+        )
+
+    payload = {
+        "inbound": {
+            "publicAccessRules": {
+                "defaultAction": inbound_policy,
+            }
+        },
+        "outbound": {
+            "publicAccessRules": {
+                "defaultAction": outbound_policy,
+            }
+        },
+    }
+    _base_api(
+        request=f"/v1/workspaces/{workspace_id}/networking/communicationPolicy",
+        client="fabric_sp",
+        payload=payload,
+        method="put",
+    )
+
+    print(
+        f"{icons.green_dot} The networking communication policy has been updated for the '{workspace_name}' workspace."
+    )

sempy_labs/admin/__init__.py

@@ -90,6 +90,10 @@ from ._tags import (
     create_tags,
     delete_tag,
 )
+from ._tenant_keys import (
+    list_tenant_keys,
+    rotate_tenant_key,
+)
 
 __all__ = [
     "list_items",
@@ -149,4 +153,6 @@ __all__ = [
     "list_tags",
     "create_tags",
     "delete_tag",
+    "list_tenant_keys",
+    "rotate_tenant_key",
 ]

sempy_labs/admin/_capacities.py

@@ -3,7 +3,7 @@ from uuid import UUID
 import sempy_labs._icons as icons
 from typing import Optional, Tuple
 from sempy._utils._log import log
-from .._helper_functions import (
+from sempy_labs._helper_functions import (
     _base_api,
     _build_url,
     _create_dataframe,
@@ -226,6 +226,7 @@ def get_capacity_state(capacity: Optional[str | UUID] = None):
 @log
 def list_capacities(
     capacity: Optional[str | UUID] = None,
+    include_tenant_key: bool = False,
 ) -> pd.DataFrame:
     """
     Shows the a list of capacities and their properties.
@@ -238,6 +239,8 @@ def list_capacities(
     ----------
     capacity : str | uuid.UUID, default=None
         The capacity name or ID to filter. If None, all capacities are returned.
+    include_tenant_key : bool, default=False
+        If True, obtains the `tenant key <https://learn.microsoft.com/rest/api/power-bi/admin/get-capacities-as-admin#example-with-expand-on-tenant-key>`_ properties.
 
     Returns
     -------
@@ -252,26 +255,46 @@ def list_capacities(
         "Region": "string",
         "State": "string",
         "Admins": "list",
+        "Users": "list",
     }
+    if include_tenant_key:
+        columns.update(
+            {
+                "Tenant Key Id": "string",
+                "Tenant Key Name": "string",
+            }
+        )
     df = _create_dataframe(columns=columns)
 
-    responses = _base_api(
-        request="/v1.0/myorg/admin/capacities", client="fabric_sp", uses_pagination=True
-    )
+    url = "/v1.0/myorg/admin/capacities"
+    if include_tenant_key:
+        url += "?$expand=tenantKey"
+
+    responses = _base_api(request=url, client="fabric_sp", uses_pagination=True)
 
     rows = []
     for r in responses:
         for i in r.get("value", []):
-            rows.append(
-                {
-                    "Capacity Id": i.get("id").lower(),
-                    "Capacity Name": i.get("displayName"),
-                    "Sku": i.get("sku"),
-                    "Region": i.get("region"),
-                    "State": i.get("state"),
-                    "Admins": [i.get("admins", [])],
-                }
-            )
+            row = {
+                "Capacity Id": i.get("id", "").lower(),
+                "Capacity Name": i.get("displayName"),
+                "Sku": i.get("sku"),
+                "Region": i.get("region"),
+                "State": i.get("state"),
+                "Admins": i.get("admins", []),
+                "Users": i.get("users", []),
+            }
+
+            if include_tenant_key:
+                tenant_key = i.get("tenantKey") or {}
+                row.update(
+                    {
+                        "Tenant Key Id": tenant_key.get("id"),
+                        "Tenant Key Name": tenant_key.get("name"),
+                    }
+                )
+
+            rows.append(row)
 
     if rows:
         df = pd.DataFrame(rows, columns=list(columns.keys()))

sempy_labs/admin/_items.py

@@ -228,8 +228,8 @@ def list_item_access_details(
             f"{icons.red_dot} The parameter 'item' and 'type' are mandatory."
         )
 
-    workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
-    item_name, item_id = _resolve_item_name_and_id(
+    (workspace_name, workspace_id) = _resolve_workspace_name_and_id(workspace)
+    (item_name, item_id) = _resolve_item_name_and_id(
         item=item, type=type, workspace=workspace_name
     )
 

sempy_labs/admin/_tenant_keys.py

@@ -0,0 +1,89 @@
+import pandas as pd
+from sempy_labs._helper_functions import (
+    _create_dataframe,
+    _update_dataframe_datatypes,
+    _base_api,
+)
+from sempy._utils._log import log
+from uuid import UUID
+import sempy_labs._icons as icons
+
+
+@log
+def list_tenant_keys() -> pd.DataFrame:
+    """
+    Returns the encryption keys for the tenant.
+
+    This is a wrapper function for the following API: `Admin - Get Power BI Encryption Keys <https://learn.microsoft.com/rest/api/power-bi/admin/get-power-bi-encryption-keys>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing the encryption keys for the tenant.
+    """
+
+    columns = {
+        "Tenant Key Id": "str",
+        "Tenant Key Name": "str",
+        "Key Vault Key Identifier": "str",
+        "Is Default": "bool",
+        "Created At": "datetime",
+        "Updated At": "datetime",
+    }
+
+    df = _create_dataframe(columns=columns)
+
+    result = _base_api(
+        request="/v1.0/myorg/admin/tenantKeys", client="fabric_sp"
+    ).json()
+
+    rows = []
+    for i in result.get("value", []):
+        rows.append(
+            {
+                "Tenant Key Id": i.get("id"),
+                "Tenant Key Name": i.get("name"),
+                "Key Vault Key Identifier": i.get("keyVaultKeyIdentifier"),
+                "Is Default": i.get("isDefault"),
+                "Created At": i.get("createdAt"),
+                "Updated At": i.get("updatedAt"),
+            }
+        )
+
+    if rows:
+        df = pd.DataFrame(rows, columns=list(columns.keys()))
+        _update_dataframe_datatypes(dataframe=df, column_map=columns)
+
+    return df
+
+
+@log
+def rotate_tenant_key(tenant_key_id: UUID, key_vault_key_identifier: str):
+    """
+    Rotate the encryption key for Power BI workspaces assigned to a capacity.
+
+    This is a wrapper function for the following API: `Admin - Rotate Power BI Encryption Key <https://learn.microsoft.com/rest/api/power-bi/admin/rotate-power-bi-encryption-key>`_.
+
+    Parameters
+    ----------
+    tenant_key_id : uuid.UUID
+        The tenant key ID
+    key_vault_key_identifier : str
+        The URI that uniquely specifies the encryption key in Azure Key Vault
+    """
+
+    payload = {
+        "keyVaultKeyIdentifier": key_vault_key_identifier,
+    }
+
+    _base_api(
+        request=f"/v1.0/myorg/admin/tenantKeys/{tenant_key_id}/Default.Rotate",
+        method="post",
+        payload=payload,
+    )
+
+    print(
+        f"{icons.green_dot} The encryption key for tenant {tenant_key_id} has been rotated successfully using the '{key_vault_key_identifier}' Key Vault key identifier."
+    )

sempy_labs/directlake/_dl_helper.py

@@ -145,7 +145,7 @@ def generate_direct_lake_semantic_model(
     expr = generate_shared_expression(
         item_name=lakehouse, item_type="Lakehouse", workspace=lakehouse_workspace
     )
-    dfD = fabric.list_datasets(workspace=workspace_id)
+    dfD = fabric.list_datasets(workspace=workspace_id, mode="rest")
     dfD_filt = dfD[dfD["Dataset Name"] == dataset]
 
     if len(dfD_filt) > 0 and not overwrite:

sempy_labs/lakehouse/__init__.py

@@ -30,6 +30,9 @@ from ._helper import (
     update_lakehouse,
     load_table,
 )
+from ._materialized_lake_views import (
+    refresh_materialized_lake_views,
+)
 
 __all__ = [
     "get_lakehouse_columns",
@@ -50,4 +53,5 @@ __all__ = [
     "delete_lakehouse",
     "update_lakehouse",
     "load_table",
+    "refresh_materialized_lake_views",
 ]

sempy_labs/lakehouse/_materialized_lake_views.py

@@ -0,0 +1,76 @@
+from typing import Optional
+from sempy_labs._helper_functions import (
+    resolve_workspace_name_and_id,
+    resolve_lakehouse_name_and_id,
+    _base_api,
+)
+from uuid import UUID
+from sempy._utils._log import log
+import sempy_labs._icons as icons
+import time
+from sempy_labs._job_scheduler import (
+    _get_item_job_instance,
+)
+import pandas as pd
+
+
+@log
+def refresh_materialized_lake_views(
+    lakehouse: Optional[str | UUID] = None, workspace: Optional[str | UUID] = None
+) -> pd.DataFrame:
+    """
+    Run on-demand Refresh MaterializedLakeViews job instance.
+
+    This is a wrapper function for the following API: `Background Jobs - Run On Demand Refresh Materialized Lake Views <https://learn.microsoft.com/rest/api/fabric/lakehouse/background-jobs/run-on-demand-refresh-materialized-lake-views>`_.
+
+    Parameters
+    ----------
+    lakehouse : str | uuid.UUID, default=None
+        The Fabric lakehouse name or ID.
+        Defaults to None which resolves to the lakehouse attached to the notebook.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID used by the lakehouse.
+        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
+    -------
+    pandas.DataFrame
+        A DataFrame containing the job instance details of the refresh materialized lake views operation.
+    """
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (lakehouse_name, lakehouse_id) = resolve_lakehouse_name_and_id(
+        lakehouse=lakehouse, workspace=workspace_id
+    )
+
+    response = _base_api(
+        request=f"/v1/workspaces/{workspace_id}/lakehouses/{lakehouse_id}/jobs/instances?jobType=RefreshMaterializedLakeViews",
+        method="post",
+        status_codes=[202],
+    )
+
+    print(
+        f"{icons.in_progress} The refresh materialized lake views job for the '{lakehouse_name}' lakehouse within the '{workspace_name}' workspace has been initiated."
+    )
+
+    status_url = response.headers.get("Location").split("fabric.microsoft.com")[1]
+    status = None
+    while status not in ["Completed", "Failed"]:
+        response = _base_api(request=status_url)
+        status = response.json().get("status")
+        time.sleep(3)
+
+    df = _get_item_job_instance(url=status_url)
+
+    if status == "Completed":
+        print(
+            f"{icons.green_dot} The refresh materialized lake views job for the '{lakehouse_name}' lakehouse within the '{workspace_name}' workspace has succeeded."
+        )
+    else:
+        print(status)
+        print(
+            f"{icons.red_dot} The refresh materialized lake views job for the '{lakehouse_name}' lakehouse within the '{workspace_name}' workspace has failed."
+        )
+
+    return df

sempy_labs/lakehouse/_shortcuts.py

@@ -2,6 +2,7 @@ import sempy.fabric as fabric
 import pandas as pd
 from .._helper_functions import (
     resolve_lakehouse_name_and_id,
+    resolve_workspace_id,
     resolve_workspace_name_and_id,
     _base_api,
     _create_dataframe,
@@ -268,6 +269,8 @@ def reset_shortcut_cache(workspace: Optional[str | UUID] = None):
 
     This is a wrapper function for the following API: `OneLake Shortcuts - Reset Shortcut Cache <https://learn.microsoft.com/rest/api/fabric/core/onelake-shortcuts/reset-shortcut-cache>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     workspace : str | uuid.UUID, default=None
@@ -281,6 +284,7 @@ def reset_shortcut_cache(workspace: Optional[str | UUID] = None):
     _base_api(
         request=f"/v1/workspaces/{workspace_id}/onelake/resetShortcutCache",
         method="post",
+        client="fabric_sp",
         lro_return_status_code=True,
         status_codes=None,
     )
@@ -318,7 +322,7 @@ def list_shortcuts(
         A pandas dataframe showing all the shortcuts which exist in the specified lakehouse.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
     (lakehouse_name, lakehouse_id) = resolve_lakehouse_name_and_id(
         lakehouse=lakehouse, workspace=workspace_id
     )
@@ -383,7 +387,9 @@ def list_shortcuts(
             source_item_id = tgt.get(sources.get(tgt_type), {}).get("itemId")
             bucket = tgt.get(sources.get(tgt_type), {}).get("bucket")
             source_workspace_name = (
-                resolve_workspace_name(workspace_id=source_workspace_id)
+                resolve_workspace_name(
+                    workspace_id=source_workspace_id, throw_error=False
+                )
                 if source_workspace_id is not None
                 else None
             )

sempy_labs/report/_download_report.py

@@ -25,6 +25,8 @@ def download_report(
 
     This is a wrapper function for the following API: `Reports - Export Report In Group <https://learn.microsoft.com/rest/api/power-bi/reports/export-report-in-group>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     report: str | uuid.UUID
@@ -59,7 +61,8 @@ def download_report(
     report_id = resolve_item_id(item=report, type="Report", workspace=workspace)
 
     response = _base_api(
-        request=f"v1.0/myorg/groups/{workspace_id}/reports/{report_id}/Export?downloadType={download_type}"
+        request=f"v1.0/myorg/groups/{workspace_id}/reports/{report_id}/Export?downloadType={download_type}",
+        client="fabric_sp",
     )
 
     # Save file to the attached lakehouse

sempy_labs/report/_export_report.py

@@ -2,7 +2,7 @@ import sempy.fabric as fabric
 import json
 import os
 import time
-from .._helper_functions import (
+from sempy_labs._helper_functions import (
     generate_embedded_filter,
     resolve_workspace_name_and_id,
     _base_api,
@@ -12,7 +12,7 @@ from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 from uuid import UUID
-from ._report_functions import (
+from sempy_labs.report._report_functions import (
     list_report_visuals,
     list_report_pages,
 )
@@ -36,6 +36,8 @@ def export_report(
 
     This is a wrapper function for the following APIs: `Reports - Export To File In Group <https://learn.microsoft.com/rest/api/power-bi/reports/export-to-file-in-group>`_, `Reports - Get Export To File Status In Group <https://learn.microsoft.com/rest/api/power-bi/reports/get-export-to-file-status-in-group>`_, `Reports - Get File Of Export To File In Group <https://learn.microsoft.com/rest/api/power-bi/reports/get-file-of-export-to-file-in-group>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     report : str
@@ -241,22 +243,27 @@ def export_report(
         method="post",
         payload=request_body,
         status_codes=202,
+        client="fabric_sp",
     )
     export_id = json.loads(response.content).get("id")
 
     get_status_url = f"{base_url}/exports/{export_id}"
-    response = _base_api(request=get_status_url, status_codes=[200, 202])
+    response = _base_api(
+        request=get_status_url, status_codes=[200, 202], client="fabric_sp"
+    )
     response_body = json.loads(response.content)
     while response_body["status"] not in ["Succeeded", "Failed"]:
         time.sleep(3)
-        response = _base_api(request=get_status_url, status_codes=[200, 202])
+        response = _base_api(
+            request=get_status_url, status_codes=[200, 202], client="fabric_sp"
+        )
         response_body = json.loads(response.content)
     if response_body["status"] == "Failed":
         raise ValueError(
             f"{icons.red_dot} The export for the '{report}' report within the '{workspace_name}' workspace in the '{export_format}' format has failed."
         )
     else:
-        response = _base_api(request=f"{get_status_url}/file")
+        response = _base_api(request=f"{get_status_url}/file", client="fabric_sp")
         print(
             f"{icons.in_progress} Saving the '{export_format}' export for the '{report}' report within the '{workspace_name}' workspace to the lakehouse..."
         )