semantic-link-labs 0.4.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

{semantic_link_labs-0.4.1.dist-info → semantic_link_labs-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: semantic-link-labs
-Version: 0.4.1
+Version: 0.5.0
 Summary: Semantic Link Labs project
 Author: Microsoft Corporation
 License: MIT License

semantic_link_labs-0.5.0.dist-info/RECORD

@@ -0,0 +1,53 @@
+sempy_labs/__init__.py,sha256=Xpq66PKo8KK7QIP0QYme_X_FyfZnpOBsli2V4bnMNYo,4843
+sempy_labs/_ai.py,sha256=WNCLh8wzZ7brXJWe2CNC79D9NHeS-2KpAWCIAlvZD7U,17784
+sempy_labs/_clear_cache.py,sha256=ELHcD4smBS3EiHEO1Ux97yV_Q7j_zB8TJJ4-kS1ylCU,1394
+sempy_labs/_connections.py,sha256=ghYtHBLTaTmYwgllh6I3zfFjG5lcxM2BAQR52B3z-t0,7795
+sempy_labs/_dax.py,sha256=u4qVxsu2dVaOJmso-ErScNZ5yI4lGQTlon_jmrAzvGs,2148
+sempy_labs/_generate_semantic_model.py,sha256=rBi1jmLQJNq9NOt56AqqYzAXBJ2DX2bpG7FYrHXyiEA,9180
+sempy_labs/_helper_functions.py,sha256=RlLtpEmBpoX285sSFKBBjkPfXqp8_FSIwYcoR4rZG58,14251
+sempy_labs/_icons.py,sha256=WkmhtLcQPO1PlcwFklb253dBLpwGUyXLwxn9_-nu3s0,215
+sempy_labs/_list_functions.py,sha256=Gc49_Q5e9CmdsVntvq0wFv1qdGInhhejxcvn-Vokyds,80791
+sempy_labs/_model_auto_build.py,sha256=SGz3ASfKJBpYdgFRPoofiU7kpsjPDBkeB3qCntKb6qs,5083
+sempy_labs/_model_bpa.py,sha256=xEp5LdqoMLBsUDzGWnfNrFCwr9tz9-2QDSi0NwnvXFI,63191
+sempy_labs/_model_dependencies.py,sha256=S8u0f7AAVL6Zk1Jm35EqbLsIEGD8KwBGc_kH1AlBI1A,12948
+sempy_labs/_one_lake_integration.py,sha256=XlGKghnYtXIprUdUI5fQj0ddshxE_AvUowb9YIrL2CE,6184
+sempy_labs/_query_scale_out.py,sha256=xkyCFz7vchxB6c6cMIhZXUWifOduanSbv1KGQbkmVls,15214
+sempy_labs/_refresh_semantic_model.py,sha256=bTVUNEdQiJDxQh1T0g4aYU3-VHaN9__qFAOYbx85-O0,6622
+sempy_labs/_translations.py,sha256=sH_-W8vdQ632lY68RxM5a2lCmy4MRDFXLxjvMuw4xQg,18054
+sempy_labs/_vertipaq.py,sha256=04hQ-A4wuW6gi4zIccmVoDDByk34Rp-QTkRXSo6XLfI,33266
+sempy_labs/directlake/__init__.py,sha256=HbfHvDvGE4H-xSbV6JO7TUb4HoLGJf2AeuqeQxIuuJ4,1689
+sempy_labs/directlake/_directlake_schema_compare.py,sha256=axqSYQHMs3zAhFAF3DMNdm2SK1k-95j_Zh-pZZXv8SQ,4636
+sempy_labs/directlake/_directlake_schema_sync.py,sha256=EMDPAVn53EN1PM769AxKspb_cVDcCazz4kHMKvWqJMQ,5119
+sempy_labs/directlake/_fallback.py,sha256=2dX5MRU2d04_jA799TaPezSQJWW-NrflDtdURge6Ceo,1995
+sempy_labs/directlake/_get_directlake_lakehouse.py,sha256=vEY1QBU7gAdoVQIGl59T_1TTYuXbHgk3pZA41EkVkl8,2358
+sempy_labs/directlake/_get_shared_expression.py,sha256=ngZCnoOjj278n6Yql7TVZ36z89HDej9JSRuoRWALxDs,1926
+sempy_labs/directlake/_guardrails.py,sha256=iReOycR6caBuBWpEOIpicZS_7kkSGJLTUDf2Ch0QUCE,2280
+sempy_labs/directlake/_list_directlake_model_calc_tables.py,sha256=C6C8WycGV4NQOQifh3XexBZx9hm30-Ac56sCo4MfqgI,2082
+sempy_labs/directlake/_show_unsupported_directlake_objects.py,sha256=oumzh7mXghkFFqFDDK7bBmNMRWnOZeE1DxuMsEPSBzo,3365
+sempy_labs/directlake/_update_directlake_model_lakehouse_connection.py,sha256=lY4JlbKwow664sXK4cZ22PFTy9Gw79R-6TYx36uz8SY,3183
+sempy_labs/directlake/_update_directlake_partition_entity.py,sha256=KzUvP4r0-UOylXgM46cVKoUlbIFEpxO3ThMQl9JU-Gw,3140
+sempy_labs/directlake/_warm_cache.py,sha256=codsJhifbn8yO8Cjo40syRtFnIArIvpWQ726WZ88ZsQ,8211
+sempy_labs/lakehouse/__init__.py,sha256=i6VRx4dR1SIN-1GxioiNwhC4FxbozRCIz5TfXjb9rKc,587
+sempy_labs/lakehouse/_get_lakehouse_columns.py,sha256=QGmuH3uFuxv_mIjm4HTXaX-s1UFb9BbHnaCx9kTHgy8,2594
+sempy_labs/lakehouse/_get_lakehouse_tables.py,sha256=g_kKtntm5uPbIxNrunOVhzuKvJfC-9g5awxul2KSmUI,9085
+sempy_labs/lakehouse/_lakehouse.py,sha256=ovm5S4hk3aLChFCzUGWIJmL3wJ47wRycoh0mbBd8pRA,2774
+sempy_labs/lakehouse/_shortcuts.py,sha256=voKile93krzWK7ccmKVk2_fv1lioyAq5rk8YrJ6qy6k,6895
+sempy_labs/migration/__init__.py,sha256=l5v8pC2INdNwbAKVmvWpuVxs6lpb6omim_4BPOmNo4E,1042
+sempy_labs/migration/_create_pqt_file.py,sha256=PGC21nhIsE9TKyz110tYaf4Nle6-eV1xqvBOqUSEDQY,8986
+sempy_labs/migration/_migrate_calctables_to_lakehouse.py,sha256=ajFvHauFdEsbgxvn9JXj2kiXaRtJLEjwX4hWQG7FQy0,20609
+sempy_labs/migration/_migrate_calctables_to_semantic_model.py,sha256=Co2f579vSwkWZ95SlBStS-XJ73YwgcdfAMlJbUv_pkk,6343
+sempy_labs/migration/_migrate_model_objects_to_semantic_model.py,sha256=q4oYBDHXvcTExeufmcOOeK3jv_9A2Xef5ksyEwG0PfA,23801
+sempy_labs/migration/_migrate_tables_columns_to_semantic_model.py,sha256=qohJC6ARjM8NiMH7nZEKqUEXMrh-IqhdeUzgrBNZ1DQ,7028
+sempy_labs/migration/_migration_validation.py,sha256=R7xz_OFmYRO4fxFWxvdl_ORZQKzqflsSBnGapmS067c,2508
+sempy_labs/migration/_refresh_calc_tables.py,sha256=VumJaTiA3bTfm8jWwyIl7gxW4-a7W_3auGjWRcvd65g,6043
+sempy_labs/report/__init__.py,sha256=fkjbkAXZuH7VnAn-k3iB4dngWZKaX-k0bxS6mBa9iAs,846
+sempy_labs/report/_generate_report.py,sha256=uFbTO_7GSItt5wM_yZrKybEQ3kQoqXKbY1bGzlOW5c0,8502
+sempy_labs/report/_report_functions.py,sha256=5rzYaS_8Gj9FEyCgysWIsLxiE-DaCgPNArI9YIe4Jvo,29515
+sempy_labs/report/_report_rebind.py,sha256=hsXXOl6C06hH-SU_TaKBInjIXHf-uRZru7uCHpbRJYA,4756
+sempy_labs/tom/__init__.py,sha256=hFwkmWk5AZ7GK1LWqoqaK1g4gDmu9mZMkfLQvLsR_eE,130
+sempy_labs/tom/_model.py,sha256=a3EXOj_yXxl0aQQfWa5bolC2q-KdCgQMF8CrrQ6aPrU,136859
+semantic_link_labs-0.5.0.dist-info/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+semantic_link_labs-0.5.0.dist-info/METADATA,sha256=TINv_gW59FHx2uvyUwYSH9MxKls6eO7XYji03ab-kYk,764
+semantic_link_labs-0.5.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+semantic_link_labs-0.5.0.dist-info/top_level.txt,sha256=kiQX1y42Dbein1l3Q8jMUYyRulDjdlc2tMepvtrvixQ,11
+semantic_link_labs-0.5.0.dist-info/RECORD,,

{semantic_link_labs-0.4.1.dist-info → semantic_link_labs-0.5.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

sempy_labs/__init__.py

@@ -5,7 +5,7 @@ from sempy_labs._clear_cache import clear_cache
 # create_connection_vnet,
 # create_connection_on_prem
 # )
-from sempy_labs._dax import run_dax
+from sempy_labs._dax import evaluate_dax_impersonation
 from sempy_labs._generate_semantic_model import (
     create_blank_semantic_model,
     create_semantic_model_from_bim,
@@ -13,6 +13,8 @@ from sempy_labs._generate_semantic_model import (
     get_semantic_model_bim,
 )
 from sempy_labs._list_functions import (
+    list_semantic_model_objects,
+    list_shortcuts,
     get_object_level_security,
     # list_annotations,
     # list_columns,
@@ -32,9 +34,21 @@ from sempy_labs._list_functions import (
     # list_sqlendpoints,
     # list_tables,
     list_warehouses,
-    # list_workspace_role_assignments,
+    list_workspace_role_assignments,
     create_warehouse,
     update_item,
+    list_custom_pools,
+    create_custom_pool,
+    update_custom_pool,
+    assign_workspace_to_capacity,
+    unassign_workspace_from_capacity,
+    get_spark_settings,
+    update_spark_settings,
+    add_user_to_workspace,
+    delete_user_from_workspace,
+    update_workspace_user,
+    list_workspace_users,
+    assign_workspace_to_dataflow_storage,
 )
 
 from sempy_labs._helper_functions import (
@@ -50,9 +64,9 @@ from sempy_labs._helper_functions import (
     resolve_dataset_name,
     resolve_report_id,
     resolve_report_name,
-    # language_validate
+    #  language_validate
 )
-from sempy_labs._model_auto_build import model_auto_build
+#  from sempy_labs._model_auto_build import model_auto_build
 from sempy_labs._model_bpa import model_bpa_rules, run_model_bpa
 from sempy_labs._model_dependencies import (
     measure_dependency_tree,
@@ -62,16 +76,15 @@ from sempy_labs._model_dependencies import (
 from sempy_labs._one_lake_integration import (
     export_model_to_onelake,
 )
-
-# from sempy_labs._query_scale_out import (
-#    qso_sync,
-#    qso_sync_status,
-#    set_qso,
-#    list_qso_settings,
-#    disable_qso,
-#    set_semantic_model_storage_format,
-#    set_workspace_default_storage_format,
-# )
+from sempy_labs._query_scale_out import (
+    qso_sync,
+    qso_sync_status,
+    set_qso,
+    list_qso_settings,
+    disable_qso,
+    set_semantic_model_storage_format,
+    set_workspace_default_storage_format,
+)
 from sempy_labs._refresh_semantic_model import (
     refresh_semantic_model,
     cancel_dataset_refresh,
@@ -82,14 +95,13 @@ from sempy_labs._vertipaq import (
     # visualize_vertipaq,
     import_vertipaq_analyzer,
 )
-from sempy_labs._tom import TOMWrapper, connect_semantic_model
 
 __all__ = [
     "clear_cache",
     # create_connection_cloud,
     # create_connection_vnet,
     # create_connection_on_prem,
-    "run_dax",
+    "evaluate_dax_impersonation",
     "create_blank_semantic_model",
     "create_semantic_model_from_bim",
     #'deploy_semantic_model',
@@ -113,7 +125,7 @@ __all__ = [
     #'list_sqlendpoints',
     #'list_tables',
     "list_warehouses",
-    #'list_workspace_role_assignments',
+    'list_workspace_role_assignments',
     "create_warehouse",
     "update_item",
     "create_abfss_path",
@@ -129,26 +141,38 @@ __all__ = [
     "resolve_report_id",
     "resolve_report_name",
     #'language_validate',
-    "model_auto_build",
+    #"model_auto_build",
     "model_bpa_rules",
     "run_model_bpa",
     "measure_dependency_tree",
     "get_measure_dependencies",
     "get_model_calc_dependencies",
     "export_model_to_onelake",
-    #'qso_sync',
-    #'qso_sync_status',
-    #'set_qso',
-    #'list_qso_settings',
-    #'disable_qso',
-    #'set_semantic_model_storage_format',
-    #'set_workspace_default_storage_format',
+    'qso_sync',
+    'qso_sync_status',
+    'set_qso',
+    'list_qso_settings',
+    'disable_qso',
+    'set_semantic_model_storage_format',
+    'set_workspace_default_storage_format',
     "refresh_semantic_model",
     "cancel_dataset_refresh",
     "translate_semantic_model",
     "vertipaq_analyzer",
     #'visualize_vertipaq',
     "import_vertipaq_analyzer",
-    "TOMWrapper",
-    "connect_semantic_model",
+    "list_semantic_model_objects",
+    "list_shortcuts",
+    "list_custom_pools",
+    "create_custom_pool",
+    "update_custom_pool",
+    "assign_workspace_to_capacity",
+    "unassign_workspace_from_capacity",
+    "get_spark_settings",
+    "update_spark_settings",
+    "add_user_to_workspace",
+    "delete_user_from_workspace",
+    "update_workspace_user",
+    "list_workspace_users",
+    "assign_workspace_to_dataflow_storage"
 ]

sempy_labs/_ai.py

@@ -6,6 +6,7 @@ from pyspark.sql.functions import col
 from pyspark.sql import SparkSession
 from typing import List, Optional, Union
 from IPython.display import display
+import sempy_labs._icons as icons
 
 
 def optimize_semantic_model(dataset: str, workspace: Optional[str] = None):
@@ -13,6 +14,7 @@ def optimize_semantic_model(dataset: str, workspace: Optional[str] = None):
     from ._model_bpa import run_model_bpa
     from .directlake._fallback import check_fallback_reason
     from ._helper_functions import format_dax_object_name
+    from sempy_labs.tom import connect_semantic_model
 
     modelBPA = run_model_bpa(
         dataset=dataset, workspace=workspace, return_dataframe=True
@@ -39,7 +41,7 @@ def optimize_semantic_model(dataset: str, workspace: Optional[str] = None):
 
         if len(fallback_filt) > 0:
             print(
-                f"The '{dataset}' semantic model is a Direct Lake semantic model which contains views. Since views always fall back to DirectQuery, it is recommended to only use lakehouse tables and not views."
+                f"{icons.yellow_dot} The '{dataset}' semantic model is a Direct Lake semantic model which contains views. Since views always fall back to DirectQuery, it is recommended to only use lakehouse tables and not views."
             )
 
     # Potential model reduction estimate
@@ -56,11 +58,11 @@ def optimize_semantic_model(dataset: str, workspace: Optional[str] = None):
         totSize = df["Total Size"].sum()
         if len(df_filt) > 0:
             print(
-                f"Potential savings of {totSize} bytes from following the '{rule}' rule."
+                f"{icons.yellow_dot} Potential savings of {totSize} bytes from following the '{rule}' rule."
             )
             display(df_filt)
         else:
-            print(f"The '{rule}' rule has been followed.")
+            print(f"{icons.green_dot} The '{rule}' rule has been followed.")
 
 
 def generate_measure_descriptions(
@@ -77,10 +79,7 @@ def generate_measure_descriptions(
 
     validModels = ["gpt-35-turbo", "gpt-35-turbo-16k", "gpt-4"]
     if gpt_model not in validModels:
-        print(
-            f"The '{gpt_model}' model is not a valid model. Enter a gpt_model from this list: {validModels}."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} The '{gpt_model}' model is not a valid model. Enter a gpt_model from this list: {validModels}.")
 
     dfM = fabric.list_measures(dataset=dataset, workspace=workspace)
 
@@ -92,7 +91,7 @@ def generate_measure_descriptions(
     df = dfM_filt[["Table Name", "Measure Name", "Measure Expression"]]
 
     df["prompt"] = (
-        f"The following is DAX code used by Microsoft Power BI. Please explain this code in simple terms:"
+        "The following is DAX code used by Microsoft Power BI. Please explain this code in simple terms:"
         + df["Measure Expression"]
     )
 
@@ -115,8 +114,8 @@ def generate_measure_descriptions(
     )
 
     # Update the model to use the new descriptions
-    tom_server = fabric.create_tom_server(readonly=False, workspace=workspace)
-    m = tom_server.Databases.GetByName(dataset).Model
+    #with connect_semantic_model(dataset=dataset, workspace=workspace, readonly=False) as tom:
+    
 
     # for t in m.Tables:
     # tName = t.Name
@@ -152,11 +151,11 @@ def generate_aggs(
     #'OrderDateKey': 'GroupBy'
     # }
 
-    if workspace == None:
+    if workspace is None:
         workspace_id = fabric.get_workspace_id()
         workspace = fabric.resolve_workspace_name(workspace_id)
 
-    if lakehouse_workspace == None:
+    if lakehouse_workspace is None:
         lakehouse_workspace = workspace
         lakehouse_workspace_id = workspace_id
     else:
@@ -172,48 +171,33 @@ def generate_aggs(
     numericTypes = ["Int64", "Double", "Decimal"]
 
     if any(value not in aggTypes for value in columns.values()):
-        print(
-            f"Invalid aggregation type(s) have been specified in the 'columns' parameter. Valid aggregation types: {aggTypes}."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} Invalid aggregation type(s) have been specified in the 'columns' parameter. Valid aggregation types: {aggTypes}.")
 
     dfC = fabric.list_columns(dataset=dataset, workspace=workspace)
     dfP = fabric.list_partitions(dataset=dataset, workspace=workspace)
     dfM = fabric.list_measures(dataset=dataset, workspace=workspace)
     dfR = fabric.list_relationships(dataset=dataset, workspace=workspace)
     if not any(r["Mode"] == "DirectLake" for i, r in dfP.iterrows()):
-        print(
-            f"The '{dataset}' semantic model within the '{workspace}' workspace is not in Direct Lake mode. This function is only relevant for Direct Lake semantic models."
-        )
-        return
-
+        raise ValueError(f"{icons.red_dot} The '{dataset}' semantic model within the '{workspace}' workspace is not in Direct Lake mode. This function is only relevant for Direct Lake semantic models.")
+        
     dfC_filtT = dfC[dfC["Table Name"] == table_name]
 
     if len(dfC_filtT) == 0:
-        print(
-            f"The '{table_name}' table does not exist in the '{dataset}' semantic model within the '{workspace}' workspace."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} The '{table_name}' table does not exist in the '{dataset}' semantic model within the '{workspace}' workspace.")
 
     dfC_filt = dfC[
         (dfC["Table Name"] == table_name) & (dfC["Column Name"].isin(columnValues))
     ]
 
     if len(columns) != len(dfC_filt):
-        print(
-            f"Columns listed in '{columnValues}' do not exist in the '{table_name}' table in the '{dataset}' semantic model within the '{workspace}' workspace."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} Columns listed in '{columnValues}' do not exist in the '{table_name}' table in the '{dataset}' semantic model within the '{workspace}' workspace.")
 
     # Check if doing sum/count/min/max etc. on a non-number column
     for col, agg in columns.items():
         dfC_col = dfC_filt[dfC_filt["Column Name"] == col]
         dataType = dfC_col["Data Type"].iloc[0]
         if agg in aggTypesAggregate and dataType not in numericTypes:
-            print(
-                f"The '{col}' column in the '{table_name}' table is of '{dataType}' data type. Only columns of '{numericTypes}' data types can be aggregated as '{aggTypesAggregate}' aggregation types."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{col}' column in the '{table_name}' table is of '{dataType}' data type. Only columns of '{numericTypes}' data types can be aggregated as '{aggTypesAggregate}' aggregation types.")            
 
     # Create/update lakehouse delta agg table
     aggSuffix = "_agg"
@@ -229,10 +213,7 @@ def generate_aggs(
     dfI_filt = dfI[(dfI["Id"] == sqlEndpointId)]
 
     if len(dfI_filt) == 0:
-        print(
-            f"The lakehouse (SQL Endpoint) used by the '{dataset}' semantic model does not reside in the '{lakehouse_workspace}' workspace. Please update the lakehouse_workspace parameter."
-        )
-        return
+        raise ValueError(f"{icons.red_dot} The lakehouse (SQL Endpoint) used by the '{dataset}' semantic model does not reside in the '{lakehouse_workspace}' workspace. Please update the lakehouse_workspace parameter.")
 
     lakehouseName = dfI_filt["Display Name"].iloc[0]
     lakehouse_id = resolve_lakehouse_id(
@@ -278,16 +259,16 @@ def generate_aggs(
         delta_table_name=aggLakeTName,
     )
     spark_df.write.mode("overwrite").format("delta").save(aggFilePath)
-    f"The '{aggLakeTName}' table has been created/updated in the lakehouse."
+    f"{icons.green_dot} The '{aggLakeTName}' table has been created/updated in the lakehouse."
 
     # Create/update semantic model agg table
     tom_server = fabric.create_tom_server(readonly=False, workspace=workspace)
     m = tom_server.Databases.GetByName(dataset).Model
-    f"\nUpdating the '{dataset}' semantic model..."
+    print(f"\n{icons.in_progress} Updating the '{dataset}' semantic model...")
     dfC_agg = dfC[dfC["Table Name"] == aggTableName]
 
     if len(dfC_agg) == 0:
-        print(f"Creating the '{aggTableName}' table...")
+        print(f"{icons.in_progress} Creating the '{aggTableName}' table...")
         exp = m.Expressions["DatabaseQuery"]
         tbl = TOM.Table()
         tbl.Name = aggTableName
@@ -318,15 +299,15 @@ def generate_aggs(
 
             tbl.Columns.Add(col)
             print(
-                f"The '{aggTableName}'[{cName}] column has been added to the '{dataset}' semantic model."
+                f"{icons.green_dot} The '{aggTableName}'[{cName}] column has been added to the '{dataset}' semantic model."
             )
 
         m.Tables.Add(tbl)
         print(
-            f"The '{aggTableName}' table has been added to the '{dataset}' semantic model."
+            f"{icons.green_dot} The '{aggTableName}' table has been added to the '{dataset}' semantic model."
         )
     else:
-        print(f"Updating the '{aggTableName}' table's columns...")
+        print(f"{icons.in_progress} Updating the '{aggTableName}' table's columns...")
         # Remove existing columns
         for t in m.Tables:
             tName = t.Name
@@ -347,12 +328,12 @@ def generate_aggs(
             col.DataType = System.Enum.Parse(TOM.DataType, dType)
 
             m.Tables[aggTableName].Columns.Add(col)
-            print(f"The '{aggTableName}'[{cName}] column has been added.")
+            print(f"{icons.green_dot} The '{aggTableName}'[{cName}] column has been added.")
 
     # Create relationships
     relMap = {"m": "Many", "1": "One", "0": "None"}
 
-    print(f"\nGenerating necessary relationships...")
+    print(f"\n{icons.in_progress} Generating necessary relationships...")
     for i, r in dfR.iterrows():
         fromTable = r["From Table"]
         fromColumn = r["From Column"]
@@ -384,27 +365,27 @@ def generate_aggs(
                 rel.FromColumn = m.Tables[aggTableName].Columns[fromColumn]
                 m.Relationships.Add(rel)
                 print(
-                    f"'{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has been added."
+                    f"{icons.green_dot} '{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has been added."
                 )
             except:
                 print(
-                    f"'{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has not been created."
+                    f"{icons.red_dot} '{aggTableName}'[{fromColumn}] -> '{toTable}'[{toColumn}] relationship has not been created."
                 )
         elif toTable == table_name:
             try:
                 rel.ToColumn = m.Tables[aggTableName].Columns[toColumn]
                 m.Relationships.Add(rel)
                 print(
-                    f"'{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has been added."
+                    f"{icons.green_dot} '{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has been added."
                 )
             except:
                 print(
-                    f"'{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has not been created."
+                    f"{icons.red_dot} '{fromTable}'[{fromColumn}] -> '{aggTableName}'[{toColumn}] relationship has not been created."
                 )
     f"Relationship creation is complete."
 
     # Create IF measure
-    f"\nCreating measure to check if the agg table can be used..."
+    f"\n{icons.in_progress} Creating measure to check if the agg table can be used..."
     aggChecker = "IF("
     dfR_filt = dfR[
         (dfR["From Table"] == table_name) & (~dfR["From Column"].isin(columnValues))
@@ -419,7 +400,7 @@ def generate_aggs(
     print(aggChecker)
 
     # Todo: add IFISFILTERED clause for columns
-    f"\n Creating the base measures in the agg table..."
+    f"\n{icons.in_progress} Creating the base measures in the agg table..."
     # Create base agg measures
     dep = fabric.evaluate_dax(
         dataset=dataset,

sempy_labs/_clear_cache.py

@@ -8,6 +8,7 @@ import sempy_labs._icons as icons
 def clear_cache(dataset: str, workspace: Optional[str] = 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
     ----------
@@ -19,9 +20,7 @@ def clear_cache(dataset: str, workspace: Optional[str] = None):
         or if no lakehouse attached, resolves to the workspace of the notebook.
     """
 
-    if workspace == None:
-        workspace_id = fabric.get_workspace_id()
-        workspace = fabric.resolve_workspace_name(workspace_id)
+    workspace = fabric.resolve_workspace_name(workspace)
 
     datasetID = resolve_dataset_id(dataset=dataset, workspace=workspace)
 

sempy_labs/_connections.py

@@ -2,6 +2,7 @@ import sempy
 import sempy.fabric as fabric
 import pandas as pd
 from typing import List, Optional, Union
+import sempy_labs._icons as icons
 
 
 def create_connection_cloud(
@@ -11,7 +12,7 @@ def create_connection_cloud(
     user_name: str,
     password: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     # https://review.learn.microsoft.com/en-us/rest/api/fabric/core/connections/create-connection?branch=features%2Fdmts&tabs=HTTP
 
@@ -60,16 +61,16 @@ def create_connection_cloud(
     if response.status_code == 200:
         o = response.json()
         new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
+            "Connection Id": o.get("id"),
+            "Connection Name": o.get("name"),
+            "Connectivity Type": o.get("connectivityType"),
+            "Connection Type": o.get("connectionDetails",{}).get("type"),
+            "Connection Path": o.get("connectionDetails",{}).get("path"),
+            "Privacy Level": o.get("privacyLevel"),
+            "Credential Type": o.get("credentialDetails",{}).get("credentialType"),
+            "Single Sign On Type": o.get("credentialDetails",{}).get("singleSignOnType"),
+            "Connection Encryption": o.get("credentialDetails",{}).get("connectionEncryption"),
+            "Skip Test Connection": o.get("credentialDetails",{}).get("skipTestConnection"),
         }
         df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
@@ -77,7 +78,7 @@ def create_connection_cloud(
 
         return df
     else:
-        print(response.status_code)
+        print(f"{icons.red_dot} {response.status_code}")
 
 
 def create_connection_on_prem(
@@ -87,7 +88,7 @@ def create_connection_on_prem(
     database_name: str,
     credentials: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     df = pd.DataFrame(
         columns=[
@@ -135,17 +136,17 @@ def create_connection_on_prem(
     if response.status_code == 200:
         o = response.json()
         new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Gateway ID": o["gatewayId"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
+            "Connection Id": o.get("id"),
+            "Connection Name": o.get("name"),
+            "Gateway ID": o.get("gatewayId"),
+            "Connectivity Type": o.get("connectivityType"),
+            "Connection Type": o.get("connectionDetails",{}).get("type"),
+            "Connection Path": o.get("connectionDetails",{}).get("path"),
+            "Privacy Level": o.get("privacyLevel"),
+            "Credential Type": o.get("credentialDetails",{}).get("credentialType"),
+            "Single Sign On Type": o.get("credentialDetails",{}).get("singleSignOnType"),
+            "Connection Encryption": o.get("credentialDetails",{}).get("connectionEncryption"),
+            "Skip Test Connection": o.get("credentialDetails",{}).get("skipTestConnection"),
         }
         df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
@@ -153,7 +154,7 @@ def create_connection_on_prem(
 
         return df
     else:
-        print(response.status_code)
+        print(f"{icons.red_dot} {response.status_code}")
 
 
 def create_connection_vnet(
@@ -164,7 +165,7 @@ def create_connection_vnet(
     user_name: str,
     password: str,
     privacy_level: str,
-):
+) -> pd.DataFrame:
 
     df = pd.DataFrame(
         columns=[
@@ -213,17 +214,17 @@ def create_connection_vnet(
     if response.status_code == 200:
         o = response.json()
         new_data = {
-            "Connection Id": o["id"],
-            "Connection Name": o["name"],
-            "Gateway ID": o["gatewayId"],
-            "Connectivity Type": o["connectivityType"],
-            "Connection Type": o["connectionDetails"]["type"],
-            "Connection Path": o["connectionDetails"]["path"],
-            "Privacy Level": o["privacyLevel"],
-            "Credential Type": o["credentialDetails"]["credentialType"],
-            "Single Sign On Type": o["credentialDetails"]["singleSignOnType"],
-            "Connection Encryption": o["credentialDetails"]["connectionEncryption"],
-            "Skip Test Connection": o["credentialDetails"]["skipTestConnection"],
+            "Connection Id": o.get("id"),
+            "Connection Name": o.get("name"),
+            "Gateway ID": o.get("gatewayId"),
+            "Connectivity Type": o.get("connectivityType"),
+            "Connection Type": o.get("connectionDetails",{}).get("type"),
+            "Connection Path": o.get("connectionDetails",{}).get("path"),
+            "Privacy Level": o.get("privacyLevel"),
+            "Credential Type": o.get("credentialDetails",{}).get("credentialType"),
+            "Single Sign On Type": o.get("credentialDetails",{}).get("singleSignOnType"),
+            "Connection Encryption": o.get("credentialDetails",{}).get("connectionEncryption"),
+            "Skip Test Connection": o.get("credentialDetails",{}).get("skipTestConnection"),
         }
         df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
 
@@ -231,4 +232,4 @@ def create_connection_vnet(
 
         return df
     else:
-        print(response.status_code)
+        print(f"{icons.red_dot} {response.status_code}")