semantic-link-labs 0.8.9__py3-none-any.whl → 0.8.11__py3-none-any.whl

sempy_labs/_helper_functions.py

@@ -102,7 +102,7 @@ def create_relationship_name(
     )
 
 
-def resolve_report_id(report: str, workspace: Optional[str] = None) -> UUID:
+def resolve_report_id(report: str, workspace: Optional[str | UUID] = None) -> UUID:
     """
     Obtains the ID of the Power BI report.
 
@@ -110,8 +110,8 @@ def resolve_report_id(report: str, workspace: Optional[str] = None) -> UUID:
     ----------
     report : str
         The name of the Power BI report.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -121,25 +121,19 @@ def resolve_report_id(report: str, workspace: Optional[str] = None) -> UUID:
         The ID of the Power BI report.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
-    obj = fabric.resolve_item_id(item_name=report, type="Report", workspace=workspace)
-
-    return obj
+    return fabric.resolve_item_id(item_name=report, type="Report", workspace=workspace)
 
 
-def resolve_report_name(report_id: UUID, workspace: Optional[str] = None) -> str:
+def resolve_report_name(report_id: UUID, workspace: Optional[str | UUID] = None) -> str:
     """
     Obtains the name of the Power BI report.
 
     Parameters
     ----------
-    report_id : UUID
+    report_id : uuid.UUID
         The name of the Power BI report.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -149,26 +143,66 @@ def resolve_report_name(report_id: UUID, workspace: Optional[str] = None) -> str
         The name of the Power BI report.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
-    obj = fabric.resolve_item_name(
+    return fabric.resolve_item_name(
         item_id=report_id, type="Report", workspace=workspace
     )
 
-    return obj
+
+def resolve_item_name_and_id(
+    item: str | UUID, type: Optional[str] = None, workspace: Optional[str | UUID] = None
+) -> Tuple[str, UUID]:
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    if _is_valid_uuid(item):
+        item_id = item
+        item_name = fabric.resolve_item_name(
+            item_id=item_id, type=type, workspace=workspace_id
+        )
+    else:
+        if type is None:
+            raise ValueError(
+                f"{icons.warning} Must specify a 'type' if specifying a name as the 'item'."
+            )
+        item_name = item
+        item_id = fabric.resolve_item_id(
+            item_name=item, type=type, workspace=workspace_id
+        )
+
+    return item_name, item_id
+
+
+def resolve_dataset_name_and_id(
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
+) -> Tuple[str, UUID]:
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    if _is_valid_uuid(dataset):
+        dataset_id = dataset
+        dataset_name = fabric.resolve_item_name(
+            item_id=dataset_id, type="SemanticModel", workspace=workspace_id
+        )
+    else:
+        dataset_name = dataset
+        dataset_id = fabric.resolve_item_id(
+            item_name=dataset, type="SemanticModel", workspace=workspace_id
+        )
+
+    return dataset_name, dataset_id
 
 
-def resolve_dataset_id(dataset: str, workspace: Optional[str] = None) -> UUID:
+def resolve_dataset_id(
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
+) -> UUID:
     """
     Obtains the ID of the semantic model.
 
     Parameters
     ----------
-    dataset : str
-        The name of the semantic model.
-    workspace : str, default=None
+    dataset : str | uuid.UUID
+        The name or ID of the semantic model.
+    workspace : str | uuid.UUID, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
@@ -179,26 +213,27 @@ def resolve_dataset_id(dataset: str, workspace: Optional[str] = None) -> UUID:
         The ID of the semantic model.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
-    obj = fabric.resolve_item_id(
-        item_name=dataset, type="SemanticModel", workspace=workspace
-    )
+    if _is_valid_uuid(dataset):
+        dataset_id = dataset
+    else:
+        dataset_id = fabric.resolve_item_id(
+            item_name=dataset, type="SemanticModel", workspace=workspace
+        )
 
-    return obj
+    return dataset_id
 
 
-def resolve_dataset_name(dataset_id: UUID, workspace: Optional[str] = None) -> str:
+def resolve_dataset_name(
+    dataset_id: UUID, workspace: Optional[str | UUID] = None
+) -> str:
     """
     Obtains the name of the semantic model.
 
     Parameters
     ----------
-    dataset_id : UUID
+    dataset_id : uuid.UUID
         The name of the semantic model.
-    workspace : str, default=None
+    workspace : str | uuid.UUID, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
@@ -209,30 +244,24 @@ def resolve_dataset_name(dataset_id: UUID, workspace: Optional[str] = None) -> s
         The name of the semantic model.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
-    obj = fabric.resolve_item_name(
+    return fabric.resolve_item_name(
         item_id=dataset_id, type="SemanticModel", workspace=workspace
     )
 
-    return obj
-
 
 def resolve_lakehouse_name(
-    lakehouse_id: Optional[UUID] = None, workspace: Optional[str] = None
+    lakehouse_id: Optional[UUID] = None, workspace: Optional[str | UUID] = None
 ) -> str:
     """
     Obtains the name of the Fabric lakehouse.
 
     Parameters
     ----------
-    lakehouse_id : UUID, default=None
+    lakehouse_id : uuid.UUID, default=None
         The name of the Fabric lakehouse.
         Defaults to None which resolves to the lakehouse attached to the notebook.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -242,21 +271,17 @@ def resolve_lakehouse_name(
         The name of the Fabric lakehouse.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
     if lakehouse_id is None:
         lakehouse_id = fabric.get_lakehouse_id()
 
-    obj = fabric.resolve_item_name(
+    return fabric.resolve_item_name(
         item_id=lakehouse_id, type="Lakehouse", workspace=workspace
     )
 
-    return obj
-
 
-def resolve_lakehouse_id(lakehouse: str, workspace: Optional[str] = None) -> UUID:
+def resolve_lakehouse_id(
+    lakehouse: str, workspace: Optional[str | UUID] = None
+) -> UUID:
     """
     Obtains the ID of the Fabric lakehouse.
 
@@ -264,38 +289,34 @@ def resolve_lakehouse_id(lakehouse: str, workspace: Optional[str] = None) -> UUI
     ----------
     lakehouse : str
         The name of the Fabric lakehouse.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
     Returns
     -------
-    UUID
+    uuid.UUID
         The ID of the Fabric lakehouse.
     """
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-
-    obj = fabric.resolve_item_id(
+    return fabric.resolve_item_id(
         item_name=lakehouse, type="Lakehouse", workspace=workspace
     )
 
-    return obj
-
 
-def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None) -> UUID:
+def get_direct_lake_sql_endpoint(
+    dataset: str | UUID, workspace: Optional[str | UUID] = None
+) -> UUID:
     """
     Obtains the SQL Endpoint ID of the semantic model.
 
     Parameters
     ----------
-    dataset : str
-        The name of the semantic model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    dataset : str | uuid.UUID
+        The 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.
 
@@ -307,9 +328,8 @@ def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None)
 
     from sempy_labs.tom import connect_semantic_model
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
 
     # dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
     # dfP_filt = dfP[dfP["Mode"] == "DirectLake"]
@@ -320,7 +340,7 @@ def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None)
     #    )
 
     with connect_semantic_model(
-        dataset=dataset, readonly=True, workspace=workspace
+        dataset=dataset_id, readonly=True, workspace=workspace_id
     ) as tom:
         sqlEndpointId = None
         for e in tom.model.Expressions:
@@ -330,7 +350,7 @@ def get_direct_lake_sql_endpoint(dataset: str, workspace: Optional[str] = None)
                 sqlEndpointId = matches[1]
 
         if sqlEndpointId is None:
-            raise ValueError("SQL Endpoint not found.")
+            raise ValueError(f"{icons.red_dot} SQL Endpoint not found.")
 
         return sqlEndpointId
 
