semantic-link-labs 0.8.8__py3-none-any.whl → 0.8.10__py3-none-any.whl

{semantic_link_labs-0.8.8.dist-info → semantic_link_labs-0.8.10.dist-info}/METADATA

@@ -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.dist-info → semantic_link_labs-0.8.10.dist-info}/RECORD

@@ -1,49 +1,49 @@
-sempy_labs/__init__.py,sha256=YFVryT8HOZZ-A8gRG7A5f_R0TtQmtUoEJk0ZeoNSNJQ,12780
+sempy_labs/__init__.py,sha256=WyNDrg52pNpAj0s3H7o1vWNYnIpXZrXrKJVYVeC47GQ,13090
 sempy_labs/_ai.py,sha256=CzsNw6Wpd2B5Rd0RcY250-_p0L-0gFoMNLEc_KmrobU,16177
-sempy_labs/_authentication.py,sha256=LWkl0yoz63tYe5u9DL1js7g88vmkVbcTIqrsaxDHMT0,3851
+sempy_labs/_authentication.py,sha256=a4sPN8IAf7w-Jftm3PvtF62HqWLkCD69tXgze_G59Zg,4642
 sempy_labs/_capacities.py,sha256=HWX1ivlWpyS7Ea_ny-39kUAQYFGMzo42kWMGdJMINos,25466
 sempy_labs/_capacity_migration.py,sha256=PCIodWXas9v7q93hqD2m8EanJHPJzke52jbCWOfnLZk,27764
 sempy_labs/_clear_cache.py,sha256=ttHsXXR6SRRw4eC0cS8I5h38UbWU9YJii1y-uR9R3KM,12493
 sempy_labs/_connections.py,sha256=-Z3rfLGlUKx5iIGmvwWOICIVZ478ydwvCjxusoJb1RI,17647
 sempy_labs/_data_pipelines.py,sha256=WdZjTELNuN_7suWj6NrZUxGnMTzAgIxFw8V6YMb8ags,5644
-sempy_labs/_dataflows.py,sha256=Wpx5jmTDewTkL2bsVZ5r_PTD0etIBdKXVigTjUF8HAg,8252
-sempy_labs/_dax.py,sha256=oYNwMaHusxoqQmfi_S6iF2X5o29dCM7cb2eIiLpFlas,8605
+sempy_labs/_dataflows.py,sha256=D-OuuoUhSrGEhW1mAMBeHULfhULXmmTqIoOdV-se3vs,8250
+sempy_labs/_dax.py,sha256=5lY0p9bS0O2OWViaqTw_9K0WfZsyBW3gK4rIv1bqjjE,9411
 sempy_labs/_deployment_pipelines.py,sha256=WBBQM85-3-MkXb5OmRPF6U83xLyhKSlYUyhRlkvcl4k,6027
 sempy_labs/_documentation.py,sha256=yVA8VPEzx_fmljtcvSxtB7-BeupYsfdMXXjp6Fpnyo8,5007
 sempy_labs/_environments.py,sha256=avpLSfZyyQFdEDIIxWv2THLjPZwbs9XGXT7ob9l_-ao,5326
 sempy_labs/_eventhouses.py,sha256=vgIFQkXcBPC4SnlrBzT7SRmembExxkm6n0gdKnc7Hlk,4036
 sempy_labs/_eventstreams.py,sha256=Rht0eWoZbYF6XKtE3AOUvGgA21smxC9gdN599z-jY3s,4061
 sempy_labs/_external_data_shares.py,sha256=lUsKy1mexNSmhyFwxSeE2jZKNdDAWDP6iC6UPTXCvyU,6799
-sempy_labs/_gateways.py,sha256=CzTS95tLpG1bIG-0XkJWniNQEyNUtRq2go4QJAvjMr4,14617
-sempy_labs/_generate_semantic_model.py,sha256=au4amuYc35n0lyNAz3dYkJAfHfd2nkRpBybVY2ZQI90,15212
-sempy_labs/_git.py,sha256=C9TYKG4g8q35R1S7iTZyjMtRis033uUWkYJe_a0f3u0,13540
-sempy_labs/_helper_functions.py,sha256=Dk1cYfctxn_0RJHRKuqelcpaNJxyhoBZwWbvb0CGumU,38051
+sempy_labs/_gateways.py,sha256=j7REuoG9vGvPcJdII-gOJagprPJctsx3bi-ekUe5q6w,16228
+sempy_labs/_generate_semantic_model.py,sha256=YsPULWDkh2VE-jU3F3jIn0AmZPl3dq03sG-QfuHG2GI,16757
+sempy_labs/_git.py,sha256=gvFR6kCZbegoO_j_hubt-fxuaxRl1KsVldtQNJfnA9U,13870
+sempy_labs/_helper_functions.py,sha256=FtkzTlIJudgGRxrZzjfyon5ehuD8w8KR6sBV74R0XtY,39124
 sempy_labs/_icons.py,sha256=ez2dx_LCti71S_-eB6WYQ-kOMyiBL8ZJN12-ev5dcmA,3579
 sempy_labs/_kql_databases.py,sha256=oNX9oKnXu5SLkzl4kTMQguh4In-i-0Forcoy1phOe1s,4621
 sempy_labs/_kql_querysets.py,sha256=A-79LiLBKxlADPTuSK9ipy_LjXKcsJZwQHknUXFpVl0,4157
-sempy_labs/_list_functions.py,sha256=6EKlqwqT0tru1uV7qXCew5lMKgJrGqEnOHu_3q-z7RM,55503
+sempy_labs/_list_functions.py,sha256=_5mxBwfTh-BUFWK2lXNPXSpZO0OtP7FlcWaSxwD-Mgs,60939
 sempy_labs/_managed_private_endpoints.py,sha256=bCuC9V4yMFBw1BNlsoxARdIEMPAySW-ljHrhvuziQfw,6179
 sempy_labs/_mirrored_databases.py,sha256=5_5phu50KIvhHNQJ-RQAxd92W4D7GUVMyjAnOb7ZY3Q,14360
 sempy_labs/_mirrored_warehouses.py,sha256=t2fBH5L0UzNahDB4lATDLvmCqYTU-V93_ZVLb5ZISSg,1764
 sempy_labs/_ml_experiments.py,sha256=UVh3cwNvpY-otCBIaKW-sgtzyjwAuu8qJDLhZGBHToE,4196
 sempy_labs/_ml_models.py,sha256=phYLySjN7MO2YYfq7ZQKMS6w18G6L1-7DdNWB4fcLjQ,4044
 sempy_labs/_model_auto_build.py,sha256=PTQo3dufzLSFcQ5shFkmBWAVSdP7cTJgpUclrcXyNbg,5105
-sempy_labs/_model_bpa.py,sha256=Tyj7JV4L0rCmeWL896S58SOA8SHQKNxQ9QnmqunKTfM,21764
-sempy_labs/_model_bpa_bulk.py,sha256=WoJhN_A0X3qkaI624NS0PTq7r8x9uSdFtQEnOmiV7Dc,16003
+sempy_labs/_model_bpa.py,sha256=nhHAoq9RtOe0lAKRIg9Hr9TBMKxxbGOSCCCs1I8oy1s,21778
+sempy_labs/_model_bpa_bulk.py,sha256=jU-kaeUeE1Slz5HEh3lSbnILzj2tfzMwvaOqOQG16Wg,16027
 sempy_labs/_model_bpa_rules.py,sha256=96_GkXQGhON-_uyUATgUibk4W9y7e9wl1QciUr96gIQ,45544
-sempy_labs/_model_dependencies.py,sha256=xMqVeOqDg06__3gaOEkjyhItmHsc_FIUWNwCFZH6Aps,11794
-sempy_labs/_notebooks.py,sha256=EUYVeRJrCL9IllQevwRxzkCUU-rzX6KEEH7x7mBYUqc,7422
+sempy_labs/_model_dependencies.py,sha256=PL_-ozj3d2L03xR1S-4b-rGhghKbed3QY47x6i5BnfI,12070
+sempy_labs/_notebooks.py,sha256=DTz0byyNMP-JIEn4h85SJ8zMXNrkoChoeV-QE_TvPhE,8280
 sempy_labs/_one_lake_integration.py,sha256=eIuLxlw8eXfUH2avKhsyLmXZbTllSwGsz2j_HMAikpQ,6234
 sempy_labs/_query_scale_out.py,sha256=xoHnuDUgPYsg-NlUplB9ieb0bClcBQeG4veJNo_4TNw,15708
-sempy_labs/_refresh_semantic_model.py,sha256=R781zVHTnoLw5mzzcSc39OPBmKpzoBxsOa1KhvqyEgw,17131
+sempy_labs/_refresh_semantic_model.py,sha256=LSfwuViimX6TFq1KlQCMHue7ylzBwaBSrcPzJuvVz2M,17465
 sempy_labs/_spark.py,sha256=RIJt9b_l5Sp5XrebhvRD0DEBKDTQdA8Rh7fByV27ngQ,20109
 sempy_labs/_sql.py,sha256=KttKi95iGxTT8UA1QOpT9ygAdwCfHHlcQSQ5d9gml0E,5358
