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

databricks/sdk/service/sql.py

@@ -303,6 +303,20 @@ class BaseChunkInfo:
                    row_offset=d.get('row_offset', None))
 
 
+@dataclass
+class CancelExecutionResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the CancelExecutionResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CancelExecutionResponse:
+        """Deserializes the CancelExecutionResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Channel:
     dbsql_version: Optional[str] = None
@@ -346,7 +360,6 @@ class ChannelInfo:
 
 
 class ChannelName(Enum):
-    """Name of the channel"""
 
     CHANNEL_NAME_CURRENT = 'CHANNEL_NAME_CURRENT'
     CHANNEL_NAME_CUSTOM = 'CHANNEL_NAME_CUSTOM'
@@ -896,6 +909,34 @@ class DataSource:
                    warehouse_id=d.get('warehouse_id', None))
 
 
+@dataclass
+class DeleteResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteWarehouseResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteWarehouseResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteWarehouseResponse:
+        """Deserializes the DeleteWarehouseResponse from a dictionary."""
+        return cls()
+
+
 class Disposition(Enum):
     """The fetch disposition provides two modes of fetching results: `INLINE` and `EXTERNAL_LINKS`.
     
@@ -1081,6 +1122,20 @@ class EditWarehouseRequestWarehouseType(Enum):
     TYPE_UNSPECIFIED = 'TYPE_UNSPECIFIED'
 
 
+@dataclass
+class EditWarehouseResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditWarehouseResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditWarehouseResponse:
+        """Deserializes the EditWarehouseResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EndpointConfPair:
     key: Optional[str] = None
@@ -2008,6 +2063,36 @@ class ListWarehousesResponse:
         return cls(warehouses=_repeated_dict(d, 'warehouses', EndpointInfo))
 
 
+@dataclass
+class MultiValuesOptions:
+    """If specified, allows multiple values to be selected for this parameter. Only applies to dropdown
+    list and query-based dropdown list parameters."""
+
+    prefix: Optional[str] = None
+    """Character that prefixes each selected parameter value."""
+
+    separator: Optional[str] = None
+    """Character that separates each selected parameter value. Defaults to a comma."""
+
+    suffix: Optional[str] = None
+    """Character that suffixes each selected parameter value."""
+
+    def as_dict(self) -> dict:
+        """Serializes the MultiValuesOptions into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.prefix is not None: body['prefix'] = self.prefix
+        if self.separator is not None: body['separator'] = self.separator
+        if self.suffix is not None: body['suffix'] = self.suffix
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> MultiValuesOptions:
+        """Deserializes the MultiValuesOptions from a dictionary."""
+        return cls(prefix=d.get('prefix', None),
+                   separator=d.get('separator', None),
+                   suffix=d.get('suffix', None))
+
+
 class ObjectType(Enum):
     """A singular noun object type."""
 
@@ -2064,9 +2149,20 @@ class OwnableObjectType(Enum):
 
 @dataclass
 class Parameter:
+    enum_options: Optional[str] = None
+    """List of valid parameter values, newline delimited. Only applies for dropdown list parameters."""
+
+    multi_values_options: Optional[MultiValuesOptions] = None
+    """If specified, allows multiple values to be selected for this parameter. Only applies to dropdown
+    list and query-based dropdown list parameters."""
+
     name: Optional[str] = None
     """The literal parameter marker that appears between double curly braces in the query text."""
 
+    query_id: Optional[str] = None
+    """The UUID of the query that provides the parameter values. Only applies for query-based dropdown
+    list parameters."""
+
     title: Optional[str] = None
     """The text displayed in a parameter picking widget."""
 
@@ -2079,7 +2175,10 @@ class Parameter:
     def as_dict(self) -> dict:
         """Serializes the Parameter into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.enum_options is not None: body['enumOptions'] = self.enum_options
+        if self.multi_values_options: body['multiValuesOptions'] = self.multi_values_options.as_dict()
         if self.name is not None: body['name'] = self.name
+        if self.query_id is not None: body['queryId'] = self.query_id
         if self.title is not None: body['title'] = self.title
         if self.type is not None: body['type'] = self.type.value
         if self.value: body['value'] = self.value
@@ -2088,7 +2187,10 @@ class Parameter:
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> Parameter:
         """Deserializes the Parameter from a dictionary."""
-        return cls(name=d.get('name', None),
+        return cls(enum_options=d.get('enumOptions', None),
+                   multi_values_options=_from_dict(d, 'multiValuesOptions', MultiValuesOptions),
+                   name=d.get('name', None),
+                   query_id=d.get('queryId', None),
                    title=d.get('title', None),
                    type=_enum(d, 'type', ParameterType),
                    value=d.get('value', None))
@@ -2098,7 +2200,9 @@ class ParameterType(Enum):
     """Parameters can have several different types."""
 
     DATETIME = 'datetime'
+    ENUM = 'enum'
     NUMBER = 'number'
+    QUERY = 'query'
     TEXT = 'text'
 
 
@@ -2536,9 +2640,6 @@ class QueryMetrics:
     photon_total_time_ms: Optional[int] = None
     """Total execution time for all individual Photon query engine tasks in the query, in milliseconds."""
 
-    planning_phases: Optional[List[Any]] = None
-    """Reserved for internal use."""
-
     planning_time_ms: Optional[int] = None
     """Reserved for internal use."""
 
@@ -2608,7 +2709,6 @@ class QueryMetrics:
         if self.overloading_queue_start_timestamp is not None:
             body['overloading_queue_start_timestamp'] = self.overloading_queue_start_timestamp
         if self.photon_total_time_ms is not None: body['photon_total_time_ms'] = self.photon_total_time_ms
-        if self.planning_phases: body['planning_phases'] = [v for v in self.planning_phases]
         if self.planning_time_ms is not None: body['planning_time_ms'] = self.planning_time_ms
         if self.provisioning_queue_start_timestamp is not None:
             body['provisioning_queue_start_timestamp'] = self.provisioning_queue_start_timestamp
@@ -2642,7 +2742,6 @@ class QueryMetrics:
                    network_sent_bytes=d.get('network_sent_bytes', None),
                    overloading_queue_start_timestamp=d.get('overloading_queue_start_timestamp', None),
                    photon_total_time_ms=d.get('photon_total_time_ms', None),
-                   planning_phases=d.get('planning_phases', None),
                    planning_time_ms=d.get('planning_time_ms', None),
                    provisioning_queue_start_timestamp=d.get('provisioning_queue_start_timestamp', None),
                    pruned_bytes=d.get('pruned_bytes', None),
@@ -2800,6 +2899,20 @@ class RepeatedEndpointConfPairs:
                    configuration_pairs=_repeated_dict(d, 'configuration_pairs', EndpointConfPair))
 
 
+@dataclass
+class RestoreResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the RestoreResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> RestoreResponse:
+        """Deserializes the RestoreResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ResultData:
     """Contains the result data of a single chunk when using `INLINE` disposition. When using
@@ -3083,6 +3196,20 @@ class SetWorkspaceWarehouseConfigRequestSecurityPolicy(Enum):
     PASSTHROUGH = 'PASSTHROUGH'
 
 
+@dataclass
+class SetWorkspaceWarehouseConfigResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the SetWorkspaceWarehouseConfigResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> SetWorkspaceWarehouseConfigResponse:
+        """Deserializes the SetWorkspaceWarehouseConfigResponse from a dictionary."""
+        return cls()
+
+
 class SpotInstancePolicy(Enum):
     """Configurations whether the warehouse should use spot instances."""
 
@@ -3091,6 +3218,20 @@ class SpotInstancePolicy(Enum):
     RELIABILITY_OPTIMIZED = 'RELIABILITY_OPTIMIZED'
 
 
+@dataclass
+class StartWarehouseResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StartWarehouseResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StartWarehouseResponse:
+        """Deserializes the StartWarehouseResponse from a dictionary."""
+        return cls()
+
+
 class State(Enum):
     """State of the warehouse"""
 
@@ -3180,6 +3321,20 @@ class Status(Enum):
     STATUS_UNSPECIFIED = 'STATUS_UNSPECIFIED'
 
 
+@dataclass
+class StopWarehouseResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StopWarehouseResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StopWarehouseResponse:
+        """Deserializes the StopWarehouseResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class Success:
     message: Optional[SuccessMessage] = None
@@ -3359,6 +3514,20 @@ class TransferOwnershipObjectId:
         return cls(new_owner=d.get('new_owner', None))
 
 
+@dataclass
+class UpdateResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateResponse:
+        """Deserializes the UpdateResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class User:
     email: Optional[str] = None
@@ -3802,6 +3971,7 @@ class AlertsAPI:
         if query_id is not None: body['query_id'] = query_id
         if rearm is not None: body['rearm'] = rearm
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/preview/sql/alerts', body=body, headers=headers)
         return Alert.from_dict(res)
 
@@ -3817,6 +3987,7 @@ class AlertsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/preview/sql/alerts/{alert_id}', headers=headers)
 
     def get(self, alert_id: str) -> Alert:
@@ -3830,6 +4001,7 @@ class AlertsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/preview/sql/alerts/{alert_id}', headers=headers)
         return Alert.from_dict(res)
 
@@ -3842,6 +4014,7 @@ class AlertsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/preview/sql/alerts', headers=headers)
         return [Alert.from_dict(v) for v in res]
 
@@ -3875,6 +4048,7 @@ class AlertsAPI:
         if query_id is not None: body['query_id'] = query_id
         if rearm is not None: body['rearm'] = rearm
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PUT', f'/api/2.0/preview/sql/alerts/{alert_id}', body=body, headers=headers)
 
 
@@ -3914,6 +4088,7 @@ class DashboardWidgetsAPI:
         if visualization_id is not None: body['visualization_id'] = visualization_id
         if width is not None: body['width'] = width
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/preview/sql/widgets', body=body, headers=headers)
         return Widget.from_dict(res)
 
@@ -3927,6 +4102,7 @@ class DashboardWidgetsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/preview/sql/widgets/{id}', headers=headers)
 
     def update(self,
@@ -3961,6 +4137,7 @@ class DashboardWidgetsAPI:
         if visualization_id is not None: body['visualization_id'] = visualization_id
         if width is not None: body['width'] = width
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', f'/api/2.0/preview/sql/widgets/{id}', body=body, headers=headers)
         return Widget.from_dict(res)
 
@@ -4009,6 +4186,7 @@ class DashboardsAPI:
         if run_as_role is not None: body['run_as_role'] = run_as_role.value
         if tags is not None: body['tags'] = [v for v in tags]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/preview/sql/dashboards', body=body, headers=headers)
         return Dashboard.from_dict(res)
 
@@ -4024,6 +4202,7 @@ class DashboardsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/preview/sql/dashboards/{dashboard_id}', headers=headers)
 
     def get(self, dashboard_id: str) -> Dashboard:
@@ -4037,6 +4216,7 @@ class DashboardsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/preview/sql/dashboards/{dashboard_id}', headers=headers)
         return Dashboard.from_dict(res)
 
@@ -4099,6 +4279,7 @@ class DashboardsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('POST', f'/api/2.0/preview/sql/dashboards/trash/{dashboard_id}', headers=headers)
 
     def update(self,
@@ -4126,6 +4307,7 @@ class DashboardsAPI:
         if name is not None: body['name'] = name
         if run_as_role is not None: body['run_as_role'] = run_as_role.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/preview/sql/dashboards/{dashboard_id}',
                            body=body,
@@ -4156,6 +4338,7 @@ class DataSourcesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/preview/sql/data_sources', headers=headers)
         return [DataSource.from_dict(v) for v in res]
 
@@ -4190,6 +4373,7 @@ class DbsqlPermissionsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/preview/sql/permissions/{object_type.value}/{object_id}',
                            headers=headers)
@@ -4217,6 +4401,7 @@ class DbsqlPermissionsAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/preview/sql/permissions/{object_type.value}/{object_id}',
                            body=body,
@@ -4244,6 +4429,7 @@ class DbsqlPermissionsAPI:
         body = {}
         if new_owner is not None: body['new_owner'] = new_owner
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/preview/sql/permissions/{object_type.value}/{object_id}/transfer',
                            body=body,
@@ -4311,6 +4497,7 @@ class QueriesAPI:
         if query is not None: body['query'] = query
         if run_as_role is not None: body['run_as_role'] = run_as_role.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/preview/sql/queries', body=body, headers=headers)
         return Query.from_dict(res)
 
@@ -4326,6 +4513,7 @@ class QueriesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/preview/sql/queries/{query_id}', headers=headers)
 
     def get(self, query_id: str) -> Query:
@@ -4340,6 +4528,7 @@ class QueriesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/preview/sql/queries/{query_id}', headers=headers)
         return Query.from_dict(res)
 
@@ -4415,6 +4604,7 @@ class QueriesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('POST', f'/api/2.0/preview/sql/queries/trash/{query_id}', headers=headers)
 
     def update(self,
@@ -4462,6 +4652,7 @@ class QueriesAPI:
         if query is not None: body['query'] = query
         if run_as_role is not None: body['run_as_role'] = run_as_role.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', f'/api/2.0/preview/sql/queries/{query_id}', body=body, headers=headers)
         return Query.from_dict(res)
 
@@ -4552,6 +4743,7 @@ class QueryVisualizationsAPI:
         if query_id is not None: body['query_id'] = query_id
         if type is not None: body['type'] = type
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/preview/sql/visualizations', body=body, headers=headers)
         return Visualization.from_dict(res)
 
@@ -4565,6 +4757,7 @@ class QueryVisualizationsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/preview/sql/visualizations/{id}', headers=headers)
 
     def update(self,
@@ -4602,6 +4795,7 @@ class QueryVisualizationsAPI:
         if type is not None: body['type'] = type
         if updated_at is not None: body['updated_at'] = updated_at
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', f'/api/2.0/preview/sql/visualizations/{id}', body=body, headers=headers)
         return Visualization.from_dict(res)
 
@@ -4707,6 +4901,7 @@ class StatementExecutionAPI:
         """
 
         headers = {}
