base api

Author: m-kovalsky

src/sempy_labs/semantic_model/_caching.py

@@ -1,14 +1,11 @@
-import requests
 from uuid import UUID
 from typing import Optional
 from sempy_labs._helper_functions import (
-    get_pbi_token_headers,
-    _get_url_prefix,
     get_model_id,
     resolve_item_name_and_id,
     resolve_workspace_name_and_id,
+    _base_api,
 )
-from sempy.fabric.exceptions import FabricHTTPException
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 
@@ -31,13 +28,12 @@ def enable_query_caching(
     enable : bool, default=True
         Set to True to enable query caching, or False to disable it.
     """
-    prefix = _get_url_prefix()
-    headers = get_pbi_token_headers()
+
     (workspace_name, workspace_id) = resolve_workspace_name_and_id(workspace)
     (item_name, item_id) = resolve_item_name_and_id(
         item=dataset, type="SemanticModel", workspace=workspace_id
     )
-    model_id = get_model_id(item_id=item_id, headers=headers, prefix=prefix)
+    model_id = get_model_id(item_id=item_id)
     if model_id is None:
         raise ValueError(f"Failed to retrieve model ID for semantic model '{item_name}'")
 
@@ -48,12 +44,7 @@ def enable_query_caching(
 
     payload = {"queryCachingState": caching_map.get(enable)}
 
-    response = requests.post(
-        f"{prefix}/metadata/models/{model_id}/caching", headers=headers, json=payload
-    )
-
-    if response.status_code != 204:
-        raise FabricHTTPException(response)
+    _base_api(request=f"metadata/models/{model_id}/caching", method="internal", payload=payload, status_codes=204)
 
     print(
         f"{icons.green_dot} Query caching has been {'enabled' if enable else 'disabled'} for the '{item_name}' semantic model within the '{workspace_name}' workspace."