semantic-link-labs 0.8.9__tar.gz → 0.8.11__tar.gz

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.8.9
+Version: 0.8.11
 Summary: Semantic Link Labs for Microsoft Fabric
 Author: Microsoft Corporation
 License: MIT License
@@ -27,7 +27,7 @@ Requires-Dist: pytest>=8.2.1; extra == "test"
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.9&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.11&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 
@@ -64,6 +64,7 @@ Check out the video below for an introduction to Semantic Link, Semantic Link La
     * [Dynamically generate a Direct Lake semantic model](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.generate_direct_lake_semantic_model)
     * [Check why a Direct Lake semantic model would fallback to DirectQuery](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.check_fallback_reason)
     * [View a measure dependency tree](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.measure_dependency_tree)
+    * [View unique columns touched in a single (or multiple) DAX query(ies)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.get_dax_query_dependencies)
 * Reports
     * [Report Best Practice Analyzer (BPA)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.report.html#sempy_labs.report.run_report_bpa)
     * [View report metadata](https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Report%20Analysis.ipynb)
@@ -141,6 +142,8 @@ An even better way to ensure the semantic-link-labs library is available in your
 2. Select your newly created environment within the 'Environment' drop down in the navigation bar at the top of the notebook
 
 ## Version History
+* [0.8.11](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.11) (December 19, 2024)
+* [0.8.10](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.10) (December 16, 2024)
 * [0.8.9](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.9) (December 4, 2024)
 * [0.8.8](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.8) (November 28, 2024)
 * [0.8.7](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.7) (November 27, 2024)

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/README.md

@@ -1,7 +1,7 @@
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.9&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.11&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 
@@ -38,6 +38,7 @@ Check out the video below for an introduction to Semantic Link, Semantic Link La
     * [Dynamically generate a Direct Lake semantic model](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.generate_direct_lake_semantic_model)
     * [Check why a Direct Lake semantic model would fallback to DirectQuery](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.check_fallback_reason)
     * [View a measure dependency tree](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.measure_dependency_tree)
+    * [View unique columns touched in a single (or multiple) DAX query(ies)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.get_dax_query_dependencies)
 * Reports
     * [Report Best Practice Analyzer (BPA)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.report.html#sempy_labs.report.run_report_bpa)
     * [View report metadata](https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Report%20Analysis.ipynb)
@@ -115,6 +116,8 @@ An even better way to ensure the semantic-link-labs library is available in your
 2. Select your newly created environment within the 'Environment' drop down in the navigation bar at the top of the notebook
 
 ## Version History
+* [0.8.11](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.11) (December 19, 2024)
+* [0.8.10](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.10) (December 16, 2024)
 * [0.8.9](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.9) (December 4, 2024)
 * [0.8.8](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.8) (November 28, 2024)
 * [0.8.7](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.7) (November 27, 2024)

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/docs/source/conf.py

@@ -13,7 +13,7 @@ sys.path.insert(0, os.path.abspath('../../src/'))
 project = 'semantic-link-labs'
 copyright = '2024, Microsoft and community'
 author = 'Microsoft and community'
-release = '0.8.9'
+release = '0.8.11'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/pyproject.toml

@@ -7,7 +7,7 @@ name="semantic-link-labs"
 authors = [
     { name = "Microsoft Corporation" },
 ]
-version="0.8.9"
+version="0.8.11"
 description="Semantic Link Labs for Microsoft Fabric"
 readme="README.md"
 requires-python=">=3.10,<3.12"

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/semantic_link_labs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.8.9
+Version: 0.8.11
 Summary: Semantic Link Labs for Microsoft Fabric
 Author: Microsoft Corporation
 License: MIT License
@@ -27,7 +27,7 @@ Requires-Dist: pytest>=8.2.1; extra == "test"
 # Semantic Link Labs
 
 [![PyPI version](https://badge.fury.io/py/semantic-link-labs.svg)](https://badge.fury.io/py/semantic-link-labs)
-[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.9&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.11&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Downloads](https://static.pepy.tech/badge/semantic-link-labs)](https://pepy.tech/project/semantic-link-labs)
 
@@ -64,6 +64,7 @@ Check out the video below for an introduction to Semantic Link, Semantic Link La
     * [Dynamically generate a Direct Lake semantic model](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.generate_direct_lake_semantic_model)
     * [Check why a Direct Lake semantic model would fallback to DirectQuery](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.directlake.html#sempy_labs.directlake.check_fallback_reason)
     * [View a measure dependency tree](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.measure_dependency_tree)
+    * [View unique columns touched in a single (or multiple) DAX query(ies)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.html#sempy_labs.get_dax_query_dependencies)
 * Reports
     * [Report Best Practice Analyzer (BPA)](https://semantic-link-labs.readthedocs.io/en/stable/sempy_labs.report.html#sempy_labs.report.run_report_bpa)
     * [View report metadata](https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Report%20Analysis.ipynb)
@@ -141,6 +142,8 @@ An even better way to ensure the semantic-link-labs library is available in your
 2. Select your newly created environment within the 'Environment' drop down in the navigation bar at the top of the notebook
 
 ## Version History
+* [0.8.11](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.11) (December 19, 2024)
+* [0.8.10](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.10) (December 16, 2024)
 * [0.8.9](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.9) (December 4, 2024)
 * [0.8.8](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.8) (November 28, 2024)
 * [0.8.7](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.7) (November 27, 2024)

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/semantic_link_labs.egg-info/SOURCES.txt

@@ -54,6 +54,7 @@ src/sempy_labs/_generate_semantic_model.py
 src/sempy_labs/_git.py
 src/sempy_labs/_helper_functions.py
 src/sempy_labs/_icons.py
+src/sempy_labs/_job_scheduler.py
 src/sempy_labs/_kql_databases.py
 src/sempy_labs/_kql_querysets.py
 src/sempy_labs/_list_functions.py

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/sempy_labs/__init__.py

@@ -1,3 +1,4 @@
+from sempy_labs._job_scheduler import list_item_job_instances
 from sempy_labs._gateways import (
     list_gateway_members,
     list_gateway_role_assigments,
@@ -8,6 +9,7 @@ from sempy_labs._gateways import (
     create_vnet_gateway,
     update_vnet_gateway,
     update_on_premises_gateway,
+    bind_semantic_model_to_gateway,
 )
 
 from sempy_labs._authentication import (
@@ -159,7 +161,8 @@ from sempy_labs._git import (
     commit_to_git,
     initialize_git_connection,
     update_from_git,
-    connect_workspace_to_git,
+    connect_workspace_to_azure_dev_ops,
+    connect_workspace_to_github,
     disconnect_workspace_from_git,
 )
 from sempy_labs._dataflows import (
@@ -205,6 +208,8 @@ from sempy_labs._list_functions import (
     list_lakehouses,
     list_sql_endpoints,
     update_item,
+    list_server_properties,
+    list_semantic_model_errors,
 )
 from sempy_labs._helper_functions import (
     convert_to_friendly_case,
@@ -230,6 +235,7 @@ from sempy_labs._helper_functions import (
     get_capacity_id,
     get_capacity_name,
     resolve_capacity_name,
+    get_tenant_id,
 )
 from sempy_labs._model_bpa_bulk import (
     run_model_bpa_bulk,
@@ -364,7 +370,8 @@ __all__ = [
     "commit_to_git",
     "initialize_git_connection",
     "update_from_git",
-    "connect_workspace_to_git",
+    "connect_workspace_to_azure_dev_ops",
+    "connect_workspace_to_github",
     "disconnect_workspace_from_git",
     "create_environment",
     "delete_environment",
@@ -458,4 +465,9 @@ __all__ = [
     "update_vnet_gateway",
     "update_on_premises_gateway",
     "get_semantic_model_definition",
+    "get_tenant_id",
+    "list_server_properties",
+    "bind_semantic_model_to_gateway",
+    "list_semantic_model_errors",
+    "list_item_job_instances",
 ]

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/sempy_labs/_authentication.py

@@ -91,11 +91,13 @@ class ServicePrincipalTokenProvider(TokenProvider):
 
         return cls(credential)
 
-    def __call__(self, audience: Literal["pbi", "storage"] = "pbi") -> str:
+    def __call__(
+        self, audience: Literal["pbi", "storage", "azure", "graph"] = "pbi"
+    ) -> str:
         """
         Parameters
         ----------
-        audience : Literal["pbi", "storage"] = "pbi") -> str
+        audience : Literal["pbi", "storage", "azure", "graph"] = "pbi") -> str
             Literal if it's for PBI/Fabric API call or OneLake/Storage Account call.
         """
         if audience == "pbi":
@@ -104,5 +106,32 @@ class ServicePrincipalTokenProvider(TokenProvider):
             ).token
         elif audience == "storage":
             return self.credential.get_token("https://storage.azure.com/.default").token
+        elif audience == "azure":
+            return self.credential.get_token(
+                "https://management.azure.com/.default"
+            ).token
+        elif audience == "graph":
+            return self.credential.get_token(
+                "https://graph.microsoft.com/.default"
+            ).token
         else:
             raise NotImplementedError
+
+
+def _get_headers(
+    token_provider: str, audience: Literal["pbi", "storage", "azure", "graph"] = "azure"
+):
+    """
+    Generates headers for an API request.
+    """
+
+    token = token_provider(audience=audience)
+
+    headers = {"Authorization": f"Bearer {token}"}
+
+    if audience == "graph":
+        headers["ConsistencyLevel"] = "eventual"
+    else:
+        headers["Content-Type"] = "application/json"
+
+    return headers

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/sempy_labs/_clear_cache.py

@@ -1,39 +1,41 @@
 import sempy.fabric as fabric
 from sempy_labs._helper_functions import (
-    resolve_dataset_id,
     is_default_semantic_model,
     _get_adls_client,
+    resolve_workspace_name_and_id,
+    resolve_dataset_name_and_id,
 )
 from typing import Optional
 import sempy_labs._icons as icons
 from sempy._utils._log import log
 import pandas as pd
 from sempy.fabric.exceptions import FabricHTTPException
+from uuid import UUID
 
 
-def clear_cache(dataset: str, workspace: Optional[str] = None):
+def clear_cache(dataset: str | UUID, workspace: Optional[str | UUID] = None):
     """
     Clears the cache of a semantic model.
     See `here <https://learn.microsoft.com/analysis-services/instances/clear-the-analysis-services-caches?view=asallproducts-allversions>`_ for documentation.
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
-    workspace : str, default=None
-        The Fabric workspace name.
+    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.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     if is_default_semantic_model(dataset=dataset, workspace=workspace):
         raise ValueError(
             f"{icons.red_dot} Cannot run XMLA operations against a default semantic model. Please choose a different semantic model. "
             "See here for more information: https://learn.microsoft.com/fabric/data-warehouse/semantic-models"
         )
 
-    dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
 
     xmla = f"""
             <ClearCache xmlns="http://schemas.microsoft.com/analysisservices/2003/engine">
@@ -42,27 +44,27 @@ def clear_cache(dataset: str, workspace: Optional[str] = None):
                 </Object>
             </ClearCache>
             """
-    fabric.execute_xmla(dataset=dataset, xmla_command=xmla, workspace=workspace)
+    fabric.execute_xmla(dataset=dataset_id, xmla_command=xmla, workspace=workspace_id)
     print(
-        f"{icons.green_dot} Cache cleared for the '{dataset}' semantic model within the '{workspace}' workspace."
+        f"{icons.green_dot} Cache cleared for the '{dataset_name}' semantic model within the '{workspace_name}' workspace."
     )
 
 
 @log
 def backup_semantic_model(
-    dataset: str,
+    dataset: str | UUID,
     file_path: str,
     allow_overwrite: bool = True,
     apply_compression: bool = True,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     `Backs up <https://learn.microsoft.com/azure/analysis-services/analysis-services-backup>`_ a semantic model to the ADLS Gen2 storage account connected to the workspace.
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
+    dataset : str | uuid.UUID
+        Name or ID of the semantic model.
     file_path : str
         The ADLS Gen2 storage account location in which to backup the semantic model. Always saves within the 'power-bi-backup/<workspace name>' folder.
         Must end in '.abf'.
@@ -72,8 +74,8 @@ def backup_semantic_model(
         If True, overwrites backup files of the same name. If False, the file you are saving cannot have the same name as a file that already exists in the same location.
     apply_compression : bool, default=True
         If True, compresses the backup file. Compressed backup files save disk space, but require slightly higher CPU utilization.
-    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.
     """
@@ -83,31 +85,32 @@ def backup_semantic_model(
             f"{icons.red_dot} The backup file for restoring must be in the .abf format."
         )
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
 
     tmsl = {
         "backup": {
-            "database": dataset,
+            "database": dataset_name,
             "file": file_path,
             "allowOverwrite": allow_overwrite,
             "applyCompression": apply_compression,
         }
     }
 
-    fabric.execute_tmsl(script=tmsl, workspace=workspace)
+    fabric.execute_tmsl(script=tmsl, workspace=workspace_id)
     print(
-        f"{icons.green_dot} The '{dataset}' semantic model within the '{workspace}' workspace has been backed up to the '{file_path}' location."
+        f"{icons.green_dot} The '{dataset_name}' semantic model within the '{workspace_name}' workspace has been backed up to the '{file_path}' location."
     )
 
 
 @log
 def restore_semantic_model(
-    dataset: str,
+    dataset: str | UUID,
     file_path: str,
     allow_overwrite: bool = True,
     ignore_incompatibilities: bool = True,
     force_restore: bool = False,
-    workspace: Optional[str] = None,
+    workspace: Optional[str | UUID] = None,
 ):
     """
     `Restores <https://learn.microsoft.com/power-bi/enterprise/service-premium-backup-restore-dataset>`_ a semantic model based on a backup (.abf) file
@@ -115,8 +118,8 @@ def restore_semantic_model(
 
     Parameters
     ----------
-    dataset : str
-        Name of the semantic model.
+    dataset : str | uuid.UUID
+        Name or ID of the semantic model.
     file_path : str
         The location in which to backup the semantic model. Must end in '.abf'.
         Example 1: file_path = 'MyModel.abf'
@@ -127,23 +130,23 @@ def restore_semantic_model(
         If True, ignores incompatibilities between Azure Analysis Services and Power BI Premium.
     force_restore: bool, default=False
         If True, restores the semantic model with the existing semantic model unloaded and offline.
-    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.
     """
-    # https://learn.microsoft.com/en-us/power-bi/enterprise/service-premium-backup-restore-dataset
 
     if not file_path.endswith(".abf"):
         raise ValueError(
             f"{icons.red_dot} The backup file for restoring must be in the .abf format."
         )
 
-    workspace = fabric.resolve_workspace_name(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(dataset, workspace_id)
 
     tmsl = {
         "restore": {
-            "database": dataset,
+            "database": dataset_name,
             "file": file_path,
             "allowOverwrite": allow_overwrite,
             "security": "copyAll",
@@ -154,10 +157,10 @@ def restore_semantic_model(
     if force_restore:
         tmsl["restore"]["forceRestore"] = force_restore
 
-    fabric.execute_tmsl(script=tmsl, workspace=workspace)
+    fabric.execute_tmsl(script=tmsl, workspace=workspace_id)
 
     print(
-        f"{icons.green_dot} The '{dataset}' semantic model has been restored to the '{workspace}' workspace based on teh '{file_path}' backup file."
+        f"{icons.green_dot} The '{dataset_name}' semantic model has been restored to the '{workspace_name}' workspace based on the '{file_path}' backup file."
     )
 
 
@@ -243,15 +246,15 @@ def copy_semantic_model_backup_file(
 
 
 @log
-def list_backups(workspace: Optional[str] = None) -> pd.DataFrame:
+def list_backups(workspace: Optional[str | UUID] = None) -> pd.DataFrame:
     """
     Shows a list of backup files contained within a workspace's ADLS Gen2 storage account.
     Requirement: An ADLS Gen2 storage account must be `connected to the workspace <https://learn.microsoft.com/power-bi/transform-model/dataflows/dataflows-azure-data-lake-storage-integration#connect-to-an-azure-data-lake-gen-2-at-a-workspace-level>`_.
 
     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.
 
@@ -262,8 +265,7 @@ def list_backups(workspace: Optional[str] = None) -> pd.DataFrame:
     """
 
     client = fabric.PowerBIRestClient()
-    workspace = fabric.resolve_workspace_name(workspace)
-    workspace_id = fabric.resolve_workspace_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     response = client.get(
         f"/v1.0/myorg/resources?resourceType=StorageAccount&folderObjectId={workspace_id}"
     )
@@ -274,7 +276,7 @@ def list_backups(workspace: Optional[str] = None) -> pd.DataFrame:
     v = response.json().get("value", [])
     if not v:
         raise ValueError(
-            f"{icons.red_dot} A storage account is not associated with the '{workspace}' workspace."
+            f"{icons.red_dot} A storage account is not associated with the '{workspace_name}' workspace."
         )
     storage_account = v[0]["resourceName"]
 

{semantic_link_labs-0.8.9 → semantic_link_labs-0.8.11}/src/sempy_labs/_connections.py

@@ -5,6 +5,7 @@ from typing import Optional
 from sempy_labs._helper_functions import (
     pagination,
     _is_valid_uuid,
+    resolve_workspace_name_and_id,
 )
 from uuid import UUID
 import sempy_labs._icons as icons
@@ -19,7 +20,7 @@ def delete_connection(connection: str | UUID):
 
     Parameters
     ----------
-    connection : str | UUID
+    connection : str | uuid.UUID
         The connection name or ID.
     """
 
@@ -42,9 +43,9 @@ def delete_connection_role_assignment(connection: str | UUID, role_assignment_id
 
     Parameters
     ----------
-    connection : str | UUID
+    connection : str | uuid.UUID
         The connection name or ID.
-    role_assignment_id : UUID
+    role_assignment_id : uuid.UUID
         The role assignment ID.
     """
 
@@ -87,7 +88,7 @@ def list_connection_role_assignments(connection: str | UUID) -> pd.DataFrame:
 
     Parameters
     ----------
-    connection : str | UUID
+    connection : str | uuid.UUID
         The connection name or ID.
 
     Returns
@@ -205,7 +206,7 @@ def list_connections() -> pd.DataFrame:
 
 
 def list_item_connections(
-    item_name: str, item_type: str, workspace: Optional[str] = None
+    item_name: str, item_type: str, workspace: Optional[str | UUID] = None
 ) -> pd.DataFrame:
     """
     Shows the list of connections that the specified item is connected to.
@@ -218,8 +219,8 @@ def list_item_connections(
         The item name.
     item_type : str
         The `item type <https://learn.microsoft.com/rest/api/fabric/core/items/update-item?tabs=HTTP#itemtype>`_.
-    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.
 
@@ -229,11 +230,10 @@ def list_item_connections(
         A pandas dataframe showing the list of connections that the specified item is connected to.
     """
 
-    workspace = fabric.resolve_workspace_name(workspace)
-    workspace_id = fabric.resolve_workspace_id(workspace)
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     item_type = item_type[0].upper() + item_type[1:]
     item_id = fabric.resolve_item_id(
-        item_name=item_name, type=item_type, workspace=workspace
+        item_name=item_name, type=item_type, workspace=workspace_id
     )
 
     client = fabric.FabricRestClient()
@@ -416,7 +416,7 @@ def create_on_prem_connection(
     ----------
     name : str
         The name of the connection.
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The name or Id of the gateway.
     server_name : str
         The name of the server.
@@ -485,7 +485,7 @@ def create_vnet_connection(
     user_name: str,
     password: str,
     privacy_level: str,
-    connection_encryption: Optional[str] = "NotEncrypted",
+    connection_encryption: str = "NotEncrypted",
     skip_test_connection: bool = False,
 ):
     """
@@ -497,7 +497,7 @@ def create_vnet_connection(
     ----------
     name : str
         The name of the connection.
-    gateway : str | UUID
+    gateway : str | uuid.UUID
         The name or Id of the gateway.
     server_name : str
         The name of the server.