semantic-link-labs 0.9.5__py3-none-any.whl → 0.9.6__py3-none-any.whl

sempy_labs/_managed_private_endpoints.py

@@ -23,6 +23,8 @@ def create_managed_private_endpoint(
 
     This is a wrapper function for the following API: `Managed Private Endpoints - Create Workspace Managed Private Endpoint <https://learn.microsoft.com/rest/api/fabric/core/managed-private-endpoints/create-workspace-managed-private-endpoint>`.
 
+    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
@@ -60,6 +62,7 @@ def create_managed_private_endpoint(
         status_codes=[201, 202],
         payload=request_body,
         lro_return_status_code=True,
+        client="fabric_sp",
     )
     _print_success(
         item_name=name,
@@ -77,6 +80,8 @@ def list_managed_private_endpoints(
 
     This is a wrapper function for the following API: `Managed Private Endpoints - List Workspace Managed Private Endpoints <https://learn.microsoft.com/rest/api/fabric/core/managed-private-endpoints/list-workspace-managed-private-endpoints>`.
 
+    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
@@ -106,7 +111,7 @@ def list_managed_private_endpoints(
     responses = _base_api(
         request=f"/v1/workspaces/{workspace_id}/managedPrivateEndpoints",
         uses_pagination=True,
-        status_codes=200,
+        client="fabric_sp",
     )
 
     for r in responses:
@@ -134,6 +139,8 @@ def delete_managed_private_endpoint(
 
     This is a wrapper function for the following API: `Managed Private Endpoints - Delete Workspace Managed Private Endpoint <https://learn.microsoft.com/rest/api/fabric/core/managed-private-endpoints/delete-workspace-managed-private-endpoint>`.
 
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     managed_private_endpoint: str | uuid.UUID
@@ -162,7 +169,7 @@ def delete_managed_private_endpoint(
     _base_api(
         request=f"/v1/workspaces/{workspace_id}/managedPrivateEndpoints/{item_id}",
         method="delete",
-        status_codes=200,
+        client="fabric_sp",
     )
 
     _print_success(

sempy_labs/_mirrored_databases.py

@@ -22,6 +22,8 @@ def list_mirrored_databases(workspace: Optional[str | UUID] = None) -> pd.DataFr
 
     This is a wrapper function for the following API: `Items - List Mirrored Databases <https://learn.microsoft.com/rest/api/fabric/mirroredwarehouse/items/list-mirrored-databases>`_.
 
+    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
@@ -50,8 +52,8 @@ def list_mirrored_databases(workspace: Optional[str | UUID] = None) -> pd.DataFr
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     responses = _base_api(
         request=f"/v1/workspaces/{workspace_id}/mirroredDatabases",
-        status_codes=200,
         uses_pagination=True,
+        client="fabric_sp",
     )
 
     for r in responses:

sempy_labs/_ml_experiments.py

@@ -73,7 +73,7 @@ def create_ml_experiment(
     name: str
         Name of the ML experiment.
     description : str, default=None
-        A description of the environment.
+        A description of the ML experiment.
     workspace : str | uuid.UUID, default=None
         The Fabric workspace name or ID.
         Defaults to None which resolves to the workspace of the attached lakehouse

sempy_labs/_model_bpa.py

@@ -6,7 +6,6 @@ from IPython.display import display, HTML
 from sempy_labs._model_dependencies import get_model_calc_dependencies
 from sempy_labs._helper_functions import (
     format_dax_object_name,
-    resolve_lakehouse_name,
     create_relationship_name,
     save_as_delta_table,
     resolve_workspace_capacity,
@@ -389,13 +388,7 @@ def run_model_bpa(
         dfExport = finalDF.copy()
         delta_table_name = "modelbparesults"
 
-        lakehouse_id = fabric.get_lakehouse_id()
-        lake_workspace = fabric.get_workspace_id()
-        lakehouse = resolve_lakehouse_name(
-            lakehouse_id=lakehouse_id, workspace=lake_workspace
-        )
-
-        lakeT = get_lakehouse_tables(lakehouse=lakehouse, workspace=lake_workspace)
+        lakeT = get_lakehouse_tables()
         lakeT_filt = lakeT[lakeT["Table Name"] == delta_table_name]
 
         dfExport["Severity"].replace(icons.severity_mapping, inplace=True)
@@ -403,9 +396,7 @@ def run_model_bpa(
         if len(lakeT_filt) == 0:
             runId = 1
         else:
-            max_run_id = _get_column_aggregate(
-                lakehouse=lakehouse, table_name=delta_table_name
-            )
+            max_run_id = _get_column_aggregate(table_name=delta_table_name)
             runId = max_run_id + 1
 
         now = datetime.datetime.now()

sempy_labs/_model_bpa_bulk.py

@@ -7,6 +7,8 @@ from sempy_labs._helper_functions import (
     resolve_workspace_capacity,
     retry,
     _get_column_aggregate,
+    resolve_workspace_id,
+    resolve_lakehouse_name_and_id,
 )
 from sempy_labs.lakehouse import (
     get_lakehouse_tables,
@@ -16,6 +18,7 @@ from sempy_labs._model_bpa import run_model_bpa
 from typing import Optional, List
 from sempy._utils._log import log
 import sempy_labs._icons as icons
+from uuid import UUID
 
 
 @log
@@ -66,17 +69,12 @@ def run_model_bpa_bulk(
 
     now = datetime.datetime.now()
     output_table = "modelbparesults"
-    lakehouse_workspace = fabric.resolve_workspace_name()
-    lakehouse_id = fabric.get_lakehouse_id()
-    lakehouse = resolve_lakehouse_name(
-        lakehouse_id=lakehouse_id, workspace=lakehouse_workspace
-    )
-    lakeT = get_lakehouse_tables(lakehouse=lakehouse, workspace=lakehouse_workspace)
+    lakeT = get_lakehouse_tables()
     lakeT_filt = lakeT[lakeT["Table Name"] == output_table]
     if len(lakeT_filt) == 0:
         runId = 1
     else:
-        max_run_id = _get_column_aggregate(lakehouse=lakehouse, table_name=output_table)
+        max_run_id = _get_column_aggregate(table_name=output_table)
         runId = max_run_id + 1
 
     if isinstance(workspace, str):
@@ -170,7 +168,7 @@ def run_model_bpa_bulk(
 
                     # Append save results individually for each workspace (so as not to create a giant dataframe)
                     print(
-                        f"{icons.in_progress} Saving the Model BPA results of the '{wksp}' workspace to the '{output_table}' within the '{lakehouse}' lakehouse within the '{lakehouse_workspace}' workspace..."
+                        f"{icons.in_progress} Saving the Model BPA results of the '{wksp}' workspace to the '{output_table}' within the lakehouse attached to this notebook..."
                     )
 
                     schema = {
@@ -195,8 +193,8 @@ def run_model_bpa_bulk(
 @log
 def create_model_bpa_semantic_model(
     dataset: Optional[str] = icons.model_bpa_name,
-    lakehouse: Optional[str] = None,
-    lakehouse_workspace: Optional[str] = None,
+    lakehouse: Optional[str | UUID] = None,
+    lakehouse_workspace: Optional[str | UUID] = None,
 ):
     """
     Dynamically generates a Direct Lake semantic model based on the 'modelbparesults' delta table which contains the Best Practice Analyzer results.
@@ -209,16 +207,15 @@ def create_model_bpa_semantic_model(
     ----------
     dataset : str, default='ModelBPA'
         Name of the semantic model to be created.
-    lakehouse : str, default=None
+    lakehouse : str | uuid.UUID, default=None
         Name of the Fabric lakehouse which contains the 'modelbparesults' delta table.
         Defaults to None which resolves to the default lakehouse attached to the notebook.
-    lakehouse_workspace : str, default=None
+    lakehouse_workspace : str | uuid.UUID, default=None
         The workspace in which the lakehouse 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.
     """
 
-    from sempy_labs._helper_functions import resolve_lakehouse_name
     from sempy_labs.directlake import (
         generate_shared_expression,
         add_table_to_direct_lake_semantic_model,
@@ -226,22 +223,21 @@ def create_model_bpa_semantic_model(
     from sempy_labs import create_blank_semantic_model, refresh_semantic_model
     from sempy_labs.tom import connect_semantic_model
 
-    lakehouse_workspace = fabric.resolve_workspace_name(lakehouse_workspace)
-
-    if lakehouse is None:
-        lakehouse_id = fabric.get_lakehouse_id()
-        lakehouse = resolve_lakehouse_name(
-            lakehouse_id=lakehouse_id, workspace=lakehouse_workspace
-        )
+    lakehouse_workspace_id = resolve_workspace_id(workspace=lakehouse_workspace)
+    (lakehouse_id, lakehouse_name) = resolve_lakehouse_name_and_id(
+        lakehouse=lakehouse, workspace=lakehouse_workspace_id
+    )
 
     # Generate the shared expression based on the lakehouse and lakehouse workspace
     expr = generate_shared_expression(
-        item_name=lakehouse, item_type="Lakehouse", workspace=lakehouse_workspace
+        item_name=lakehouse_name,
+        item_type="Lakehouse",
+        workspace=lakehouse_workspace_id,
     )
 
     # Create blank model
     create_blank_semantic_model(
-        dataset=dataset, workspace=lakehouse_workspace, overwrite=True
+        dataset=dataset, workspace=lakehouse_workspace_id, overwrite=True
     )
 
     @retry(
@@ -250,7 +246,7 @@ def create_model_bpa_semantic_model(
     )
     def dyn_connect():
         with connect_semantic_model(
-            dataset=dataset, readonly=True, workspace=lakehouse_workspace
+            dataset=dataset, readonly=True, workspace=lakehouse_workspace_id
         ) as tom:
 
             tom.model
@@ -259,7 +255,7 @@ def create_model_bpa_semantic_model(
     icons.sll_tags.append("ModelBPABulk")
     table_exists = False
     with connect_semantic_model(
-        dataset=dataset, readonly=False, workspace=lakehouse_workspace
+        dataset=dataset, readonly=False, workspace=lakehouse_workspace_id
     ) as tom:
         t_name = "BPAResults"
         t_name_full = f"'{t_name}'"
@@ -274,11 +270,11 @@ def create_model_bpa_semantic_model(
             dataset=dataset,
             table_name=t_name,
             lakehouse_table_name="modelbparesults",
-            workspace=lakehouse_workspace,
+            workspace=lakehouse_workspace_id,
             refresh=False,
         )
     with connect_semantic_model(
-        dataset=dataset, readonly=False, workspace=lakehouse_workspace
+        dataset=dataset, readonly=False, workspace=lakehouse_workspace_id
     ) as tom:
         # Fix column names
         for c in tom.all_columns():
@@ -377,4 +373,4 @@ def create_model_bpa_semantic_model(
         # tom.add_measure(table_name=t_name, measure_name='Rules Followed', expression="[Rules] - [Rules Violated]")
 
     # Refresh the model
-    refresh_semantic_model(dataset=dataset, workspace=lakehouse_workspace)
+    refresh_semantic_model(dataset=dataset, workspace=lakehouse_workspace_id)

sempy_labs/_one_lake_integration.py

@@ -5,6 +5,7 @@ from sempy._utils._log import log
 from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_dataset_name_and_id,
+    resolve_workspace_id,
 )
 import sempy_labs._icons as icons
 from uuid import UUID
@@ -43,7 +44,7 @@ def export_model_to_onelake(
         destination_workspace = workspace_name
         destination_workspace_id = workspace_id
     else:
-        destination_workspace_id = fabric.resolve_workspace_id(destination_workspace)
+        destination_workspace_id = resolve_workspace_id(workspace=destination_workspace)
 
     tmsl = f"""
     {{

sempy_labs/_semantic_models.py

@@ -7,6 +7,7 @@ from sempy_labs._helper_functions import (
     _update_dataframe_datatypes,
     resolve_workspace_name_and_id,
     resolve_dataset_name_and_id,
+    delete_item,
 )
 import sempy_labs._icons as icons
 
@@ -115,3 +116,22 @@ def enable_semantic_model_scheduled_refresh(
         print(
             f"{icons.green_dot} Scheduled refresh for the '{dataset_name}' within the '{workspace_name}' workspace has been enabled."
         )
+
+
+def delete_semantic_model(dataset: str | UUID, workspace: Optional[str | UUID] = None):
+    """
+    Deletes a semantic model.
+
+    This is a wrapper function for the following API: `Items - Delete Semantic Model <https://learn.microsoft.com/rest/api/fabric/semanticmodel/items/delete-semantic-model>`_.
+
+    Parameters
+    ----------
+    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.
+    """
+
+    delete_item(item=dataset, type="SemanticModel", workspace=workspace)

sempy_labs/_sql.py

@@ -60,14 +60,18 @@ class ConnectBase:
                 item=item, workspace=workspace_id, type=endpoint_type.capitalize()
             )
 
+        endpoint_for_url = (
+            "sqlDatabases" if endpoint_type == "sqldatabase" else f"{endpoint_type}s"
+        )
+
         # Get the TDS endpoint
         response = _base_api(
-            request=f"v1/workspaces/{workspace_id}/{endpoint_type}s/{resource_id}"
+            request=f"v1/workspaces/{workspace_id}/{endpoint_for_url}/{resource_id}"
         )
 
         if endpoint_type == "warehouse":
             tds_endpoint = response.json().get("properties", {}).get("connectionString")
-        if endpoint_type == "sqldatabase":
+        elif endpoint_type == "sqldatabase":
             tds_endpoint = response.json().get("properties", {}).get("serverFqdn")
         else:
             tds_endpoint = (

sempy_labs/_sqldatabase.py

@@ -1,70 +1,69 @@
-import sempy.fabric as fabric
 from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     _base_api,
     _create_dataframe,
     _update_dataframe_datatypes,
+    create_item,
+    delete_item,
 )
 import pandas as pd
 from typing import Optional
-import sempy_labs._icons as icons
 from uuid import UUID
 
-## Still debugging the creation of an sql database
-# def create_warehouse(
-#     warehouse: str,
-#     description: Optional[str] = None,
-#     case_insensitive_collation: bool = False,
-#     workspace: Optional[str | UUID] = None,
-# ):
-#     """
-#     Creates a Fabric warehouse.
-
-#     This is a wrapper function for the following API: `Items - Create Warehouse <https://learn.microsoft.com/rest/api/fabric/warehouse/items/create-warehouse>`_.
-
-#     Parameters
-#     ----------
-#     warehouse: str
-#         Name of the warehouse.
-#     description : str, default=None
-#         A description of the warehouse.
-#     case_insensitive_collation: bool, default=False
-#         If True, creates the warehouse with case-insensitive collation.
-#     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.
-#     """
-
-#     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-#     payload = {"displayName": warehouse}
-
-#     if description:
-#         payload["description"] = description
-#     if case_insensitive_collation:
-#         payload.setdefault("creationPayload", {})
-#         payload["creationPayload"][
-#             "defaultCollation"
-#         ] = "Latin1_General_100_CI_AS_KS_WS_SC_UTF8"
-
-#     _base_api(
-#         request=f"/v1/workspaces/{workspace_id}/warehouses",
-#         payload=payload,
-#         method="post",
-#         lro_return_status_code=True,
-#         status_codes=[201, 202],
-#     )
-
-#     print(
-#         f"{icons.green_dot} The '{warehouse}' warehouse has been created within the '{workspace_name}' workspace."
-#     )
-
-
-def _list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
+
+def create_sql_database(
+    name: str, description: Optional[str] = None, workspace: Optional[str | UUID] = None
+):
+    """
+    Creates a SQL database.
+
+    This is a wrapper function for the following API: `Items - Create SQL Database <https://learn.microsoft.com/rest/api/fabric/sqldatabase/items/create-sql-database>`_.
+
+    Parameters
+    ----------
+    name: str
+        Name of the SQL database.
+    description : str, default=None
+        A description of the SQL database.
+    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.
+    """
+
+    create_item(
+        name=name, description=description, type="SQLDatabase", workspace=workspace
+    )
+
+
+def delete_sql_database(
+    sql_database: str | UUID, workspace: Optional[str | UUID] = None
+):
+    """
+    Deletes a SQL Database.
+
+    This is a wrapper function for the following API: `Items - Delete SQL Database <https://learn.microsoft.com/rest/api/fabric/sqldatabase/items/delete-sql-database>`_.
+
+    Parameters
+    ----------
+    sql_database: str | uuid.UUID
+        Name of the SQL database.
+    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.
     """
-    Shows the databses within a workspace.
 
+    delete_item(item=sql_database, type="SQLDatabase", workspace=workspace)
+
+
+def list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
+    """
+    Lists all SQL databases in the Fabric workspace.
+
+    This is a wrapper function for the following API: `Items - List SQL Databases <https://learn.microsoft.com/rest/api/fabric/sqldatabase/items/list-sql-databases>`_.
+
+    Service Principal Authentication is supported (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
 
     Parameters
     ----------
@@ -76,45 +75,39 @@ def _list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     Returns
     -------
     pandas.DataFrame
-        A pandas dataframe showing the SQLDabatases within a workspace.
+        A pandas dataframe showing a list of SQL databases in the Fabric workspace.
     """
 
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+
     columns = {
         "SQL Database Name": "string",
         "SQL Database Id": "string",
         "Description": "string",
-        "Connection Type": "string",
         "Connection Info": "string",
         "Database Name": "string",
         "Server FQDN": "string",
-        "Provisioning Status": "string",
-        "Created Date": "datetime",
-        "Last Updated Time UTC": "datetime",
     }
     df = _create_dataframe(columns=columns)
 
-    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-
     responses = _base_api(
-        request=f"/v1/workspaces/{workspace_id}/sqldatabases", uses_pagination=True
+        request=f"/v1/workspaces/{workspace_id}/SQLDatabases",
+        uses_pagination=True,
+        client="fabric_sp",
     )
 
     for r in responses:
         for v in r.get("value", []):
             prop = v.get("properties", {})
-
             new_data = {
                 "SQL Database Name": v.get("displayName"),
                 "SQL Database Id": v.get("id"),
                 "Description": v.get("description"),
-                "Connection Type": v.get("type"),
                 "Connection Info": prop.get("connectionInfo"),
                 "Database Name": prop.get("databaseName"),
                 "Server FQDN": prop.get("serverFqdn"),
-                "Provisioning Status": prop.get("provisioningState"),
-                "Created Date": prop.get("createdDate"),
-                "Last Updated Time UTC": prop.get("lastUpdatedTimeUtc"),
             }
+
             df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
     _update_dataframe_datatypes(dataframe=df, column_map=columns)
@@ -122,38 +115,6 @@ def _list_sql_databases(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     return df
 
 
-## Still debugging the deletion of an sql database
-# def delete_warehouse(name: str, workspace: Optional[str | UUID] = None):
-#     """
-#     Deletes a Fabric warehouse.
-
-#     This is a wrapper function for the following API: `Items - Delete Warehouse <https://learn.microsoft.com/rest/api/fabric/warehouse/items/delete-warehouse>`_.
-
-#     Parameters
-#     ----------
-#     name: str
-#         Name of the warehouse.
-#     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.
-#     """
-
-#     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-#     item_id = fabric.resolve_item_id(
-#         item_name=name, type="Warehouse", workspace=workspace_id
-#     )
-
-#     _base_api(
-#         request=f"/v1/workspaces/{workspace_id}/warehouses/{item_id}", method="delete"
-#     )
-
-#     print(
-#         f"{icons.green_dot} The '{name}' warehouse within the '{workspace_name}' workspace has been deleted."
-#     )
-
-
 def get_sql_database_tables(
     sql_database: str | UUID, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:

sempy_labs/_vertipaq.py

@@ -15,6 +15,8 @@ from sempy_labs._helper_functions import (
     resolve_workspace_name_and_id,
     resolve_dataset_name_and_id,
     _create_spark_session,
+    resolve_workspace_id,
+    resolve_workspace_name,
 )
 from sempy_labs._list_functions import list_relationships, list_tables
 from sempy_labs.lakehouse import lakehouse_attached, get_lakehouse_tables
@@ -196,8 +198,10 @@ def vertipaq_analyzer(
                 & (~dfC["Column Name"].str.startswith("RowNumber-"))
             ]
 
-            object_workspace = fabric.resolve_workspace_name(lakehouse_workspace_id)
-            current_workspace_id = fabric.get_workspace_id()
+            object_workspace = resolve_workspace_name(
+                workspace_id=lakehouse_workspace_id
+            )
+            current_workspace_id = resolve_workspace_id()
             if current_workspace_id != lakehouse_workspace_id:
                 lakeTables = get_lakehouse_tables(
                     lakehouse=lakehouse_name, workspace=object_workspace
@@ -526,22 +530,15 @@ def vertipaq_analyzer(
             )
 
     if export == "table":
-        lakehouse_id = fabric.get_lakehouse_id()
-        lake_workspace = fabric.resolve_workspace_name()
-        lakehouse = resolve_lakehouse_name(
-            lakehouse_id=lakehouse_id, workspace=lake_workspace
-        )
         lakeTName = "vertipaqanalyzer_model"
 
-        lakeT = get_lakehouse_tables(lakehouse=lakehouse, workspace=lake_workspace)
+        lakeT = get_lakehouse_tables()
         lakeT_filt = lakeT[lakeT["Table Name"] == lakeTName]
 
         if len(lakeT_filt) == 0:
             runId = 1
         else:
-            max_run_id = _get_column_aggregate(
-                lakehouse=lakehouse, table_name=lakeTName
-            )
+            max_run_id = _get_column_aggregate(table_name=lakeTName)
             runId = max_run_id + 1
 
         dfMap = {

sempy_labs/_warehouses.py

@@ -16,7 +16,7 @@ def create_warehouse(
     description: Optional[str] = None,
     case_insensitive_collation: bool = False,
     workspace: Optional[str | UUID] = None,
-):
+) -> UUID:
     """
     Creates a Fabric warehouse.
 
@@ -34,6 +34,11 @@ def create_warehouse(
         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.
+
+    Returns
+    -------
+    uuid.UUID
+        The ID of the created warehouse.
     """
 
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
@@ -48,7 +53,7 @@ def create_warehouse(
             "defaultCollation"
         ] = "Latin1_General_100_CI_AS_KS_WS_SC_UTF8"
 
-    _base_api(
+    response = _base_api(
         request=f"/v1/workspaces/{workspace_id}/warehouses",
         payload=payload,
         method="post",
@@ -60,6 +65,8 @@ def create_warehouse(
         f"{icons.green_dot} The '{warehouse}' warehouse has been created within the '{workspace_name}' workspace."
     )
 
+    return response.get("id")
+
 
 def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
@@ -67,6 +74,8 @@ def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
 
     This is a wrapper function for the following API: `Items - List Warehouses <https://learn.microsoft.com/rest/api/fabric/warehouse/items/list-warehouses>`_.
 
+    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
@@ -93,7 +102,9 @@ def list_warehouses(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     responses = _base_api(
-        request=f"/v1/workspaces/{workspace_id}/warehouses", uses_pagination=True
+        request=f"/v1/workspaces/{workspace_id}/warehouses",
+        uses_pagination=True,
+        client="fabric_sp",
     )
 
     for r in responses: