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

sempy_labs/_helper_functions.py

@@ -21,6 +21,7 @@ from jsonpath_ng.ext import parse
 from jsonpath_ng.jsonpath import Fields, Index
 from sempy._utils._log import log
 from os import PathLike
+import sempy_labs._utils as utils
 
 
 def _build_url(url: str, params: dict) -> str:
@@ -226,8 +227,6 @@ def delete_item(
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    from sempy_labs._utils import item_types
-
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     (item_name, item_id) = resolve_item_name_and_id(item, type, workspace_id)
     item_type = item_types.get(type)[0].lower()
@@ -281,11 +280,10 @@ def create_item(
         The folder within the workspace where the item will be created.
         Defaults to None which places the item in the root of the workspace.
     """
-    from sempy_labs._utils import item_types
 
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-    item_type = item_types.get(type)[0].lower()
-    item_type_url = item_types.get(type)[1]
+    item_type = utils.item_types.get(type)[0].lower()
+    item_type_url = utils.item_types.get(type)[1]
 
     payload = {
         "displayName": name,
@@ -319,10 +317,13 @@ def copy_item(
     source_workspace: Optional[str | UUID] = None,
     target_workspace: Optional[str | UUID] = None,
     overwrite: bool = False,
+    keep_existing_bindings: bool = False,
 ):
     """
     Copies an item (with its definition) from one location to another location.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     item : str | uuid.UUID
@@ -339,40 +340,14 @@ def copy_item(
         The workspace name or ID to which the item will be copied.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
-    overwrite: bool, default=False
+    overwrite : bool, default=False
         If True, overwrites the item in the target workspace if it already exists.
+    keep_existing_bindings : bool, default=False
+        If True, ensures that reports are re-bound to the original semantic model.
+        If False, reports are binded to the semantic model to which the item is bound.
     """
 
-    items = {
-        "CopyJob": "copyJobs",
-        "Dataflow": "dataflows",
-        "Eventhouse": "eventhouses",
-        "GraphQLApi": "GraphQLApis",
-        "Report": "reports",
-        "SemanticModel": "semanticModels",
-        # "Environment": "environments",
-        "KQLDatabase": "kqlDatabases",
-        "KQLDashboard": "kqlDashboards",
-        "KQLQueryset": "kqlQuerysets",
-        "DataPipeline": "dataPipelines",
-        "Notebook": "notebooks",
-        "SparkJobDefinition": "sparkJobDefinitions",
-        "Eventstream": "eventstreams",
-        "MirroredWarehouse": "mirroredWarehouses",
-        "MirroredDatabase": "mirroredDatabases",
-        "MountedDataFactory": "mountedDataFactories",
-        "VariableLibrary": "variableLibraries",
-        "ApacheAirFlowJob": "ApacheAirflowJobs",
-        "WarehouseSnapshot": "warehousesnapshots",
-        "DigitalTwinBuilder": "digitaltwinbuilders",
-        "DigitalTwinBuilderFlow": "DigitalTwinBuilderFlows",
-        "MirroredAzureDatabricksCatalog": "mirroredAzureDatabricksCatalogs",
-    }
-    if type not in items:
-        raise ValueError(
-            f"{icons.red_dot} The '{type}' item type does not have a definition and cannot be copied."
-        )
-    type_url = items.get(type)
+    from sempy_labs.report import report_rebind
 
     (item_name, item_id) = resolve_item_name_and_id(
         item=item, type=type, workspace=source_workspace
@@ -392,18 +367,19 @@ def copy_item(
             f"{icons.red_dot} The source and target workspaces are the same and the target name is the same as the source name. No action taken."
         )
 
+    type_url = utils.items.get(type)
     result = _base_api(
         request=f"v1/workspaces/{source_workspace_id}/{type_url}/{item_id}",
         client="fabric_sp",
     )
     description = result.json().get("description")
 
-    payload = _base_api(
-        request=f"v1/workspaces/{source_workspace_id}/{type_url}/{item_id}/getDefinition",
-        method="post",
-        client="fabric_sp",
-        status_codes=None,
-        lro_return_json=True,
+    payload = get_item_definition(
+        item=item_id,
+        type=type,
+        workspace=source_workspace_id,
+        return_dataframe=False,
+        decode=False,
     )
     payload["displayName"] = target_name
     if description:
@@ -428,6 +404,13 @@ def copy_item(
         print(
             f"{icons.in_progress} Updating existing item '{target_name}' of type '{type}' in the target workspace '{target_workspace_name}'..."
         )
+        # Get the existing source model
+        if type == "Report" and keep_existing_bindings:
+            result = _base_api(
+                request=f"v1.0/myorg/groups/{target_workspace_id}/reports/{target_item_id}"
+            ).json()
+            dataset_id = result.get("datasetId")
+            dataset_workspace_id = result.get("datasetWorkspaceId")
         _base_api(
             request=f"/v1/workspaces/{target_workspace_id}/{type_url}/{target_item_id}/updateDefinition",
             method="post",
@@ -439,6 +422,15 @@ def copy_item(
         print(
             f"{icons.green_dot} The item '{target_name}' of type '{type}' has been successfully updated in the target workspace '{target_workspace_name}'."
         )
+
+        if keep_existing_bindings:
+            report_rebind(
+                report=target_item_id,
+                dataset=dataset_id,
+                report_workspace=target_workspace,
+                dataset_workspace=dataset_workspace_id,
+            )
+
     else:
         print(
             f"{icons.in_progress} Creating new item '{target_name}' of type '{type}' in the target workspace '{target_workspace_name}'..."
@@ -451,8 +443,115 @@ def copy_item(
         )
 
 
+@log
+def is_base64(s):
+    try:
+        # Add padding if needed
+        s_padded = s + "=" * (-len(s) % 4)
+        decoded = base64.b64decode(s_padded, validate=True)
+        # Optional: check if re-encoding gives the original (excluding padding)
+        return base64.b64encode(decoded).decode().rstrip("=") == s.rstrip("=")
+    except Exception:
+        return False
+
+
+@log
+def decode_payload(payload):
+
+    if is_base64(payload):
+        try:
+            decoded_payload = json.loads(base64.b64decode(payload).decode("utf-8"))
+        except Exception:
+            decoded_payload = base64.b64decode(payload)
+    elif isinstance(payload, dict):
+        decoded_payload = payload
+    else:
+        raise ValueError("Payload must be a dictionary or a base64 encoded value.")
+
+    return decoded_payload
+
+
 @log
 def get_item_definition(
+    item: str | UUID,
+    type: str,
+    workspace: Optional[str | UUID] = None,
+    return_dataframe: bool = False,
+    decode: bool = True,
+    format: Optional[str] = None,
+) -> dict | pd.DataFrame:
+    """
+    Gets a Fabric item's defintion.
+
+    This is a wrapper function for the following API: `<https://learn.microsoft.com/rest/api/fabric/core/items/get-item-definition>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    item : str | uuid.UUID
+        The name or ID of the item to be copied.
+    type : str
+        The `type <https://learn.microsoft.com/rest/api/fabric/core/items/list-items?tabs=HTTP#itemtype>`_ of the item.
+    target_name: str, default=None
+        The name of the item in the target workspace. Defaults to the same name as the source item.
+    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.
+    return_dataframe : bool, default=False
+        If True, returns a pandas dataframe.
+        If False, returns a dictionary.
+    decode : bool, default=True
+        If True, decodes the base64 payload.
+    format : str, default=None
+        The `format <https://learn.microsoft.com/rest/api/fabric/core/items/get-item-definition?tabs=HTTP#itemdefinition>`_ of the item definition.
+    """
+
+    workspace_id = resolve_workspace_id(workspace)
+    item_id = resolve_item_id(item=item, type=type, workspace=workspace_id)
+
+    item_type_url = utils.items.get(type)
+    if not item_type_url:
+        raise ValueError(f"{icons.red_dot} Invalid item type '{type}'.")
+
+    url = f"/v1/workspaces/{workspace_id}/{item_type_url}/{item_id}/getDefinition"
+    if format:
+        url += f"?format={format}"
+
+    result = _base_api(
+        request=url,
+        method="post",
+        status_codes=None,
+        lro_return_json=True,
+    )
+
+    if return_dataframe:
+        return pd.json_normalize(result["definition"]["parts"]).rename(
+            columns={
+                "path": "Path",
+                "payload": "Payload",
+                "payloadType": "Payload Type",
+            }
+        )
+
+    definition = {"definition": {"parts": []}}
+    if decode:
+        for part in result.get("definition", {}).get("parts", []):
+            path = part.get("path")
+            payload = part.get("payload")
+            decoded_payload = decode_payload(payload)
+
+            # Keep structure similar to original but replace payload with decoded version
+            definition["definition"]["parts"].append(
+                {"path": path, "payload": decoded_payload}
+            )
+    else:
+        return result
+
+
+@log
+def _get_item_definition(
     item: str | UUID,
     type: str,
     workspace: Optional[str | UUID] = None,
@@ -460,12 +559,11 @@ def get_item_definition(
     return_dataframe: bool = True,
     decode: bool = True,
 ):
-    from sempy_labs._utils import item_types
 
     workspace_id = resolve_workspace_id(workspace)
     item_id = resolve_item_id(item, type, workspace_id)
-    item_type_url = item_types.get(type)[1]
-    path = item_types.get(type)[2]
+    item_type_url = utils.item_types.get(type)[1]
+    path = utils.item_types.get(type)[2]
 
     url = f"/v1/workspaces/{workspace_id}/{item_type_url}/{item_id}/getDefinition"
     if format:
@@ -986,7 +1084,9 @@ def resolve_workspace_id(
 
 
 @log
-def resolve_workspace_name(workspace_id: Optional[UUID] = None) -> str:
+def resolve_workspace_name(
+    workspace_id: Optional[UUID] = None, throw_error: bool = True
+) -> str:
 
     if workspace_id is None:
         workspace_id = _get_fabric_context_setting(name="trident.workspace.id")
@@ -996,9 +1096,12 @@ def resolve_workspace_name(workspace_id: Optional[UUID] = None) -> str:
             request=f"/v1/workspaces/{workspace_id}", client="fabric_sp"
         ).json()
     except FabricHTTPException:
-        raise ValueError(
-            f"{icons.red_dot} The '{workspace_id}' workspace was not found."
-        )
+        if throw_error:
+            raise ValueError(
+                f"{icons.red_dot} The '{workspace_id}' workspace was not found."
+            )
+        else:
+            return workspace_id
 
     return response.get("displayName")
 
@@ -2090,17 +2193,21 @@ def _base_api(
     if (lro_return_json or lro_return_status_code) and status_codes is None:
         status_codes = [200, 202]
 
-    def get_token(audience="pbi"):
-        return notebookutils.credentials.getToken(audience)
+    class FabricDefaultCredential(TokenCredential):
+
+        def get_token(self, *scopes, **kwargs) -> AccessToken:
+            from sempy.fabric._credentials import build_access_token
+
+            return build_access_token(notebookutils.credentials.getToken("pbi"))
 
     if isinstance(status_codes, int):
         status_codes = [status_codes]
 
     if client == "fabric":
-        c = fabric.FabricRestClient(token_provider=get_token)
+        c = fabric.FabricRestClient(credential=FabricDefaultCredential())
     elif client == "fabric_sp":
-        token = auth.token_provider.get() or get_token
-        c = fabric.FabricRestClient(token_provider=token)
+        token = auth.token_provider.get() or FabricDefaultCredential()
+        c = fabric.FabricRestClient(credential=token)
     elif client in ["azure", "graph"]:
         pass
     else:

sempy_labs/_job_scheduler.py

@@ -1,7 +1,7 @@
 from sempy._utils._log import log
 import pandas as pd
 from typing import Optional, List
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_item_name_and_id,
     _update_dataframe_datatypes,
@@ -22,6 +22,8 @@ def list_item_job_instances(
 
     This is a wrapper function for the following API: `Job Scheduler - List Item Job Instances <https://learn.microsoft.com/rest/api/fabric/core/job-scheduler/list-item-job-instances>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     item : str | uuid.UUID
@@ -62,6 +64,7 @@ def list_item_job_instances(
     responses = _base_api(
         request=f"v1/workspaces/{workspace_id}/items/{item_id}/jobs/instances",
         uses_pagination=True,
+        client="fabric_sp",
     )
 
     if not responses[0].get("value"):
@@ -185,6 +188,9 @@ def list_item_schedules(
         "Times": "string",
         "Owner Id": "string",
         "Owner Type": "string",
+        "Recurrence": "int",
+        "Occurrence Type": "string",
+        "Occurrence Day of Month": "int",
     }
     df = _create_dataframe(columns=columns)
 
@@ -210,6 +216,11 @@ def list_item_schedules(
                 "Times": config.get("times"),
                 "Owner Id": own.get("id"),
                 "Owner Type": own.get("type"),
+                "Recurrence": config.get("recurrence"),
+                "Occurrence Type": config.get("occurence", {}).get("occurrenceType"),
+                "Occurrence Day of Month": config.get("occurrence", {}).get(
+                    "dayOfMonth"
+                ),
             }
         )
 

sempy_labs/_kql_databases.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     _base_api,
     _create_dataframe,
     delete_item,

sempy_labs/_kql_querysets.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import sempy_labs._icons as icons
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_id,
     _base_api,
     _create_dataframe,
@@ -19,6 +19,8 @@ def list_kql_querysets(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
 
     This is a wrapper function for the following API: `Items - List KQL Querysets <https://learn.microsoft.com/rest/api/fabric/kqlqueryset/items/list-kql-querysets>`_.
 
+    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
@@ -42,7 +44,9 @@ def list_kql_querysets(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     workspace_id = resolve_workspace_id(workspace)
 
     responses = _base_api(
-        request=f"v1/workspaces/{workspace_id}/kqlQuerysets", uses_pagination=True
+        request=f"v1/workspaces/{workspace_id}/kqlQuerysets",
+        uses_pagination=True,
+        client="fabric_sp",
     )
 
     rows = []
@@ -71,6 +75,8 @@ def create_kql_queryset(
 
     This is a wrapper function for the following API: `Items - Create KQL Queryset <https://learn.microsoft.com/rest/api/fabric/kqlqueryset/items/create-kql-queryset>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     name: str
@@ -97,6 +103,8 @@ def delete_kql_queryset(
 
     This is a wrapper function for the following API: `Items - Delete KQL Queryset <https://learn.microsoft.com/rest/api/fabric/kqlqueryset/items/delete-kql-queryset>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     kql_queryset: str | uuid.UUID

sempy_labs/_kusto.py

@@ -5,8 +5,8 @@ from sempy._utils._log import log
 import sempy_labs._icons as icons
 from typing import Optional
 from uuid import UUID
-from ._kql_databases import _resolve_cluster_uri
-from ._helper_functions import resolve_item_id
+from sempy_labs._kql_databases import _resolve_cluster_uri
+from sempy_labs._helper_functions import resolve_item_id
 
 
 @log

sempy_labs/_list_functions.py

@@ -1,5 +1,5 @@
 import sempy.fabric as fabric
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     create_relationship_name,
     format_dax_object_name,

sempy_labs/_managed_private_endpoints.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import sempy_labs._icons as icons
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _is_valid_uuid,
     _base_api,

sempy_labs/_mirrored_databases.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _update_dataframe_datatypes,
     _base_api,
@@ -8,7 +8,7 @@ from ._helper_functions import (
     _create_dataframe,
     delete_item,
     create_item,
-    get_item_definition,
+    _get_item_definition,
     resolve_workspace_id,
 )
 import sempy_labs._icons as icons
@@ -91,6 +91,8 @@ def create_mirrored_database(
 
     This is a wrapper function for the following API: `Items - Create Mirrored Database <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/items/create-mirrored-database>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     name: str
@@ -117,6 +119,8 @@ def delete_mirrored_database(
 
     This is a wrapper function for the following API: `Items - Delete Mirrored Database <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/items/delete-mirrored-database>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database: str
@@ -139,6 +143,8 @@ def get_mirroring_status(
 
     This is a wrapper function for the following API: `Mirroring - Get Mirroring Status <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/mirroring/get-mirroring-status>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database: str | uuid.UUID
@@ -154,13 +160,14 @@ def get_mirroring_status(
         The status of a mirrored database.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
     item_id = resolve_item_id(
         item=mirrored_database, type="MirroredDatabase", workspace=workspace
     )
     response = _base_api(
         request=f"/v1/workspaces/{workspace_id}/mirroredDatabases/{item_id}/getMirroringStatus",
         status_codes=200,
+        client="fabric_sp",
     )
 
     return response.json().get("status", {})
@@ -175,6 +182,8 @@ def get_tables_mirroring_status(
 
     This is a wrapper function for the following API: `Mirroring - Get Tables Mirroring Status <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/mirroring/get-tables-mirroring-status>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database: str | uuid.UUID
@@ -190,7 +199,7 @@ def get_tables_mirroring_status(
         A pandas dataframe showing the mirroring status of the tables.
     """
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    workspace_id = resolve_workspace_id(workspace)
     item_id = resolve_item_id(
         item=mirrored_database, type="MirroredDatabase", workspace=workspace
     )
@@ -199,6 +208,7 @@ def get_tables_mirroring_status(
         method="post",
         status_codes=200,
         uses_pagination=True,
+        client="fabric_sp",
     )
 
     columns = {
@@ -211,21 +221,24 @@ def get_tables_mirroring_status(
     }
     df = _create_dataframe(columns=columns)
 
+    rows = []
     for r in responses:
         for v in r.get("data", []):
             m = v.get("metrics", {})
-            new_data = {
-                "Source Schema Name": v.get("sourceSchemaName"),
-                "Source Table Name": v.get("sourceTableName"),
-                "Status": v.get("status"),
-                "Processed Bytes": m.get("processedBytes"),
-                "Processed Rows": m.get("processedRows"),
-                "Last Sync Date": m.get("lastSyncDateTime"),
-            }
-
-            df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+            rows.append(
+                {
+                    "Source Schema Name": v.get("sourceSchemaName"),
+                    "Source Table Name": v.get("sourceTableName"),
+                    "Status": v.get("status"),
+                    "Processed Bytes": m.get("processedBytes"),
+                    "Processed Rows": m.get("processedRows"),
+                    "Last Sync Date": m.get("lastSyncDateTime"),
+                }
+            )
 
-    _update_dataframe_datatypes(dataframe=df, column_map=columns)
+    if rows:
+        df = pd.DataFrame(rows, columns=list(columns.keys()))
+        _update_dataframe_datatypes(dataframe=df, column_map=columns)
 
     return df
 
@@ -239,6 +252,8 @@ def start_mirroring(
 
     This is a wrapper function for the following API: `Mirroring - Start Mirroring <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/mirroring/start-mirroring>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database: str | uuid.UUID
@@ -257,6 +272,7 @@ def start_mirroring(
         request=f"/v1/workspaces/{workspace_id}/mirroredDatabases/{item_id}/startMirroring",
         method="post",
         status_codes=200,
+        client="fabric_sp",
     )
 
     print(
@@ -273,6 +289,8 @@ def stop_mirroring(
 
     This is a wrapper function for the following API: `Mirroring - Stop Mirroring <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/mirroring/stop-mirroring>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database: str | uuid.UUID
@@ -291,6 +309,7 @@ def stop_mirroring(
         request=f"/v1/workspaces/{workspace_id}/mirroredDatabases/{item_id}/stopMirroring",
         method="post",
         status_codes=200,
+        client="fabric_sp",
     )
 
     print(
@@ -309,6 +328,8 @@ def get_mirrored_database_definition(
 
     This is a wrapper function for the following API: `Items - Get Mirrored Database Definition <https://learn.microsoft.com/rest/api/fabric/mirroreddatabase/items/get-mirrored-database-definition>`_.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database : str | uuid.UUID
@@ -327,7 +348,7 @@ def get_mirrored_database_definition(
         The mirrored database definition.
     """
 
-    return get_item_definition(
+    return _get_item_definition(
         item=mirrored_database,
         type="MirroredDatabase",
         workspace=workspace,
@@ -345,6 +366,8 @@ def update_mirrored_database_definition(
     """
     Updates an existing notebook with a new definition.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     mirrored_database : str | uuid.UUID
@@ -383,6 +406,7 @@ def update_mirrored_database_definition(
         json=request_body,
         status_codes=None,
         lro_return_status_code=True,
+        client="fabric_sp",
     )
 
     print(

sempy_labs/_mirrored_warehouses.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_id,
     _base_api,
     _create_dataframe,

sempy_labs/_ml_experiments.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from typing import Optional
-from ._helper_functions import (
+from sempy_labs._helper_functions import (
     resolve_workspace_id,
     _base_api,
     delete_item,