semantic-link-labs 0.8.8__tar.gz → 0.8.10__tar.gz

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.8.8
+Version: 0.8.10
 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.8&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.10&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.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)
 * [0.8.6](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.6) (November 14, 2024)

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/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.8&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.10&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.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)
 * [0.8.6](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.6) (November 14, 2024)

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/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.8'
+release = '0.8.10'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/pyproject.toml

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

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/src/semantic_link_labs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.8.8
+Version: 0.8.10
 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.8&style=flat)](https://readthedocs.org/projects/semantic-link-labs/)
+[![Read The Docs](https://readthedocs.org/projects/semantic-link-labs/badge/?version=0.8.10&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.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)
 * [0.8.6](https://github.com/microsoft/semantic-link-labs/releases/tag/0.8.6) (November 14, 2024)

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/src/sempy_labs/__init__.py

@@ -8,6 +8,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 (
@@ -190,6 +191,7 @@ from sempy_labs._generate_semantic_model import (
     get_semantic_model_bim,
     get_semantic_model_size,
     update_semantic_model_from_bim,
+    get_semantic_model_definition,
 )
 from sempy_labs._list_functions import (
     list_reports_using_semantic_model,
@@ -204,6 +206,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,
@@ -229,6 +233,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,
@@ -456,4 +461,9 @@ __all__ = [
     "create_vnet_gateway",
     "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",
 ]

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/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.8 → semantic_link_labs-0.8.10}/src/sempy_labs/_dataflows.py

@@ -42,7 +42,7 @@ def list_dataflows(workspace: Optional[str] = None):
             "Dataflow Id": v.get("objectId"),
             "Dataflow Name": v.get("name"),
             "Configured By": v.get("configuredBy"),
-            "Users": [v.get("users")],
+            "Users": v.get("users"),
             "Generation": v.get("generation"),
         }
         df = pd.concat(

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/src/sempy_labs/_dax.py

@@ -6,7 +6,7 @@ from sempy_labs._helper_functions import (
     format_dax_object_name,
 )
 from sempy_labs._model_dependencies import get_model_calc_dependencies
-from typing import Optional
+from typing import Optional, List
 from sempy._utils._log import log
 from tqdm.auto import tqdm
 
@@ -67,8 +67,9 @@ def evaluate_dax_impersonation(
 @log
 def get_dax_query_dependencies(
     dataset: str,
-    dax_string: str,
+    dax_string: str | List[str],
     put_in_memory: bool = False,
+    show_vertipaq_stats: bool = True,
     workspace: Optional[str] = None,
 ) -> pd.DataFrame:
     """
@@ -78,10 +79,12 @@ def get_dax_query_dependencies(
     ----------
     dataset : str
         Name of the semantic model.
-    dax_string : str
-        The DAX query.
+    dax_string : str | List[str]
+        The DAX query or list of DAX queries.
     put_in_memory : bool, default=False
         If True, ensures that the dependent columns are put into memory in order to give realistic Vertipaq stats (i.e. Total Size etc.).
+    show_vertipaq_stats : bool, default=True
+        If True, shows vertipaq stats (i.e. Total Size, Data Size, Dictionary Size, Hierarchy Size)
     workspace : str, default=None
         The Fabric workspace name.
         Defaults to None which resolves to the workspace of the attached lakehouse
@@ -96,67 +99,79 @@ def get_dax_query_dependencies(
     if workspace is None:
         workspace = fabric.resolve_workspace_name(workspace)
 
-    # Escape quotes in dax
-    dax_string = dax_string.replace('"', '""')
-    final_query = f"""
-        EVALUATE
-        VAR source_query = "{dax_string}"
-        VAR all_dependencies = SELECTCOLUMNS(
-            INFO.CALCDEPENDENCY("QUERY", source_query),
-                "Referenced Object Type",[REFERENCED_OBJECT_TYPE],
-                "Referenced Table", [REFERENCED_TABLE],
-                "Referenced Object", [REFERENCED_OBJECT]
-            )             
-        RETURN all_dependencies
-        """
-    dep = fabric.evaluate_dax(
-        dataset=dataset, workspace=workspace, dax_string=final_query
-    )
+    if isinstance(dax_string, str):
+        dax_string = [dax_string]
 
-    # Clean up column names and values (remove outside square brackets, underscorees in object type)
-    dep.columns = dep.columns.map(lambda x: x[1:-1])
-    dep["Referenced Object Type"] = (
-        dep["Referenced Object Type"].str.replace("_", " ").str.title()
-    )
-    dep
-
-    # Dataframe df will contain the output of all dependencies of the objects used in the query
-    df = dep.copy()
+    final_df = pd.DataFrame(columns=["Object Type", "Table", "Object"])
 
     cd = get_model_calc_dependencies(dataset=dataset, workspace=workspace)
 
-    for _, r in dep.iterrows():
-        ot = r["Referenced Object Type"]
-        object_name = r["Referenced Object"]
-        table_name = r["Referenced Table"]
-        cd_filt = cd[
-            (cd["Object Type"] == ot)
-            & (cd["Object Name"] == object_name)
-            & (cd["Table Name"] == table_name)
-        ]
-
-        # Adds in the dependencies of each object used in the query (i.e. relationship etc.)
-        if len(cd_filt) > 0:
-            subset = cd_filt[
-                ["Referenced Object Type", "Referenced Table", "Referenced Object"]
+    for dax in dax_string:
+        # Escape quotes in dax
+        dax = dax.replace('"', '""')
+        final_query = f"""
+            EVALUATE
+            VAR source_query = "{dax}"
+            VAR all_dependencies = SELECTCOLUMNS(
+                INFO.CALCDEPENDENCY("QUERY", source_query),
+                    "Referenced Object Type",[REFERENCED_OBJECT_TYPE],
+                    "Referenced Table", [REFERENCED_TABLE],
+                    "Referenced Object", [REFERENCED_OBJECT]
+                )             
+            RETURN all_dependencies
+            """
+        dep = fabric.evaluate_dax(
+            dataset=dataset, workspace=workspace, dax_string=final_query
+        )
+
+        # Clean up column names and values (remove outside square brackets, underscorees in object type)
+        dep.columns = dep.columns.map(lambda x: x[1:-1])
+        dep["Referenced Object Type"] = (
+            dep["Referenced Object Type"].str.replace("_", " ").str.title()
+        )
+
+        # Dataframe df will contain the output of all dependencies of the objects used in the query
+        df = dep.copy()
+
+        for _, r in dep.iterrows():
+            ot = r["Referenced Object Type"]
+            object_name = r["Referenced Object"]
+            table_name = r["Referenced Table"]
+            cd_filt = cd[
+                (cd["Object Type"] == ot)
+                & (cd["Object Name"] == object_name)
+                & (cd["Table Name"] == table_name)
             ]
-            df = pd.concat([df, subset], ignore_index=True)
-
-    df.columns = df.columns.map(lambda x: x.replace("Referenced ", ""))
-    # Remove duplicates
-    df = df.drop_duplicates().reset_index(drop=True)
-    # Only show columns and remove the rownumber column
-    df = df[
-        (df["Object Type"].isin(["Column", "Calc Column"]))
-        & (~df["Object"].str.startswith("RowNumber-"))
+
+            # Adds in the dependencies of each object used in the query (i.e. relationship etc.)
+            if len(cd_filt) > 0:
+                subset = cd_filt[
+                    ["Referenced Object Type", "Referenced Table", "Referenced Object"]
+                ]
+                df = pd.concat([df, subset], ignore_index=True)
+
+        df.columns = df.columns.map(lambda x: x.replace("Referenced ", ""))
+        final_df = pd.concat([df, final_df], ignore_index=True)
+
+    final_df = final_df[
+        (final_df["Object Type"].isin(["Column", "Calc Column"]))
+        & (~final_df["Object"].str.startswith("RowNumber-"))
     ]
+    final_df = final_df.drop_duplicates().reset_index(drop=True)
+    final_df = final_df.rename(columns={"Table": "Table Name", "Object": "Column Name"})
+    final_df.drop(columns=["Object Type"], inplace=True)
+
+    if not show_vertipaq_stats:
+        return final_df
 
     # Get vertipaq stats, filter to just the objects in the df dataframe
-    df["Full Object"] = format_dax_object_name(df["Table"], df["Object"])
+    final_df["Full Object"] = format_dax_object_name(
+        final_df["Table Name"], final_df["Column Name"]
+    )
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace, extended=True)
     dfC["Full Object"] = format_dax_object_name(dfC["Table Name"], dfC["Column Name"])
 
-    dfC_filtered = dfC[dfC["Full Object"].isin(df["Full Object"].values)][
+    dfC_filtered = dfC[dfC["Full Object"].isin(final_df["Full Object"].values)][
         [
             "Table Name",
             "Column Name",

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/src/sempy_labs/_gateways.py

@@ -6,6 +6,8 @@ from sempy_labs._helper_functions import (
     pagination,
     _is_valid_uuid,
     resolve_capacity_id,
+    resolve_workspace_name_and_id,
+    resolve_dataset_name_and_id,
 )
 from uuid import UUID
 import sempy_labs._icons as icons
@@ -437,3 +439,47 @@ def update_vnet_gateway(
         raise FabricHTTPException(response)
 
     print(f"{icons.green_dot} The '{gateway}' has been updated accordingly.")
+
+
+def bind_semantic_model_to_gateway(
+    dataset: str | UUID, gateway: str | UUID, workspace: Optional[str | UUID] = None
+):
+    """
+    Binds the specified dataset from the specified workspace to the specified gateway.
+
+    This is a wrapper function for the following API: `Datasets - Bind To Gateway In Group <https://learn.microsoft.com/rest/api/power-bi/datasets/bind-to-gateway-in-group>`_.
+
+    Parameters
+    ----------
+    dataset : str | UUID
+        The name or ID of the semantic model.
+    gateway : str | UUID
+        The name or ID of the gateway.
+    workspace : str | 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.
+    """
+
+    (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
+    (dataset_name, dataset_id) = resolve_dataset_name_and_id(
+        dataset, workspace=workspace_id
+    )
+
+    gateway_id = _resolve_gateway_id(gateway)
+    payload = {
+        "gatewayObjectId": gateway_id,
+    }
+
+    client = fabric.FabricRestClient()
+    response = client.post(
+        f"/v1.0/myorg/groups/{workspace_id}/datasets/{dataset_id}/Default.BindToGateway",
+        json=payload,
+    )
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(
+        f"{icons.green_dot} The '{dataset_name}' semantic model within the '{workspace_name}' workspace has been binded to the '{gateway_id}' gateway."
+    )

{semantic_link_labs-0.8.8 → semantic_link_labs-0.8.10}/src/sempy_labs/_generate_semantic_model.py

@@ -2,7 +2,7 @@ import sempy.fabric as fabric
 import pandas as pd
 import json
 import os
-from typing import Optional
+from typing import Optional, List
 from sempy_labs._helper_functions import (
     resolve_lakehouse_name,
     resolve_workspace_name_and_id,
@@ -203,14 +203,7 @@ def update_semantic_model_from_bim(
     """
 
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-    dfD = fabric.list_datasets(workspace=workspace, mode="rest")
-    dfD_filt = dfD[dfD["Dataset Name"] == dataset]
-    if len(dfD_filt) == 0:
-        raise ValueError(
-            f"{icons.red_dot} The '{dataset}' semantic model within the '{workspace}' workspace does not exist."
-        )
-    dataset_id = dfD_filt["Dataset Id"].iloc[0]
+    dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
 
     client = fabric.FabricRestClient()
     defPBIDataset = {"version": "1.0", "settings": {}}
@@ -303,12 +296,11 @@ def deploy_semantic_model(
     bim = get_semantic_model_bim(dataset=source_dataset, workspace=source_workspace)
 
     # Create the semantic model if the model does not exist
-    if len(dfD_filt) == 0:
+    if dfD_filt.empty:
         create_semantic_model_from_bim(
             dataset=target_dataset,
             bim_file=bim,
             workspace=target_workspace,
-            overwrite=overwrite,
         )
     # Update the semantic model if the model exists
     else:
@@ -329,8 +321,6 @@ def get_semantic_model_bim(
     """
     Extracts the Model.bim file for a given semantic model.
 
-    This is a wrapper function for the following API: `Items - Get Semantic Model Definition <https://learn.microsoft.com/rest/api/fabric/semanticmodel/items/get-semantic-model-definition>`_.
-
     Parameters
     ----------
     dataset : str
@@ -352,20 +342,9 @@ def get_semantic_model_bim(
         The Model.bim file for the semantic model.
     """
 
-    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
-
-    fmt = "TMSL"
-    client = fabric.FabricRestClient()
-    dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
-    response = client.post(
-        f"/v1/workspaces/{workspace_id}/semanticModels/{dataset_id}/getDefinition?format={fmt}",
+    bimJson = get_semantic_model_definition(
+        dataset=dataset, workspace=workspace, format="TMSL", return_dataframe=False
     )
-    result = lro(client, response).json()
-    df_items = pd.json_normalize(result["definition"]["parts"])
-    df_items_filt = df_items[df_items["path"] == "model.bim"]
-    payload = df_items_filt["payload"].iloc[0]
-    bimFile = _decode_b64(payload)
-    bimJson = json.loads(bimFile)
 
     if save_to_file_name is not None:
         if not lakehouse_attached():
@@ -384,12 +363,80 @@ def get_semantic_model_bim(
         with open(filePath, "w") as json_file:
             json.dump(bimJson, json_file, indent=4)
         print(
-            f"{icons.green_dot} The .bim file for the '{dataset}' semantic model has been saved to the '{lakehouse}' in this location: '{filePath}'.\n\n"
+            f"{icons.green_dot} The {fileExt} file for the '{dataset}' semantic model has been saved to the '{lakehouse}' in this location: '{filePath}'.\n\n"
         )
 
     return bimJson
 
 
+def get_semantic_model_definition(
+    dataset: str,
+    format: str = "TMSL",
+    workspace: Optional[str] = None,
+    return_dataframe: bool = True,
+) -> pd.DataFrame | dict | List:
+    """
+    Extracts the semantic model definition.
+
+    This is a wrapper function for the following API: `Items - Get Semantic Model Definition <https://learn.microsoft.com/rest/api/fabric/semanticmodel/items/get-semantic-model-definition>`_.
+
+    Parameters
+    ----------
+    dataset : str
+        Name of the semantic model.
+    format : str, default="TMSL"
+        The output format. Valid options are "TMSL" or "TMDL". "TMSL" returns the .bim file whereas "TMDL" returns the collection of TMDL files. Can also enter 'bim' for the TMSL version.
+    workspace : str, default=None
+        The Fabric workspace name 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.
+    return_dataframe : bool, default=True
+        If True, returns a dataframe.
+        If False, returns the .bim file for TMSL format. Returns a list of the TMDL files (decoded) for TMDL format.
+
+    Returns
+    -------
+    pandas.DataFrame | dict | List
+        A pandas dataframe with the semantic model definition or the file or files comprising the semantic model definition.
+    """
+
+    valid_formats = ["TMSL", "TMDL"]
+
+    format = format.upper()
+    if format == "BIM":
+        format = "TMSL"
+    if format not in valid_formats:
+        raise ValueError(
+            f"{icons.red_dot} Invalid format. Valid options: {valid_formats}."
+        )
+
+    (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
+
+    client = fabric.FabricRestClient()
+    dataset_id = resolve_dataset_id(dataset=dataset, workspace=workspace)
+    response = client.post(
+        f"/v1/workspaces/{workspace_id}/semanticModels/{dataset_id}/getDefinition?format={format}",
+    )
+    result = lro(client, response).json()
+
+    files = result["definition"]["parts"]
+
+    if return_dataframe:
+        return pd.json_normalize(files)
+    elif format == "TMSL":
+        payload = next(
+            (part["payload"] for part in files if part["path"] == "model.bim"), None
+        )
+        return json.loads(_decode_b64(payload))
+    else:
+        decoded_parts = [
+            {"file_name": part["path"], "content": _decode_b64(part["payload"])}
+            for part in files
+        ]
+
+        return decoded_parts
+
+
 def get_semantic_model_size(dataset: str, workspace: Optional[str] = None):
 
     workspace = fabric.resolve_workspace_name(workspace)