+
         self._api.do('POST', f'/api/2.0/sql/statements/{statement_id}/cancel', headers=headers)
 
     def execute_statement(self,
@@ -4862,6 +5057,7 @@ class StatementExecutionAPI:
         if wait_timeout is not None: body['wait_timeout'] = wait_timeout
         if warehouse_id is not None: body['warehouse_id'] = warehouse_id
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/sql/statements/', body=body, headers=headers)
         return ExecuteStatementResponse.from_dict(res)
 
@@ -4884,6 +5080,7 @@ class StatementExecutionAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/sql/statements/{statement_id}', headers=headers)
         return GetStatementResponse.from_dict(res)
 
@@ -4906,6 +5103,7 @@ class StatementExecutionAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/sql/statements/{statement_id}/result/chunks/{chunk_index}',
                            headers=headers)
@@ -5077,6 +5275,7 @@ class WarehousesAPI:
         if tags is not None: body['tags'] = tags.as_dict()
         if warehouse_type is not None: body['warehouse_type'] = warehouse_type.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         op_response = self._api.do('POST', '/api/2.0/sql/warehouses', body=body, headers=headers)
         return Wait(self.wait_get_warehouse_running,
                     response=CreateWarehouseResponse.from_dict(op_response),
@@ -5126,6 +5325,7 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/sql/warehouses/{id}', headers=headers)
 
     def edit(
@@ -5226,8 +5426,11 @@ class WarehousesAPI:
         if tags is not None: body['tags'] = tags.as_dict()
         if warehouse_type is not None: body['warehouse_type'] = warehouse_type.value
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
-        self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/edit', body=body, headers=headers)
-        return Wait(self.wait_get_warehouse_running, id=id)
+
+        op_response = self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/edit', body=body, headers=headers)
+        return Wait(self.wait_get_warehouse_running,
+                    response=EditWarehouseResponse.from_dict(op_response),
+                    id=id)
 
     def edit_and_wait(
         self,
@@ -5275,6 +5478,7 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/sql/warehouses/{id}', headers=headers)
         return GetWarehouseResponse.from_dict(res)
 
@@ -5290,6 +5494,7 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/warehouses/{warehouse_id}/permissionLevels',
                            headers=headers)
@@ -5308,6 +5513,7 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/permissions/warehouses/{warehouse_id}', headers=headers)
         return WarehousePermissions.from_dict(res)
 
@@ -5320,6 +5526,7 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', '/api/2.0/sql/config/warehouses', headers=headers)
         return GetWorkspaceWarehouseConfigResponse.from_dict(res)
 
@@ -5338,6 +5545,7 @@ class WarehousesAPI:
         query = {}
         if run_as_user_id is not None: query['run_as_user_id'] = run_as_user_id
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET', '/api/2.0/sql/warehouses', query=query, headers=headers)
         parsed = ListWarehousesResponse.from_dict(json).warehouses
         return parsed if parsed is not None else []
@@ -5361,6 +5569,7 @@ class WarehousesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PUT',
                            f'/api/2.0/permissions/warehouses/{warehouse_id}',
                            body=body,
@@ -5422,6 +5631,7 @@ class WarehousesAPI:
         if sql_configuration_parameters is not None:
             body['sql_configuration_parameters'] = sql_configuration_parameters.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PUT', '/api/2.0/sql/config/warehouses', body=body, headers=headers)
 
     def start(self, id: str) -> Wait[GetWarehouseResponse]:
@@ -5438,8 +5648,11 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
-        self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/start', headers=headers)
-        return Wait(self.wait_get_warehouse_running, id=id)
+
+        op_response = self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/start', headers=headers)
+        return Wait(self.wait_get_warehouse_running,
+                    response=StartWarehouseResponse.from_dict(op_response),
+                    id=id)
 
     def start_and_wait(self, id: str, timeout=timedelta(minutes=20)) -> GetWarehouseResponse:
         return self.start(id=id).result(timeout=timeout)
@@ -5458,8 +5671,11 @@ class WarehousesAPI:
         """
 
         headers = {'Accept': 'application/json', }
-        self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/stop', headers=headers)
-        return Wait(self.wait_get_warehouse_stopped, id=id)
+
+        op_response = self._api.do('POST', f'/api/2.0/sql/warehouses/{id}/stop', headers=headers)
+        return Wait(self.wait_get_warehouse_stopped,
+                    response=StopWarehouseResponse.from_dict(op_response),
+                    id=id)
 
     def stop_and_wait(self, id: str, timeout=timedelta(minutes=20)) -> GetWarehouseResponse:
         return self.stop(id=id).result(timeout=timeout)
@@ -5484,6 +5700,7 @@ class WarehousesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH',
                            f'/api/2.0/permissions/warehouses/{warehouse_id}',
                            body=body,