@@ -407,7 +427,7 @@ def save_as_delta_table(
     merge_schema: bool = False,
     schema: Optional[dict] = None,
     lakehouse: Optional[str] = None,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     Saves a pandas dataframe as a delta table in a Fabric lakehouse.
@@ -427,8 +447,8 @@ def save_as_delta_table(
     lakehouse : str, default=None
         The Fabric lakehouse used by the Direct Lake semantic model.
         Defaults to None which resolves to the lakehouse attached to the notebook.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
@@ -447,19 +467,15 @@ def save_as_delta_table(
         TimestampType,
     )
 
-    if workspace is None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
-    else:
-        workspace_id = fabric.resolve_workspace_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
         lakehouse = resolve_lakehouse_name(
-            lakehouse_id=lakehouse_id, workspace=workspace
+            lakehouse_id=lakehouse_id, workspace=workspace_id
         )
     else:
-        lakehouse_id = resolve_lakehouse_id(lakehouse, workspace)
+        lakehouse_id = resolve_lakehouse_id(lakehouse, workspace_id)
 
     writeModes = ["append", "overwrite"]
     write_mode = write_mode.lower()
@@ -515,7 +531,7 @@ def save_as_delta_table(
     else:
         spark_df.write.mode(write_mode).format("delta").save(filePath)
     print(
-        f"{icons.green_dot} The dataframe has been saved as the '{delta_table_name}' table in the '{lakehouse}' lakehouse within the '{workspace}' workspace."
+        f"{icons.green_dot} The dataframe has been saved as the '{delta_table_name}' table in the '{lakehouse}' lakehouse within the '{workspace_name}' workspace."
     )
 
 
@@ -555,14 +571,16 @@ def language_validate(language: str):
     return lang
 
 
-def resolve_workspace_name_and_id(workspace: Optional[str] = None) -> Tuple[str, str]:
+def resolve_workspace_name_and_id(
+    workspace: Optional[str | UUID] = None,
+) -> Tuple[str, str]:
     """
     Obtains the name and ID of the Fabric workspace.
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -574,11 +592,15 @@ def resolve_workspace_name_and_id(workspace: Optional[str] = None) -> Tuple[str,
 
     if workspace is None:
         workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+        workspace_name = fabric.resolve_workspace_name(workspace_id)
+    elif _is_valid_uuid(workspace):
+        workspace_id = workspace
+        workspace_name = fabric.resolve_workspace_name(workspace_id)
     else:
-        workspace_id = fabric.resolve_workspace_id(workspace)
+        workspace_name = workspace
+        workspace_id = fabric.resolve_workspace_id(workspace_name)
 
-    return str(workspace), str(workspace_id)
+    return str(workspace_name), str(workspace_id)
 
 
 def _extract_json(dataframe: pd.DataFrame) -> dict:
@@ -604,7 +626,9 @@ def _decode_b64(file, format: Optional[str] = "utf-8"):
     return result
 
 
-def is_default_semantic_model(dataset: str, workspace: Optional[str] = None) -> bool:
+def is_default_semantic_model(
+    dataset: str, workspace: Optional[str | UUID] = None
+) -> bool:
     """
     Identifies whether a semantic model is a default semantic model.
 
@@ -612,8 +636,8 @@ def is_default_semantic_model(dataset: str, workspace: Optional[str] = None) ->
     ----------
     dataset : str
         The name of the semantic model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -623,9 +647,9 @@ def is_default_semantic_model(dataset: str, workspace: Optional[str] = None) ->
         A True/False value indicating whether the semantic model is a default semantic model.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    dfI = fabric.list_items(workspace=workspace)
+    dfI = fabric.list_items(workspace=workspace_id)
     filtered_df = dfI.groupby("Display Name").filter(
         lambda x: set(["Warehouse", "SemanticModel"]).issubset(set(x["Type"]))
         or set(["Lakehouse", "SemanticModel"]).issubset(set(x["Type"]))
@@ -635,16 +659,16 @@ def is_default_semantic_model(dataset: str, workspace: Optional[str] = None) ->
     return dataset in default_semantic_models
 
 
-def resolve_item_type(item_id: UUID, workspace: Optional[str] = None) -> str:
+def resolve_item_type(item_id: UUID, workspace: Optional[str | UUID] = None) -> str:
     """
     Obtains the item type for a given Fabric Item Id within a Fabric workspace.
 
     Parameters
     ----------
-    item_id : UUID
+    item_id : uuid.UUID
         The item/artifact Id.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -654,21 +678,19 @@ def resolve_item_type(item_id: UUID, workspace: Optional[str] = None) -> str:
         The item type for the item Id.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
-    dfI = fabric.list_items(workspace=workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    dfI = fabric.list_items(workspace=workspace_id)
     dfI_filt = dfI[dfI["Id"] == item_id]
 
-    if len(dfI_filt) == 0:
+    if dfI_filt.empty:
         raise ValueError(
-            f"Invalid 'item_id' parameter. The '{item_id}' item was not found in the '{workspace}' workspace."
+            f"Invalid 'item_id' parameter. The '{item_id}' item was not found in the '{workspace_name}' workspace."
         )
-    item_type = dfI_filt["Type"].iloc[0]
-
-    return item_type
+    return dfI_filt["Type"].iloc[0]
 
 
 def resolve_dataset_from_report(
-    report: str, workspace: Optional[str] = None
+    report: str, workspace: Optional[str | UUID] = None
 ) -> Tuple[UUID, str, UUID, str]:
     """
     Obtains the basic semantic model properties from which the report's data is sourced.
@@ -677,8 +699,8 @@ def resolve_dataset_from_report(
     ----------
     report : str
         The name of the Power BI report.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -688,13 +710,13 @@ def resolve_dataset_from_report(
         The semantic model UUID, semantic model name, semantic model workspace UUID, semantic model workspace name
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
 
-    dfR = fabric.list_reports(workspace=workspace)
+    dfR = fabric.list_reports(workspace=workspace_id)
     dfR_filt = dfR[dfR["Name"] == report]
     if len(dfR_filt) == 0:
         raise ValueError(
-            f"{icons.red_dot} The '{report}' report does not exist within the '{workspace}' workspace."
+            f"{icons.red_dot} The '{report}' report does not exist within the '{workspace_name}' workspace."
         )
     dataset_id = dfR_filt["Dataset Id"].iloc[0]
     dataset_workspace_id = dfR_filt["Dataset Workspace Id"].iloc[0]
@@ -713,14 +735,16 @@ def _add_part(target_dict, path, payload):
     target_dict["definition"]["parts"].append(part)
 
 
-def resolve_workspace_capacity(workspace: Optional[str] = None) -> Tuple[UUID, str]:
+def resolve_workspace_capacity(
+    workspace: Optional[str | UUID] = None,
+) -> Tuple[UUID, str]:
     """
     Obtains the capacity Id and capacity name for a given workspace.
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or UUID.
         Defaults to None which resolves to the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the workspace of the notebook.
 
@@ -730,9 +754,9 @@ def resolve_workspace_capacity(workspace: Optional[str] = None) -> Tuple[UUID, s
         capacity Id; capacity came.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
-    filter_condition = urllib.parse.quote(workspace)
-    dfW = fabric.list_workspaces(filter=f"name eq '{filter_condition}'")
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    filter_condition = urllib.parse.quote(workspace_id)
+    dfW = fabric.list_workspaces(filter=f"id eq '{filter_condition}'")
     capacity_id = dfW["Capacity Id"].iloc[0]
     dfC = fabric.list_capacities()
     dfC_filt = dfC[dfC["Id"] == capacity_id]
@@ -744,14 +768,14 @@ def resolve_workspace_capacity(workspace: Optional[str] = None) -> Tuple[UUID, s
     return capacity_id, capacity_name
 
 
-def get_capacity_id(workspace: Optional[str] = None) -> UUID:
+def get_capacity_id(workspace: Optional[str | UUID] = None) -> UUID:
     """
     Obtains the Capacity Id for a given workspace.
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -761,23 +785,28 @@ def get_capacity_id(workspace: Optional[str] = None) -> UUID:
         The capacity Id.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
-    filter_condition = urllib.parse.quote(workspace)
-    dfW = fabric.list_workspaces(filter=f"name eq '{filter_condition}'")
-    if len(dfW) == 0:
-        raise ValueError(f"{icons.red_dot} The '{workspace}' does not exist'.")
+    if workspace is None:
+        capacity_id = _get_fabric_context_setting(name="trident.capacity.id")
+    else:
+        (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+        filter_condition = urllib.parse.quote(workspace_id)
+        dfW = fabric.list_workspaces(filter=f"id eq '{filter_condition}'")
+        if len(dfW) == 0:
+            raise ValueError(f"{icons.red_dot} The '{workspace_name}' does not exist'.")
+
+        capacity_id = dfW["Capacity Id"].iloc[0]
 
-    return dfW["Capacity Id"].iloc[0]
+    return capacity_id
 
 
-def get_capacity_name(workspace: Optional[str] = None) -> str:
+def get_capacity_name(workspace: Optional[str | UUID] = None) -> str:
     """
     Obtains the capacity name for a given workspace.
 
     Parameters
     ----------
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
 
@@ -804,7 +833,7 @@ def resolve_capacity_name(capacity_id: Optional[UUID] = None) -> str:
 
     Parameters
     ----------
-    capacity_id : UUID, default=None
+    capacity_id : uuid.UUID, default=None
         The capacity Id.
         Defaults to None which resolves to the capacity name of the workspace of the attached lakehouse
         or if no lakehouse attached, resolves to the capacity name of the workspace of the notebook.
@@ -995,7 +1024,7 @@ def _get_adls_client(account_name):
     return service_client
 
 
-def resolve_warehouse_id(warehouse: str, workspace: Optional[str]) -> UUID:
+def resolve_warehouse_id(warehouse: str, workspace: Optional[str | UUID]) -> UUID:
     """
     Obtains the Id for a given warehouse.
 
@@ -1003,6 +1032,10 @@ def resolve_warehouse_id(warehouse: str, workspace: Optional[str]) -> UUID:
     ----------
     warehouse : str
         The warehouse name
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID in which the semantic model resides.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
 
     Returns
     -------
@@ -1010,7 +1043,6 @@ def resolve_warehouse_id(warehouse: str, workspace: Optional[str]) -> UUID:
         The warehouse Id.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
     return fabric.resolve_item_id(
         item_name=warehouse, type="Warehouse", workspace=workspace
     )
@@ -1072,7 +1104,9 @@ def convert_to_alphanumeric_lowercase(input_string):
     return cleaned_string
 
 
-def resolve_environment_id(environment: str, workspace: Optional[str] = None) -> UUID:
+def resolve_environment_id(
+    environment: str, workspace: Optional[str | UUID] = None
+) -> UUID:
     """
     Obtains the environment Id for a given environment.
 
@@ -1080,6 +1114,10 @@ def resolve_environment_id(environment: str, workspace: Optional[str] = None) ->
     ----------
     environment: str
         Name of the environment.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID in which the semantic model resides.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
 
     Returns
     -------
@@ -1087,7 +1125,6 @@ def resolve_environment_id(environment: str, workspace: Optional[str] = None) ->
         The environment Id.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
     return fabric.resolve_item_id(
         item_name=environment, type="Environment", workspace=workspace
     )
@@ -1122,7 +1159,7 @@ def convert_to_friendly_case(text: str) -> str:
     return text
 
 
-def resolve_notebook_id(notebook: str, workspace: Optional[str] = None) -> UUID:
+def resolve_notebook_id(notebook: str, workspace: Optional[str | UUID] = None) -> UUID:
     """
     Obtains the notebook Id for a given notebook.
 
@@ -1130,6 +1167,10 @@ def resolve_notebook_id(notebook: str, workspace: Optional[str] = None) -> UUID:
     ----------
     notebook: str
         Name of the notebook.
+    workspace : str | uuid.UUID, default=None
+        The Fabric workspace name or ID in which the semantic model resides.
+        Defaults to None which resolves to the workspace of the attached lakehouse
+        or if no lakehouse attached, resolves to the workspace of the notebook.
 
     Returns
     -------
@@ -1137,7 +1178,6 @@ def resolve_notebook_id(notebook: str, workspace: Optional[str] = None) -> UUID:
         The notebook Id.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
     return fabric.resolve_item_id(
         item_name=notebook, type="Notebook", workspace=workspace
     )
@@ -1165,22 +1205,24 @@ def _make_list_unique(my_list):
     return list(set(my_list))
 
 
-def _get_partition_map(dataset: str, workspace: Optional[str] = None) -> pd.DataFrame:
+def _get_partition_map(
+    dataset: str, workspace: Optional[str | UUID] = None
+) -> pd.DataFrame:
 
-    if workspace is None:
-        workspace = fabric.resolve_workspace_name()
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
 
     partitions = fabric.evaluate_dax(
-        dataset=dataset,
-        workspace=workspace,
+        dataset=dataset_id,
+        workspace=workspace_id,
         dax_string="""
     select [ID] AS [PartitionID], [TableID], [Name] AS [PartitionName] from $system.tmschema_partitions
     """,
     )
 
     tables = fabric.evaluate_dax(
-        dataset=dataset,
-        workspace=workspace,
+        dataset=dataset_id,
+        workspace=workspace_id,
         dax_string="""
     select [ID] AS [TableID], [Name] AS [TableName] from $system.tmschema_tables
     """,
@@ -1352,3 +1394,15 @@ def _is_valid_uuid(
         return True
     except ValueError:
         return False
+
+
+def _get_fabric_context_setting(name: str):
+
+    from synapse.ml.internal_utils.session_utils import get_fabric_context
+
+    return get_fabric_context().get(name)
+
+
+def get_tenant_id():
+
+    _get_fabric_context_setting(name="trident.tenant.id")