-sempy_labs/_translations.py,sha256=2DpP--U3d8Gp7o9LF-OnZa10onN2unvqSHVQHv3CBZg,19838
+sempy_labs/_translations.py,sha256=CVRd_yJ1pjUzxY_6H8tSCLh67bHhxRyS7DICY20Lqlc,16112
 sempy_labs/_vertipaq.py,sha256=sS9wFPxZfr_5dsOIXd-oeQIeCyXkVeCHbp30Kd7raUU,37662
 sempy_labs/_warehouses.py,sha256=KI7Ww5Slw4jfhby4ensGVlDHLWq6u2SvdMCa2R9i778,7205
 sempy_labs/_workloads.py,sha256=x3dS2mOkrS9rA-p70z8849DZlMIvMbzTjMzO_YmnHRg,4449
 sempy_labs/_workspace_identity.py,sha256=d5cdiiqjyUVoSoDiqU8mzWYOvbt2oJrt7sm-ZGEEkDk,2261
-sempy_labs/_workspaces.py,sha256=YdLuKBWBqvzYoGZSkWeAce-XxardzjpnpiaCMiE1aGI,11200
+sempy_labs/_workspaces.py,sha256=Ya_F-2b9LU08xN9bKKRS5LeGs_Ie2o4h26Xgy57IExk,11234
 sempy_labs/_bpa_translation/_model/_translations_am-ET.po,sha256=zQVjJ-t0vtgIYan-HaXtUVJLB_PJvB53Nf5BNoOReU4,39199
 sempy_labs/_bpa_translation/_model/_translations_ar-AE.po,sha256=QP1PjDLFccLDs9zq456crdAST57wrcWVk5rRiqqoCws,36959
 sempy_labs/_bpa_translation/_model/_translations_bg-BG.po,sha256=sqezjpS3wfk09WD7x27bHoCBtgmqeHtyHNKTwG7-bkI,44132
@@ -82,17 +82,17 @@ sempy_labs/_bpa_translation/_model/_translations_tr-TR.po,sha256=NdW-X4E0QmeLKM0
 sempy_labs/_bpa_translation/_model/_translations_uk-UA.po,sha256=3NsFN8hoor_5L6738FjpJ8o4spwp8FNFGbVItHD-_ec,43500
 sempy_labs/_bpa_translation/_model/_translations_zh-CN.po,sha256=ipMbnet7ZI5mZoC8KonYKVwGmFLHFB_9KIDOoBgSNfo,26815
 sempy_labs/_bpa_translation/_model/_translations_zu-ZA.po,sha256=5v6tVKGruqneAeMoa6F3tyg_JBL8qOpqOJofWpq2W3U,31518
-sempy_labs/admin/__init__.py,sha256=mWwIKFM2mTEMh4A7hhr27fnNSASrZ-4p5PzpCHn1FQY,1785
-sempy_labs/admin/_basic_functions.py,sha256=W21ZUjIMKG2FLCbh9QliT46E3R_SPRQmNljByAIZ7LY,33949
-sempy_labs/admin/_domains.py,sha256=tVjUiV4bLdVKl665ouYGfzYPFsRhPwYHYy7efCncvsE,12337
+sempy_labs/admin/__init__.py,sha256=YYOwKRfRr6alAez4BImlPcX9bQExATrb856BRq7a3O8,1945
+sempy_labs/admin/_basic_functions.py,sha256=5EEHWDhIFEt94-9yKNecIAMN-KzdBg1uqfdZNoXmnwA,37068
+sempy_labs/admin/_domains.py,sha256=5mv2SzIZCibvHwd4tgm-Lelj0zi66A2KKzQjDQgT9ms,12385
 sempy_labs/admin/_external_data_share.py,sha256=ITsPDgRDfgvZn1cjzpUWyR6lpnoOP0-gJVxjRA3Mp8w,3489
 sempy_labs/admin/_git.py,sha256=OY2F5ICKBXrB1HhlYDWdXQPnhTwSrMfWzEa2xcutClc,2181
 sempy_labs/admin/_items.py,sha256=LqjBYWL3NZCX8f0H-zzjOzy9zlBC7XR4LiknJv_JLT0,8428
-sempy_labs/admin/_scanner.py,sha256=GjBJUQg5qoeF492MkEaJOY_vIB7guFXv6YL7WFoPQy0,4500
+sempy_labs/admin/_scanner.py,sha256=OkP2Nc_s-DkYEmfLqiMIf8i3EhVyHfvnT1bPSSnVVss,4476
 sempy_labs/directlake/__init__.py,sha256=etaj-3wqe5t93mu74tGYjEOQ6gtHWUogidOygiVvlq8,2131
 sempy_labs/directlake/_directlake_schema_compare.py,sha256=ocHFU6E6HSKgcNLywGM0dx0ie9AXYwk-E7o7EYcqiN4,4422
 sempy_labs/directlake/_directlake_schema_sync.py,sha256=fhh6Xjd42HjI5x_Ejwq1N4qqnXQsKpXmyPcYl7cNG6A,4151
