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

sempy_labs/admin/_basic_functions.py

@@ -8,11 +8,14 @@ from sempy_labs._helper_functions import (
     _is_valid_uuid,
     _build_url,
 )
+from sempy._utils._log import log
 import numpy as np
 import pandas as pd
 from dateutil.parser import parse as dtparser
+import sempy_labs._authentication as auth
 
 
+@log
 def list_workspaces(
     capacity: Optional[str | UUID] = None,
     workspace: Optional[str | UUID] = None,
@@ -25,11 +28,13 @@ def list_workspaces(
 
     This is a wrapper function for the following API: `Workspaces - List Workspaces - REST API (Admin) <https://learn.microsoft.com/en-us/rest/api/fabric/admin/workspaces/list-workspaces>`_.
 
+    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, default=None
+    capacity : str | uuid.UUID, default=None
         Returns only the workspaces in the specified Capacity.
-    workspace : str | UUID, default=None
+    workspace : str | uuid.UUID, default=None
         Returns the workspace with the specific name.
     workspace_state : str, default=None
         Return only the workspace with the requested state. You can find the possible states in `Workspace States <https://learn.microsoft.com/en-us/rest/api/fabric/admin/workspaces/list-workspaces?tabs=HTTP#workspacestate>`_.
@@ -59,7 +64,7 @@ def list_workspaces(
         )
         del kwargs["skip"]
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     df = pd.DataFrame(
         columns=[
@@ -121,6 +126,7 @@ def list_workspaces(
     return df
 
 
+@log
 def list_capacities(
     capacity: Optional[str | UUID] = None,
 ) -> pd.DataFrame:
@@ -129,9 +135,11 @@ def list_capacities(
 
     This is a wrapper function for the following API: `Admin - Get Capacities As Admin <https://learn.microsoft.com/rest/api/power-bi/admin/get-capacities-as-admin>`_.
 
+    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, default=None
+    capacity : str | uuid.UUID, default=None
         Capacity name or id to filter.
 
     Returns
@@ -139,7 +147,7 @@ def list_capacities(
     pandas.DataFrame
         A pandas dataframe showing the capacities and their properties.
     """
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     df = pd.DataFrame(
         columns=["Capacity Id", "Capacity Name", "Sku", "Region", "State", "Admins"]
@@ -173,6 +181,7 @@ def list_capacities(
     return df
 
 
+@log
 def assign_workspaces_to_capacity(
     source_capacity: Optional[str | UUID] = None,
     target_capacity: Optional[str | UUID] = None,
@@ -185,11 +194,11 @@ def assign_workspaces_to_capacity(
 
     Parameters
     ----------
-    source_capacity : str | UUID, default=None
+    source_capacity : str | uuid.UUID, default=None
         The name of the source capacity. If the Workspace is not specified, this is parameter mandatory.
-    target_capacity : str | UUID, default=None
+    target_capacity : str | uuid.UUID, default=None
         The name of the target capacity.
-    workspace : str | List[str] | UUID | List[UUID], default=None
+    workspace : str | List[str] | uuid.UUID | List[uuid.UUID], default=None
         The name or id of the workspace(s).
         Defaults to None which resolves to migrating all workspaces within the source capacity to the target capacity.
     """
@@ -264,6 +273,7 @@ def assign_workspaces_to_capacity(
     )
 
 
+@log
 def unassign_workspaces_from_capacity(
     workspaces: str | List[str] | UUID | List[UUID],
 ):
@@ -274,7 +284,7 @@ def unassign_workspaces_from_capacity(
 
     Parameters
     ----------
-    workspaces : str | List[str] | UUID | List[UUID]
+    workspaces : str | List[str] | uuid.UUID | List[uuid.UUID]
         The Fabric workspace name(s) or id(s).
     """
     if isinstance(workspaces, str):
@@ -305,18 +315,21 @@ def unassign_workspaces_from_capacity(
     )
 
 
+@log
 def list_tenant_settings() -> pd.DataFrame:
     """
     Lists all tenant settings.
 
     This is a wrapper function for the following API: `Tenants - List Tenant Settings <https://learn.microsoft.com/rest/api/fabric/admin/tenants/list-tenant-settings>`_.
 
+    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 tenant settings.
     """
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     response = client.get("/v1/admin/tenantsettings")
 
@@ -359,6 +372,8 @@ def list_capacities_delegated_tenant_settings(
 
     This is a wrapper function for the following API: `Tenants - List Capacities Tenant Settings Overrides <https://learn.microsoft.com/rest/api/fabric/admin/tenants/list-capacities-tenant-settings-overrides>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     return_dataframe : bool, default=True
@@ -384,7 +399,7 @@ def list_capacities_delegated_tenant_settings(
         ]
     )
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     response = client.get("/v1/admin/capacities/delegatedTenantSettingOverrides")
 
     if response.status_code != 200:
@@ -450,6 +465,8 @@ def list_modified_workspaces(
 
     This is a wrapper function for the following API: `Admin - WorkspaceInfo GetModifiedWorkspaces <https://learn.microsoft.com/rest/api/power-bi/admin/workspace-info-get-modified-workspaces>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     modified_since : str, default=None
@@ -464,7 +481,7 @@ def list_modified_workspaces(
     pandas.DataFrame
         A pandas dataframe showing a list of workspace IDs in the organization.
     """
-    client = fabric.PowerBIRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     params = {}
 
@@ -504,6 +521,8 @@ def list_datasets(
 
     This is a wrapper function for the following API: `Admin - Datasets GetDatasetsAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/datasets-get-datasets-as-admin>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     top : int, default=None
@@ -512,39 +531,38 @@ def list_datasets(
         Returns a subset of a results based on Odata filter query parameter condition.
     skip : int, default=None
         Skips the first n results.
-    token_provider : Optional[TokenProvider] = None,
-        Authentication provider used to be use in the request. Supports Service Principal.
 
     Returns
     -------
     pandas.DataFrame
         A pandas dataframe showing a list of datasets for the organization.
     """
-    df = pd.DataFrame(
-        columns=[
-            "Dataset Id",
-            "Dataset Name",
-            "Web URL",
-            "Add Rows API Enabled",
-            "Configured By",
-            "Is Refreshable",
-            "Is Effective Identity Required",
-            "Is Effective Identity Roles Required",
-            "Target Storage Mode",
-            "Created Date",
-            "Content Provider Type",
-            "Create Report Embed URL",
-            "QnA Embed URL",
-            "Upstream Datasets",
-            "Users",
-            "Is In Place Sharing Enabled",
-            "Workspace Id",
-            "Auto Sync Read Only Replicas",
-            "Max Read Only Replicas",
-        ]
-    )
 
-    client = fabric.PowerBIRestClient()
+    columns = [
+        "Dataset Id",
+        "Dataset Name",
+        "Web URL",
+        "Add Rows API Enabled",
+        "Configured By",
+        "Is Refreshable",
+        "Is Effective Identity Required",
+        "Is Effective Identity Roles Required",
+        "Target Storage Mode",
+        "Created Date",
+        "Content Provider Type",
+        "Create Report Embed URL",
+        "QnA Embed URL",
+        "Upstream Datasets",
+        "Users",
+        "Is In Place Sharing Enabled",
+        "Workspace Id",
+        "Auto Sync Read Only Replicas",
+        "Max Read Only Replicas",
+    ]
+
+    df = pd.DataFrame(columns=columns)
+
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     params = {}
     url = "/v1.0/myorg/admin/datasets"
@@ -565,35 +583,40 @@ def list_datasets(
     if response.status_code != 200:
         raise FabricHTTPException(response)
 
+    rows = []
     for v in response.json().get("value", []):
-        new_data = {
-            "Dataset Id": v.get("id"),
-            "Dataset Name": v.get("name"),
-            "Web URL": v.get("webUrl"),
-            "Add Rows API Enabled": v.get("addRowsAPIEnabled"),
-            "Configured By": v.get("configuredBy"),
-            "Is Refreshable": v.get("isRefreshable"),
-            "Is Effective Identity Required": v.get("isEffectiveIdentityRequired"),
-            "Is Effective Identity Roles Required": v.get(
-                "isEffectiveIdentityRolesRequired"
-            ),
-            "Target Storage Mode": v.get("targetStorageMode"),
-            "Created Date": pd.to_datetime(v.get("createdDate")),
-            "Content Provider Type": v.get("contentProviderType"),
-            "Create Report Embed URL": v.get("createReportEmbedURL"),
-            "QnA Embed URL": v.get("qnaEmbedURL"),
-            "Upstream Datasets": v.get("upstreamDatasets", []),
-            "Users": v.get("users", []),
-            "Is In Place Sharing Enabled": v.get("isInPlaceSharingEnabled"),
-            "Workspace Id": v.get("workspaceId"),
-            "Auto Sync Read Only Replicas": v.get("queryScaleOutSettings", {}).get(
-                "autoSyncReadOnlyReplicas"
-            ),
-            "Max Read Only Replicas": v.get("queryScaleOutSettings", {}).get(
-                "maxReadOnlyReplicas"
-            ),
-        }
-        df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
+        rows.append(
+            {
+                "Dataset Id": v.get("id"),
+                "Dataset Name": v.get("name"),
+                "Web URL": v.get("webUrl"),
+                "Add Rows API Enabled": v.get("addRowsAPIEnabled"),
+                "Configured By": v.get("configuredBy"),
+                "Is Refreshable": v.get("isRefreshable"),
+                "Is Effective Identity Required": v.get("isEffectiveIdentityRequired"),
+                "Is Effective Identity Roles Required": v.get(
+                    "isEffectiveIdentityRolesRequired"
+                ),
+                "Target Storage Mode": v.get("targetStorageMode"),
+                "Created Date": pd.to_datetime(v.get("createdDate")),
+                "Content Provider Type": v.get("contentProviderType"),
+                "Create Report Embed URL": v.get("createReportEmbedURL"),
+                "QnA Embed URL": v.get("qnaEmbedURL"),
+                "Upstream Datasets": v.get("upstreamDatasets", []),
+                "Users": v.get("users", []),
+                "Is In Place Sharing Enabled": v.get("isInPlaceSharingEnabled"),
+                "Workspace Id": v.get("workspaceId"),
+                "Auto Sync Read Only Replicas": v.get("queryScaleOutSettings", {}).get(
+                    "autoSyncReadOnlyReplicas"
+                ),
+                "Max Read Only Replicas": v.get("queryScaleOutSettings", {}).get(
+                    "maxReadOnlyReplicas"
+                ),
+            }
+        )
+
+    if rows:
+        df = pd.DataFrame(rows, columns=columns)
 
     bool_cols = [
         "Add Rows API Enabled",
@@ -619,6 +642,8 @@ def list_access_entities(
 
     This is a wrapper function for the following API: `Users - List Access Entities <https://learn.microsoft.com/rest/api/fabric/admin/users/list-access-entities>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     user_email_address : str
@@ -638,7 +663,7 @@ def list_access_entities(
             "Additional Permissions",
         ]
     )
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     response = client.get(f"/v1/admin/users/{user_email_address}/access")
 
@@ -664,16 +689,18 @@ def list_access_entities(
 
 
 def list_workspace_access_details(
-    workspace: Optional[Union[str, UUID]] = None
+    workspace: Optional[Union[str, UUID]] = None,
 ) -> pd.DataFrame:
     """
     Shows a list of users (including groups and Service Principals) that have access to the specified workspace.
 
     This is a wrapper function for the following API: `Workspaces - List Workspace Access Details <https://learn.microsoft.com/rest/api/fabric/admin/workspaces/list-workspace-access-details>`_.
 
+    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, default=None
+    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.
@@ -683,7 +710,7 @@ def list_workspace_access_details(
     pandas.DataFrame
         A pandas dataframe showing a list of users (including groups and Service Principals) that have access to the specified workspace.
     """
-    workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
+    (workspace_name, workspace_id) = _resolve_workspace_name_and_id(workspace)
 
     df = pd.DataFrame(
         columns=[
@@ -696,7 +723,7 @@ def list_workspace_access_details(
         ]
     )
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
 
     response = client.get(f"/v1/admin/workspaces/{workspace_id}/users")
     if response.status_code != 200:
@@ -721,13 +748,15 @@ def list_activity_events(
     end_time: str,
     activity_filter: Optional[str] = None,
     user_id_filter: Optional[str] = None,
-    return_dataframe: Optional[bool] = True,
+    return_dataframe: bool = True,
 ) -> pd.DataFrame | dict:
     """
     Shows a list of audit activity events for a tenant.
 
     This is a wrapper function for the following API: `Admin - Get Activity Events <https://learn.microsoft.com/rest/api/power-bi/admin/get-activity-events>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     start_time : str
@@ -798,7 +827,7 @@ def list_activity_events(
     )
 
     response_json = {"activityEventEntities": []}
-    client = fabric.PowerBIRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     url = f"/v1.0/myorg/admin/activityevents?startDateTime='{start_time}'&endDateTime='{end_time}'"
 
     conditions = []
@@ -951,38 +980,51 @@ def _resolve_workspace_name_and_id(
 
 
 def list_reports(
-    top: Optional[int] = None, skip: Optional[int] = None, filter: Optional[str] = None
+    top: Optional[int] = None,
+    skip: Optional[int] = None,
+    filter: Optional[str] = None,
 ) -> pd.DataFrame:
     """
     Shows a list of reports for the organization.
 
     This is a wrapper function for the following API: `Admin - Reports GetReportsAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/reports-get-reports-as-admin>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    top : int, default=None
+        Returns only the first n results.
+    skip : int, default=None
+        Skips the first n results.
+    filter : str, default=None
+        Returns a subset of a results based on Odata filter query parameter condition.
+
     Returns
     -------
     pandas.DataFrame
         A pandas dataframe showing a list of reports for the organization.
     """
 
-    df = pd.DataFrame(
-        columns=[
-            "Report Id",
-            "Report Name",
-            "Type",
-            "Web URL",
-            "Embed URL",
-            "Dataset Id",
-            "Created Date",
-            "Modified Date",
-            "Created By",
-            "Modified By",
-            "Sensitivity Label Id",
-            "Users",
-            "Subscriptions",
-            "Workspace Id",
-            "Report Flags",
-        ]
-    )
+    columns = [
+        "Report Id",
+        "Report Name",
+        "Type",
+        "Web URL",
+        "Embed URL",
+        "Dataset Id",
+        "Created Date",
+        "Modified Date",
+        "Created By",
+        "Modified By",
+        "Sensitivity Label Id",
+        "Users",
+        "Subscriptions",
+        "Workspace Id",
+        "Report Flags",
+    ]
+
+    df = pd.DataFrame(columns=columns)
 
     url = "/v1.0/myorg/admin/reports?"
     if top is not None:
@@ -994,31 +1036,36 @@ def list_reports(
 
     url.rstrip("$").rstrip("?")
 
-    client = fabric.PowerBIRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     response = client.get(url)
 
     if response.status_code != 200:
         raise FabricHTTPException(response)
 
+    rows = []
     for v in response.json().get("value", []):
-        new_data = {
-            "Report Id": v.get("id"),
-            "Report Name": v.get("name"),
-            "Type": v.get("reportType"),
-            "Web URL": v.get("webUrl"),
-            "Embed URL": v.get("embedUrl"),
-            "Dataset Id": v.get("datasetId"),
-            "Created Date": v.get("createdDateTime"),
-            "Modified Date": v.get("modifiedDateTime"),
-            "Created By": v.get("createdBy"),
-            "Modified By": v.get("modifiedBy"),
-            "Sensitivity Label Id": v.get("sensitivityLabel", {}).get("labelId"),
-            "Users": v.get("users"),
-            "Subscriptions": v.get("subscriptions"),
-            "Workspace Id": v.get("workspaceId"),
-            "Report Flags": v.get("reportFlags"),
-        }
-        df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
+        rows.append(
+            {
+                "Report Id": v.get("id"),
+                "Report Name": v.get("name"),
+                "Type": v.get("reportType"),
+                "Web URL": v.get("webUrl"),
+                "Embed URL": v.get("embedUrl"),
+                "Dataset Id": v.get("datasetId"),
+                "Created Date": v.get("createdDateTime"),
+                "Modified Date": v.get("modifiedDateTime"),
+                "Created By": v.get("createdBy"),
+                "Modified By": v.get("modifiedBy"),
+                "Sensitivity Label Id": v.get("sensitivityLabel", {}).get("labelId"),
+                "Users": v.get("users"),
+                "Subscriptions": v.get("subscriptions"),
+                "Workspace Id": v.get("workspaceId"),
+                "Report Flags": v.get("reportFlags"),
+            }
+        )
+
+    if rows:
+        df = pd.DataFrame(rows, columns=columns)
 
     int_cols = ["Report Flags"]
     df[int_cols] = df[int_cols].astype(int)
@@ -1029,15 +1076,19 @@ def list_reports(
     return df
 
 
-def get_capacity_assignment_status(workspace: Optional[str | UUID] = None):
+def get_capacity_assignment_status(
+    workspace: Optional[str | UUID] = None,
+) -> pd.DataFrame:
     """
     Gets the status of the assignment-to-capacity operation for the specified workspace.
 
     This is a wrapper function for the following API: `Capacities - Groups CapacityAssignmentStatus <https://learn.microsoft.com/rest/api/power-bi/capacities/groups-capacity-assignment-status>`_.
 
+    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, default=None
+    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.
@@ -1061,7 +1112,7 @@ def get_capacity_assignment_status(workspace: Optional[str | UUID] = None):
         ]
     )
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     response = client.get(f"/v1.0/myorg/groups/{workspace_id}/CapacityAssignmentStatus")
 
     if response.status_code != 200:

sempy_labs/admin/_domains.py

@@ -6,6 +6,7 @@ from sempy.fabric.exceptions import FabricHTTPException
 import pandas as pd
 from uuid import UUID
 from sempy_labs.admin._basic_functions import list_workspaces
+import sempy_labs._authentication as auth
 
 
 def resolve_domain_id(domain_name: str) -> UUID:
@@ -37,6 +38,8 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
 
     This is a wrapper function for the following API: `Domains - List Domains <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domains>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     non_empty_only : bool, default=False
@@ -59,7 +62,7 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
         ]
     )
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     url = "/v1/admin/domains"
     if non_empty_only:
         url = f"{url}?nonEmptyOnly=True"
@@ -87,6 +90,8 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
 
     This is a wrapper function for the following API: `Domains - List Domain Workspaces <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domain-workspaces>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     domain_name : str
@@ -102,7 +107,7 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
 
     df = pd.DataFrame(columns=["Workspace ID", "Workspace Name"])
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     response = client.get(f"/v1/admin/domains/{domain_id}/workspaces")
 
     if response.status_code != 200:

sempy_labs/admin/_external_data_share.py

@@ -75,11 +75,11 @@ def revoke_external_data_share(
 
     Parameters
     ----------
-    external_data_share_id : UUID
+    external_data_share_id : uuid.UUID
         The external data share ID.
-    item_id : int, default=None
+    item_id : uuid.UUID, default=None
         The Item ID
-    workspace : str
+    workspace : str | uuid.UUID
         The Fabric workspace name or id.
     """
     (workspace, workspace_id) = _resolve_workspace_name_and_id(workspace)

sempy_labs/admin/_git.py

@@ -5,6 +5,7 @@ from sempy_labs._helper_functions import (
 )
 import pandas as pd
 from sempy_labs.admin._basic_functions import list_workspaces
+import sempy_labs._authentication as auth
 
 
 def list_git_connections() -> pd.DataFrame:
@@ -13,13 +14,15 @@ def list_git_connections() -> pd.DataFrame:
 
     This is a wrapper function for the following API: `Workspaces - List Git Connections <https://learn.microsoft.com/rest/api/fabric/admin/workspaces/list-git-connections>`_.
 
+    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 a list of Git connections.
     """
 
-    client = fabric.FabricRestClient()
+    client = fabric.FabricRestClient(token_provider=auth.token_provider.get())
     response = client.get("/v1/admin/workspaces/discoverGitConnections")
 
     df = pd.DataFrame(