semantic-link-labs 0.8.3__py3-none-any.whl → 0.8.5__py3-none-any.whl

sempy_labs/admin/_basic_functions.py

@@ -1,5 +1,5 @@
 import sempy.fabric as fabric
-from typing import Optional, List, Union
+from typing import Optional, List, Union, Tuple
 from uuid import UUID
 import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
@@ -10,18 +10,26 @@ from sempy_labs._helper_functions import (
 import numpy as np
 import pandas as pd
 import time
+import urllib.parse
+from datetime import datetime
 
 
 def list_workspaces(
-    top: Optional[int] = 5000, skip: Optional[int] = None
+    top: Optional[int] = 5000,
+    filter: Optional[str] = None,
+    skip: Optional[int] = None,
 ) -> pd.DataFrame:
     """
     Lists workspaces for the organization. This function is the admin version of list_workspaces.
 
+    This is a wrapper function for the following API: `Admin - Groups GetGroupsAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/groups-get-groups-as-admin>`_.
+
     Parameters
     ----------
     top : int, default=5000
         Returns only the first n results. This parameter is mandatory and must be in the range of 1-5000.
+    filter : str, default=None
+        Returns a subset of a results based on `Odata filter <https://docs.oasis-open.org/odata/odata/v4.01/odata-v4.01-part2-url-conventions.html#sec_SystemQueryOptions>`_ query parameter condition.
     skip : int, default=None
         Skips the first n results. Use with top to fetch results beyond the first 5000.
 
@@ -48,6 +56,8 @@ def list_workspaces(
     url = f"/v1.0/myorg/admin/groups?$top={top}"
     if skip is not None:
         url = f"{url}&$skip={skip}"
+    if filter is not None:
+        url = f"{url}&$filter={filter}"
 
     client = fabric.PowerBIRestClient()
     response = client.get(url)
@@ -91,6 +101,8 @@ def assign_workspaces_to_capacity(
     """
     Assigns a workspace to a capacity. This function is the admin version.
 
+    This is a wrapper function for the following API: `Admin - Capacities AssignWorkspacesToCapacity <https://learn.microsoft.com/rest/api/power-bi/admin/capacities-assign-workspaces-to-capacity>`_.
+
     Parameters
     ----------
     source_capacity : str
@@ -113,9 +125,6 @@ def assign_workspaces_to_capacity(
     target_capacity_id = dfC_filt["Capacity Id"].iloc[0]
 
     if workspace is None:
-        # workspaces = fabric.list_workspaces(
-        #    filter=f"capacityId eq '{source_capacity_id.upper()}'"
-        # )["Id"].values
         dfW = list_workspaces()
         dfW = dfW[dfW["Capacity Id"].str.upper() == source_capacity_id.upper()]
         workspaces = dfW["Id"].tolist()
@@ -153,6 +162,8 @@ def list_capacities() -> pd.DataFrame:
     """
     Shows the a list of capacities and their properties. This function is the admin version.
 
+    This is a wrapper function for the following API: `Admin - Get Capacities As Admin <https://learn.microsoft.com/rest/api/power-bi/admin/get-capacities-as-admin>`_.
+
     Returns
     -------
     pandas.DataFrame
@@ -189,14 +200,14 @@ def list_tenant_settings() -> pd.DataFrame:
     """
     Lists all tenant settings.
 
+    This is a wrapper function for the following API: `Tenants - List Tenant Settings <https://learn.microsoft.com/rest/api/fabric/admin/tenants/list-tenant-settings>`_.
+
     Returns
     -------
     pandas.DataFrame
         A pandas dataframe showing the tenant settings.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/tenants/list-tenant-settings?tabs=HTTP
-
     client = fabric.FabricRestClient()
     response = client.get("/v1/admin/tenantsettings")
 
@@ -262,7 +273,9 @@ def _list_capacities_meta() -> pd.DataFrame:
 
 def unassign_workspaces_from_capacity(workspaces: str | List[str]):
     """
-    Unassigns workspace(s) from their capacity. This function is the admin version of list_workspaces.
+    Unassigns workspace(s) from their capacity.
+
+    This is a wrapper function for the following API: `Admin - Capacities UnassignWorkspacesFromCapacity <https://learn.microsoft.com/rest/api/power-bi/admin/capacities-unassign-workspaces-from-capacity>`_.
 
     Parameters
     ----------
@@ -270,8 +283,6 @@ def unassign_workspaces_from_capacity(workspaces: str | List[str]):
         The Fabric workspace name(s).
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/power-bi/admin/capacities-unassign-workspaces-from-capacity
-
     if isinstance(workspaces, str):
         workspaces = [workspaces]
 
@@ -293,14 +304,14 @@ def list_external_data_shares():
     """
     Lists external data shares in the tenant. This function is for admins.
 
+    This is a wrapper function for the following API: `External Data Shares - List External Data Shares <https://learn.microsoft.com/rest/api/fabric/admin/external-data-shares/list-external-data-shares>`_.
+
     Returns
     -------
     pandas.DataFrame
         A pandas dataframe showing a list of external data shares in the tenant.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/external-data-shares/list-external-data-shares?tabs=HTTP
-
     df = pd.DataFrame(
         columns=[
             "External Data Share Id",
@@ -355,6 +366,8 @@ def revoke_external_data_share(
     """
     Revokes the specified external data share. Note: This action cannot be undone.
 
+    This is a wrapper function for the following API: `External Data Shares - Revoke External Data Share <https://learn.microsoft.com/rest/api/fabric/admin/external-data-shares/revoke-external-data-share>`_.
+
     Parameters
     ----------
     external_data_share_id : UUID
@@ -365,8 +378,6 @@ def revoke_external_data_share(
         The Fabric workspace name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/external-data-shares/revoke-external-data-share?tabs=HTTP
-
     (workspace, workspace_id) = resolve_workspace_name_and_id(workspace)
 
     client = fabric.FabricRestClient()
@@ -384,23 +395,23 @@ def revoke_external_data_share(
 
 def list_capacities_delegated_tenant_settings(
     return_dataframe: bool = True,
-) -> Optional[pd.DataFrame | dict]:
+) -> pd.DataFrame | dict:
     """
     Returns list of tenant setting overrides that override at the capacities.
 
+    This is a wrapper function for the following API: `Tenants - List Capacities Tenant Settings Overrides <https://learn.microsoft.com/rest/api/fabric/admin/tenants/list-capacities-tenant-settings-overrides>`_.
+
     Parameters
     ----------
     return_dataframe : bool, default=True
-        If True, returns a dataframe. If False, returns a dictionary
+        If True, returns a dataframe. If False, returns a dictionary.
 
     Returns
     -------
-    pandas.DataFrame
+    pandas.DataFrame | dict
         A pandas dataframe showing a list of tenant setting overrides that override at the capacities.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/tenants/list-capacities-tenant-settings-overrides?tabs=HTTP
-
     df = pd.DataFrame(
         columns=[
             "Capacity Id",
@@ -519,14 +530,14 @@ def list_datasets() -> pd.DataFrame:
     """
     Shows a list of datasets for the organization.
 
+    This is a wrapper function for the following API: `Admin - Datasets GetDatasetsAsAdmin <https://learn.microsoft.com/rest/api/power-bi/admin/datasets-get-datasets-as-admin>`_.
+
     Returns
     -------
     pandas.DataFrame
         A pandas dataframe showing a list of datasets for the organization.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/power-bi/admin/datasets-get-datasets-as-admin
-
     df = pd.DataFrame(
         columns=[
             "Dataset Id",
@@ -609,6 +620,8 @@ def list_item_access_details(
     """
     Returns a list of users (including groups and service principals) and lists their workspace roles.
 
+    This is a wrapper function for the following API: `Items - List Item Access Details <https://learn.microsoft.com/rest/api/fabric/admin/items/list-item-access-details>`_.
+
     Parameters
     ----------
     item_name : str
@@ -626,13 +639,8 @@ def list_item_access_details(
         A pandas dataframe showing a list of users (including groups and service principals) and lists their workspace roles.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/items/list-item-access-details?tabs=HTTP
-
-    workspace = fabric.resolve_workspace_name(workspace)
-    workspace_id = fabric.resolve_workspace_id(workspace)
-    item_id = fabric.resolve_item_id(
-        item_name=item_name, type=type, workspace=workspace
-    )
+    workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
+    item_id = _resolve_item_id(item_name=item_name, type=type, workspace=workspace_name)
 
     df = pd.DataFrame(
         columns=[
@@ -678,7 +686,9 @@ def list_access_entities(
     user_email_address: str,
 ) -> pd.DataFrame:
     """
-    Shows a list of permission details for Fabric and PowerBI items the specified user can access.
+    Shows a list of permission details for Fabric and Power BI items the specified user can access.
+
+    This is a wrapper function for the following API: `Users - List Access Entities <https://learn.microsoft.com/rest/api/fabric/admin/users/list-access-entities>`_.
 
     Parameters
     ----------
@@ -688,11 +698,9 @@ def list_access_entities(
     Returns
     -------
     pandas.DataFrame
-        A pandas dataframe showing a list of permission details for Fabric and PowerBI items the specified user can access.
+        A pandas dataframe showing a list of permission details for Fabric and Power BI items the specified user can access.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/users/list-access-entities?tabs=HTTP
-
     df = pd.DataFrame(
         columns=[
             "Item Id",
@@ -732,6 +740,8 @@ def list_workspace_access_details(
     """
     Shows a list of users (including groups and Service Principals) that have access to the specified workspace.
 
+    This is a wrapper function for the following API: `Workspaces - List Workspace Access Details <https://learn.microsoft.com/rest/api/fabric/admin/workspaces/list-workspace-access-details>`_.
+
     Parameters
     ----------
     workspace : str, default=None
@@ -745,10 +755,7 @@ def list_workspace_access_details(
         A pandas dataframe showing a list of users (including groups and Service Principals) that have access to the specified workspace.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/workspaces/list-workspace-access-details?tabs=HTTP
-
-    workspace_name = fabric.resolve_workspace_name(workspace)
-    workspace_id = fabric.resolve_workspace_id(workspace_name)
+    workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
 
     df = pd.DataFrame(
         columns=[
@@ -779,6 +786,32 @@ def list_workspace_access_details(
     return df
 
 
+def _resolve_item_id(
+    item_name: str, type: str, workspace: Optional[str] = None
+) -> UUID:
+
+    workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
+    dfI = list_items(workspace=workspace_name, type=type)
+    dfI_filt = dfI[dfI["Item Name"] == item_name]
+
+    if len(dfI_filt) == 0:
+        raise ValueError(
+            f"The '{item_name}' {type} does not exist within the '{workspace_name}' workspace."
+        )
+
+    return dfI_filt["Item Id"].iloc[0]
+
+
+def _resolve_workspace_name_and_id(workspace: str) -> Tuple[str, UUID]:
+
+    filter_condition = urllib.parse.quote(workspace)
+    dfW_filt = list_workspaces(filter=f"name eq '{filter_condition}'")
+    workspace_name = dfW_filt["Name"].iloc[0]
+    workspace_id = dfW_filt["Id"].iloc[0]
+
+    return workspace_name, workspace_id
+
+
 def list_items(
     capacity_name: Optional[str] = None,
     workspace: Optional[str] = None,
@@ -786,7 +819,9 @@ def list_items(
     type: Optional[str] = None,
 ) -> pd.DataFrame:
     """
-    Shows a list of active Fabric and PowerBI items.
+    Shows a list of active Fabric and Power BI items.
+
+    This is a wrapper function for the following API: `Items - List Items <https://learn.microsoft.com/rest/api/fabric/admin/items/list-items>`_.
 
     Parameters
     ----------
@@ -807,8 +842,6 @@ def list_items(
         A pandas dataframe showing a list of active Fabric and Power BI items.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/items/list-items?tabs=HTTP
-
     url = "/v1/admin/items?"
 
     df = pd.DataFrame(
@@ -829,8 +862,7 @@ def list_items(
     )
 
     if workspace is not None:
-        workspace = fabric.resolve_workspace_name(workspace)
-        workspace_id = fabric.resolve_workspace_id(workspace)
+        workspace_name, workspace_id = _resolve_workspace_name_and_id(workspace)
         url += f"workspaceId={workspace_id}&"
     if capacity_name is not None:
         dfC = list_capacities()
@@ -891,6 +923,8 @@ def list_activity_events(
     """
     Shows a list of audit activity events for a tenant.
 
+    This is a wrapper function for the following API: `Admin - Get Activity Events <https://learn.microsoft.com/rest/api/power-bi/admin/get-activity-events>`_.
+
     Parameters
     ----------
     start_time : str
@@ -908,7 +942,13 @@ def list_activity_events(
         A pandas dataframe showing a list of audit activity events for a tenant.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/power-bi/admin/get-activity-events
+    start_dt = datetime.strptime(start_time, "%Y-%m-%dT%H:%M:%S")
+    end_dt = datetime.strptime(end_time, "%Y-%m-%dT%H:%M:%S")
+
+    if not start_dt.date() == end_dt.date():
+        raise ValueError(
+            f"{icons.red_dot} Start and End Times must be within the same UTC day. Please refer to the documentation here: https://learn.microsoft.com/rest/api/power-bi/admin/get-activity-events#get-audit-activity-events-within-a-time-window-and-for-a-specific-activity-type-and-user-id-example"
+        )
 
     df = pd.DataFrame(
         columns=[
@@ -948,7 +988,7 @@ def list_activity_events(
         conditions.append(f"UserId{space}eq{space}{tic}{user_id_filter}{tic}")
 
     filter_value = (
-        f"&filter={f'{space}and{space}'.join(conditions)}" if conditions else ""
+        f"&$filter={f'{space}and{space}'.join(conditions)}" if conditions else ""
     )
 
     full_url = f"{base_url}?startDateTime={tic}{start_time}{tic}&endDateTime={tic}{end_time}{tic}{filter_value}"
@@ -988,4 +1028,52 @@ def list_activity_events(
                 ignore_index=True,
             )
 
+    df["Creation Time"] = pd.to_datetime(df["Creation Time"])
+
+    return df
+
+
+def list_modified_workspaces(
+    modified_since: Optional[str] = None,
+    exclude_inactive_workspaces: bool = False,
+    exclude_personal_workspaces: bool = False,
+) -> pd.DataFrame:
+    """
+    Gets a list of workspace IDs in the organization.
+
+    This is a wrapper function for the following API: `Admin - WorkspaceInfo GetModifiedWorkspaces <https://learn.microsoft.com/rest/api/power-bi/admin/workspace-info-get-modified-workspaces>`_.
+
+    Parameters
+    ----------
+    modified_since : str
+        Last modified date (must be in ISO 8601 compliant UTC format). Example: "2024-11-02T05:51:30.0000000Z".
+    exclude_inactive_workspaces : bool, default=False
+        Whether to exclude inactive workspaces.
+    exclude_personal_workspaces : bool, default=False
+        Whether to exclude personal workspaces.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of workspace IDs in the organization.
+    """
+
+    client = fabric.PowerBIRestClient()
+    url = "/v1.0/myorg/admin/workspaces/modified?"
+
+    if modified_since is not None:
+        url += f"modifiedSince={modified_since}&"
+    if exclude_inactive_workspaces:
+        url += f"excludeInActiveWorkspaces={exclude_inactive_workspaces}&"
+    if exclude_personal_workspaces:
+        url += f"excludePersonalWorkspaces={exclude_personal_workspaces}&"
+
+    url = url.rstrip("&").rstrip("?")
+
+    response = client.get(url)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    df = pd.DataFrame(response.json()).rename(columns={"id": "Workspace Id"})
+
     return df

sempy_labs/admin/_domains.py

@@ -34,6 +34,8 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
     """
     Shows a list of domains.
 
+    This is a wrapper function for the following API: `Domains - List Domains <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domains>`_.
+
     Parameters
     ----------
     non_empty_only : bool, default=False
@@ -46,8 +48,6 @@ def list_domains(non_empty_only: bool = False) -> pd.DataFrame:
         A pandas dataframe showing a list of the domains.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/list-domains?tabs=HTTP
-
     df = pd.DataFrame(
         columns=[
             "Domain ID",
@@ -84,6 +84,8 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
     """
     Shows a list of workspaces within the domain.
 
+    This is a wrapper function for the following API: `Domains - List Domain Workspaces <https://learn.microsoft.com/rest/api/fabric/admin/domains/list-domain-workspaces>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -95,8 +97,6 @@ def list_domain_workspaces(domain_name: str) -> pd.DataFrame:
         A pandas dataframe showing a list of workspaces within the domain.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/list-domain-workspaces?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name)
 
     df = pd.DataFrame(columns=["Workspace ID", "Workspace Name"])
@@ -125,6 +125,8 @@ def create_domain(
     """
     Creates a new domain.
 
+    This is a wrapper function for the following API: `Domains - Create Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/create-domain>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -135,8 +137,6 @@ def create_domain(
         The parent domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/create-domain?tabs=HTTP
-
     if parent_domain_name is not None:
         parent_domain_id = resolve_domain_id(parent_domain_name)
 
@@ -160,14 +160,14 @@ def delete_domain(domain_name: str):
     """
     Deletes a domain.
 
+    This is a wrapper function for the following API: `Domains - Delete Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/delete-domain>`_.
+
     Parameters
     ----------
     domain_name : str
         The domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/delete-domain?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name)
 
     client = fabric.FabricRestClient()
@@ -187,6 +187,8 @@ def update_domain(
     """
     Updates a domain's properties.
 
+    This is a wrapper function for the following API: `Domains - Update Domain <https://learn.microsoft.com/rest/api/fabric/admin/domains/update-domain>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -197,8 +199,6 @@ def update_domain(
         The domain `contributor scope <https://learn.microsoft.com/rest/api/fabric/admin/domains/update-domain?tabs=HTTP#contributorsscopetype>`_.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/update-domain?tabs=HTTP
-
     contributors_scopes = ["AdminsOnly", "AllTenant", "SpecificUsersAndGroups"]
 
     if contributors_scope not in contributors_scopes:
@@ -231,6 +231,8 @@ def assign_domain_workspaces_by_capacities(
     """
     Assigns all workspaces that reside on the specified capacities to the specified domain.
 
+    This is a wrapper function for the following API: `Domains - Assign Domain Workspaces By Capacities <https://learn.microsoft.com/rest/api/fabric/admin/domains/assign-domain-workspaces-by-capacities>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -239,8 +241,6 @@ def assign_domain_workspaces_by_capacities(
         The capacity names.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/assign-domain-workspaces-by-capacities?tabs=HTTP
-
     from sempy_labs.admin import list_capacities
 
     domain_id = resolve_domain_id(domain_name)
@@ -287,6 +287,8 @@ def assign_domain_workspaces(domain_name: str, workspace_names: str | List[str])
     """
     Assigns workspaces to the specified domain by workspace.
 
+    This is a wrapper function for the following API: `Domains - Assign Domain Workspaces By Ids <https://learn.microsoft.com/rest/api/fabric/admin/domains/assign-domain-workspaces-by-ids>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -295,8 +297,6 @@ def assign_domain_workspaces(domain_name: str, workspace_names: str | List[str])
         The Fabric workspace(s).
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/assign-domain-workspaces-by-ids?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     if isinstance(workspace_names, str):
@@ -340,14 +340,14 @@ def unassign_all_domain_workspaces(domain_name: str):
     """
     Unassigns all workspaces from the specified domain.
 
+    This is a wrapper function for the following API: `Domains - Unassign All Domain Workspaces <https://learn.microsoft.com/rest/api/fabric/admin/domains/unassign-all-domain-workspaces>`_.
+
     Parameters
     ----------
     domain_name : str
         The domain name.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/unassign-all-domain-workspaces?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     client = fabric.FabricRestClient()
@@ -364,6 +364,8 @@ def unassign_domain_workspaces(domain_name: str, workspace_names: str | List[str
     """
     Unassigns workspaces from the specified domain by workspace.
 
+    This is a wrapper function for the following API: `Domains - Unassign Domain Workspaces By Ids <https://learn.microsoft.com/rest/api/fabric/admin/domains/unassign-domain-workspaces-by-ids>`_.
+
     Parameters
     ----------
     domain_name : str
@@ -372,8 +374,6 @@ def unassign_domain_workspaces(domain_name: str, workspace_names: str | List[str
         The Fabric workspace(s).
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/fabric/admin/domains/unassign-domain-workspaces-by-ids?tabs=HTTP
-
     domain_id = resolve_domain_id(domain_name=domain_name)
 
     if isinstance(workspace_names, str):

sempy_labs/directlake/__init__.py

@@ -21,6 +21,7 @@ from sempy_labs.directlake._show_unsupported_directlake_objects import (
 )
 from sempy_labs.directlake._update_directlake_model_lakehouse_connection import (
     update_direct_lake_model_lakehouse_connection,
+    update_direct_lake_model_connection,
 )
 from sempy_labs.directlake._update_directlake_partition_entity import (
     update_direct_lake_partition_entity,
@@ -50,4 +51,5 @@ __all__ = [
     "add_table_to_direct_lake_semantic_model",
     "generate_direct_lake_semantic_model",
     "get_direct_lake_source",
+    "update_direct_lake_model_connection",
 ]

sempy_labs/directlake/_directlake_schema_sync.py

@@ -3,6 +3,7 @@ import sempy.fabric as fabric
 from sempy_labs.lakehouse import get_lakehouse_columns
 from sempy_labs.directlake._dl_helper import get_direct_lake_source
 from sempy_labs.tom import connect_semantic_model
+from sempy_labs._helper_functions import _convert_data_type
 from typing import Optional
 from sempy._utils._log import log
 import sempy_labs._icons as icons
@@ -88,7 +89,7 @@ def direct_lake_schema_sync(
                         f"{icons.yellow_dot} The '{lakeCName}' column exists in the '{lakeTName}' lakehouse table but not in the '{dataset}' semantic model within the '{workspace}' workspace."
                     )
                     if add_to_model:
-                        dt = icons.data_type_mapping.get(dType)
+                        dt = _convert_data_type(dType)
                         tom.add_data_column(
                             table_name=table_name,
                             column_name=lakeCName,

sempy_labs/directlake/_dl_helper.py

@@ -9,6 +9,7 @@ from sempy_labs._helper_functions import (
     retry,
     resolve_dataset_id,
     resolve_lakehouse_name,
+    _convert_data_type,
 )
 
 
@@ -130,6 +131,8 @@ def generate_direct_lake_semantic_model(
 
     dfLT = get_lakehouse_tables(lakehouse=lakehouse, workspace=lakehouse_workspace)
 
+    icons.sll_tags.append("GenerateDLModel")
+
     # Validate lakehouse tables
     for t in lakehouse_tables:
         if t not in dfLT["Table Name"].values:
@@ -178,7 +181,7 @@ def generate_direct_lake_semantic_model(
             for i, r in dfLC_filt.iterrows():
                 lakeCName = r["Column Name"]
                 dType = r["Data Type"]
-                dt = icons.data_type_mapping.get(dType)
+                dt = _convert_data_type(dType)
                 tom.add_data_column(
                     table_name=t,
                     column_name=lakeCName,

sempy_labs/directlake/_get_shared_expression.py

@@ -30,7 +30,7 @@ def get_shared_expression(
     workspace = fabric.resolve_workspace_name(workspace)
     if lakehouse is None:
         lakehouse_id = fabric.get_lakehouse_id()
-        lakehouse = resolve_lakehouse_name(lakehouse_id)
+        lakehouse = resolve_lakehouse_name(lakehouse_id, workspace)
 
     dfL = list_lakehouses(workspace=workspace)
     lakeDetail = dfL[dfL["Lakehouse Name"] == lakehouse]
@@ -39,6 +39,12 @@ def get_shared_expression(
     sqlepid = lakeDetail["SQL Endpoint ID"].iloc[0]
     provStatus = lakeDetail["SQL Endpoint Provisioning Status"].iloc[0]
 
+    parts = sqlEPCS.split(".", 1)
+    if parts:
+        parts[0] = parts[0].upper()
+
+    sqlEPCS = ".".join(parts)
+
     if provStatus == "InProgress":
         raise ValueError(
             f"{icons.red_dot} The SQL Endpoint for the '{lakehouse}' lakehouse within the '{workspace}' workspace has not yet been provisioned. Please wait until it has been provisioned."

sempy_labs/directlake/_guardrails.py

@@ -81,6 +81,7 @@ def get_directlake_guardrails_for_sku(sku_size: str) -> pd.DataFrame:
     """
 
     df = get_direct_lake_guardrails()
-    filtered_df = df[df["Fabric SKUs"] == sku_size]
+    col_name = df.columns[0]
+    filtered_df = df[df[col_name] == sku_size]
 
     return filtered_df

sempy_labs/directlake/_show_unsupported_directlake_objects.py

@@ -59,13 +59,7 @@ def show_unsupported_direct_lake_objects(
     dfR["From Column Data Type"] = merged_from["Data Type"]
     dfR["To Column Data Type"] = merged_to["Data Type"]
 
-    dfR_filt = dfR[
-        (
-            (dfR["From Column Data Type"] == "DateTime")
-            | (dfR["To Column Data Type"] == "DateTime")
-        )
-        | (dfR["From Column Data Type"] != dfR["To Column Data Type"])
-    ]
+    dfR_filt = dfR[(dfR["From Column Data Type"] != dfR["To Column Data Type"])]
     r = dfR_filt[
         [
             "From Table",