-sempy_labs/directlake/_dl_helper.py,sha256=lnvq99QCtxba0ViAEfL_fjhvlV1QMU-xDF0cU9s6ih4,9413
+sempy_labs/directlake/_dl_helper.py,sha256=tG3b0-BJbk-Kwk2B0fyPwoaMgTXS920L61Qz55COex8,9647
 sempy_labs/directlake/_generate_shared_expression.py,sha256=EauK1M4fabCZjsHYAWxEYaVJKqxJ99nZQaN2pKdd1lg,3077
 sempy_labs/directlake/_get_directlake_lakehouse.py,sha256=sovI4ds2SEgkp4Fi465jtJ4seRvQxdYgcixRDvsUwNM,2321
 sempy_labs/directlake/_get_shared_expression.py,sha256=rJ2twFSAMpjdjXl4zkqei_qxzxmGn5DxiDW2KxLcUog,1081
@@ -126,7 +126,7 @@ sempy_labs/report/_report_functions.py,sha256=nKqsVsjGrv8TUXsBXpb5ejEopAaFELc7Yz
 sempy_labs/report/_report_helper.py,sha256=pKIsca-XWaioQd97FgbEfsGPWy4X_NxSyYm22W3C23E,10461
 sempy_labs/report/_report_list_functions.py,sha256=Y0fkoo1gGoVDnihveKruNXFgPJNSiEQ5Fus8bw0nqcU,3381
 sempy_labs/report/_report_rebind.py,sha256=GbOfEb9qz4SdXVGopiWSkGMDKnneJxd7wx4_OWKZ1Js,5188
-sempy_labs/report/_reportwrapper.py,sha256=LMc-cZP7DbFhLn5EmOdKHFWhaBUZ7u2gY7Sx7GZxxJs,82712
+sempy_labs/report/_reportwrapper.py,sha256=ZFcWuZMo9SOzvZ8aPG-DvlZq_MkXRALUJusVIHn7dxE,83049
 sempy_labs/report/_bpareporttemplate/.platform,sha256=kWRa6B_KwSYLsvVFDx372mQriQO8v7dJ_YzQV_cfD-Q,303
 sempy_labs/report/_bpareporttemplate/definition.pbir,sha256=bttyHZYKqjA8OBb_cezGlX4H82cDvGZVCl1QB3fij4E,343
 sempy_labs/report/_bpareporttemplate/.pbi/localSettings.json,sha256=kzjBlNdjbsSBBSHBwbQc298AJCr9Vp6Ex0D5PemUuT0,1578
@@ -157,9 +157,9 @@ sempy_labs/report/_bpareporttemplate/definition/pages/c597da16dc7e63222a82/visua
 sempy_labs/report/_bpareporttemplate/definition/pages/d37dce724a0ccc30044b/page.json,sha256=wBVuNc8S2NaUA0FC708w6stmR2djNZp8nAsHMqesgsc,293
 sempy_labs/report/_bpareporttemplate/definition/pages/d37dce724a0ccc30044b/visuals/ce8532a7e25020271077/visual.json,sha256=mlY6t9OlSe-Y6_QmXJpS1vggU6Y3FjISUKECL8FVSg8,931
 sempy_labs/tom/__init__.py,sha256=Qbs8leW0fjzvWwOjyWK3Hjeehu7IvpB1beASGsi28bk,121
-sempy_labs/tom/_model.py,sha256=ebtKiXIP96BU8cK62OAQZq2PAB2TYM-4pIoOIA73_68,172073
-semantic_link_labs-0.8.8.dist-info/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
-semantic_link_labs-0.8.8.dist-info/METADATA,sha256=EQozPndrWN2DZX1jRCrvQClTroh5fCRuis4Ywt7GE6U,20813
-semantic_link_labs-0.8.8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-semantic_link_labs-0.8.8.dist-info/top_level.txt,sha256=kiQX1y42Dbein1l3Q8jMUYyRulDjdlc2tMepvtrvixQ,11
-semantic_link_labs-0.8.8.dist-info/RECORD,,
+sempy_labs/tom/_model.py,sha256=rBMI9BTSuTEH1MnJsUIVtsc45x7EOQ2fAxcYkwYgfZw,173575
+semantic_link_labs-0.8.10.dist-info/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+semantic_link_labs-0.8.10.dist-info/METADATA,sha256=eITmLRcqLr0Yuitj1Q1LXmWYDH5AA-fbpwWUDGWQuO4,21196
+semantic_link_labs-0.8.10.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+semantic_link_labs-0.8.10.dist-info/top_level.txt,sha256=kiQX1y42Dbein1l3Q8jMUYyRulDjdlc2tMepvtrvixQ,11
+semantic_link_labs-0.8.10.dist-info/RECORD,,

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",
 ]

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

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(

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",

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."
+    )

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)