databricks-sdk 0.19.1__py3-none-any.whl → 0.20.0__py3-none-any.whl

databricks/sdk/__init__.py

@@ -15,6 +15,7 @@ from databricks.sdk.service.catalog import (AccountMetastoreAssignmentsAPI,
                                             ExternalLocationsAPI, FunctionsAPI,
                                             GrantsAPI, LakehouseMonitorsAPI,
                                             MetastoresAPI, ModelVersionsAPI,
+                                            OnlineTablesAPI,
                                             RegisteredModelsAPI, SchemasAPI,
                                             StorageCredentialsAPI,
                                             SystemSchemasAPI,
@@ -180,6 +181,7 @@ class WorkspaceClient:
         self._metastores = MetastoresAPI(self._api_client)
         self._model_registry = ModelRegistryAPI(self._api_client)
         self._model_versions = ModelVersionsAPI(self._api_client)
+        self._online_tables = OnlineTablesAPI(self._api_client)
         self._permissions = PermissionsAPI(self._api_client)
         self._pipelines = PipelinesAPI(self._api_client)
         self._policy_families = PolicyFamiliesAPI(self._api_client)
@@ -322,7 +324,7 @@ class WorkspaceClient:
 
     @property
     def files(self) -> FilesAPI:
-        """The Files API allows you to read, write, and delete files and directories in Unity Catalog volumes."""
+        """The Files API allows you to read, write, list, and delete files and directories."""
         return self._files
 
     @property
@@ -400,6 +402,11 @@ class WorkspaceClient:
         """Databricks provides a hosted version of MLflow Model Registry in Unity Catalog."""
         return self._model_versions
 
+    @property
+    def online_tables(self) -> OnlineTablesAPI:
+        """Online tables provide lower latency and higher QPS access to data from Delta tables."""
+        return self._online_tables
+
     @property
     def permissions(self) -> PermissionsAPI:
         """Permissions API are used to create read, write, edit, update and manage access for various users on different objects and endpoints."""

databricks/sdk/core.py

@@ -135,11 +135,13 @@ class ApiClient:
                                             raw=raw,
                                             files=files,
                                             data=data)
-        if raw:
-            return StreamingResponse(response)
+
         resp = dict()
         for header in response_headers if response_headers else []:
             resp[header] = response.headers.get(Casing.to_header_case(header))
+        if raw:
+            resp["contents"] = StreamingResponse(response)
+            return resp
         if not len(response.content):
             return resp
 

databricks/sdk/mixins/workspace.py

@@ -102,4 +102,5 @@ class WorkspaceExt(WorkspaceAPI):
         """
         query = {'path': path, 'direct_download': 'true'}
         if format: query['format'] = format.value
-        return self._api.do('GET', '/api/2.0/workspace/export', query=query, raw=True)
+        response = self._api.do('GET', '/api/2.0/workspace/export', query=query, raw=True)
+        return response["contents"]

databricks/sdk/service/billing.py

@@ -335,6 +335,17 @@ class DeliveryStatus(Enum):
 class DownloadResponse:
     contents: Optional[BinaryIO] = None
 
+    def as_dict(self) -> dict:
+        """Serializes the DownloadResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.contents: body['contents'] = self.contents
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DownloadResponse:
+        """Deserializes the DownloadResponse from a dictionary."""
+        return cls(contents=d.get('contents', None))
+
 
 class LogDeliveryConfigStatus(Enum):
     """Status of log delivery configuration. Set to `ENABLED` (enabled) or `DISABLED` (disabled).
@@ -704,12 +715,13 @@ class BillableUsageAPI:
         if personal_data is not None: query['personal_data'] = personal_data
         if start_month is not None: query['start_month'] = start_month
         headers = {'Accept': 'text/plain', }
+
         res = self._api.do('GET',
                            f'/api/2.0/accounts/{self._api.account_id}/usage/download',
                            query=query,
                            headers=headers,
                            raw=True)
-        return DownloadResponse(contents=res)
+        return DownloadResponse.from_dict(res)
 
 
 class BudgetsAPI:
@@ -732,6 +744,7 @@ class BudgetsAPI:
         body = {}
         if budget is not None: body['budget'] = budget.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/accounts/{self._api.account_id}/budget',
                            body=body,
@@ -750,6 +763,7 @@ class BudgetsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE',
                      f'/api/2.0/accounts/{self._api.account_id}/budget/{budget_id}',
                      headers=headers)
@@ -767,6 +781,7 @@ class BudgetsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/accounts/{self._api.account_id}/budget/{budget_id}',
                            headers=headers)
@@ -782,6 +797,7 @@ class BudgetsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', f'/api/2.0/accounts/{self._api.account_id}/budget', headers=headers)
         parsed = BudgetList.from_dict(json).budgets
         return parsed if parsed is not None else []
@@ -801,6 +817,7 @@ class BudgetsAPI:
         body = {}
         if budget is not None: body['budget'] = budget.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PATCH',
                      f'/api/2.0/accounts/{self._api.account_id}/budget/{budget_id}',
                      body=body,
@@ -894,6 +911,7 @@ class LogDeliveryAPI:
         if log_delivery_configuration is not None:
             body['log_delivery_configuration'] = log_delivery_configuration.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/accounts/{self._api.account_id}/log-delivery',
                            body=body,
@@ -912,6 +930,7 @@ class LogDeliveryAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do(
             'GET',
             f'/api/2.0/accounts/{self._api.account_id}/log-delivery/{log_delivery_configuration_id}',
@@ -942,6 +961,7 @@ class LogDeliveryAPI:
         if status is not None: query['status'] = status.value
         if storage_configuration_id is not None: query['storage_configuration_id'] = storage_configuration_id
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET',
                             f'/api/2.0/accounts/{self._api.account_id}/log-delivery',
                             query=query,
@@ -970,6 +990,7 @@ class LogDeliveryAPI:
         body = {}
         if status is not None: body['status'] = status.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PATCH',
                      f'/api/2.0/accounts/{self._api.account_id}/log-delivery/{log_delivery_configuration_id}',
                      body=body,