anyscale 0.26.63__py3-none-any.whl → 0.26.65__py3-none-any.whl

anyscale/client/openapi_client/api/default_api.py

@@ -7019,6 +7019,120 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete(self, connection_id, **kwargs):  # noqa: E501
+        """Delete Databricks Connection  # noqa: E501
+
+        Delete a Databricks connection by ID  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete(connection_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str connection_id: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: DatabricksconnectionresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete_with_http_info(connection_id, **kwargs)  # noqa: E501
+
+    def delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete_with_http_info(self, connection_id, **kwargs):  # noqa: E501
+        """Delete Databricks Connection  # noqa: E501
+
+        Delete a Databricks connection by ID  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete_with_http_info(connection_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str connection_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(DatabricksconnectionresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'connection_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'connection_id' is set
+        if self.api_client.client_side_validation and ('connection_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['connection_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `connection_id` when calling `delete_databricks_connection_api_v2_integrations_connections_databricks_connection_id_delete`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'connection_id' in local_var_params:
+            path_params['connection_id'] = local_var_params['connection_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/integrations/connections/databricks/{connection_id}', 'DELETE',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='DatabricksconnectionresponseResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def delete_hosted_cloud_api_v2_clouds_hosted_cloud_id_delete(self, cloud_id, **kwargs):  # noqa: E501
         """Delete Hosted Cloud  # noqa: E501
 
@@ -10180,6 +10294,136 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post(self, aggregated_usage_query, **kwargs):  # noqa: E501
+        """Fetch Usage Group By Cluster Type  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post(aggregated_usage_query, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :param str paging_token:
+        :param int count:
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: UsagebyclustertypeListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post_with_http_info(aggregated_usage_query, **kwargs)  # noqa: E501
+
+    def fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post_with_http_info(self, aggregated_usage_query, **kwargs):  # noqa: E501
+        """Fetch Usage Group By Cluster Type  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post_with_http_info(aggregated_usage_query, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :param str paging_token:
+        :param int count:
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(UsagebyclustertypeListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'aggregated_usage_query' is set
+        if self.api_client.client_side_validation and ('aggregated_usage_query' not in local_var_params or  # noqa: E501
+                                                        local_var_params['aggregated_usage_query'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post`")  # noqa: E501
+
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post`, must be a value less than or equal to `1000`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_type_api_v2_aggregated_instance_usage_cluster_type_post`, must be a value greater than or equal to `0`")  # noqa: E501
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
+            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
+        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
+            query_params.append(('count', local_var_params['count']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'aggregated_usage_query' in local_var_params:
+            body_params = local_var_params['aggregated_usage_query']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/aggregated_instance_usage/cluster_type', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='UsagebyclustertypeListResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post(self, aggregated_usage_query, **kwargs):  # noqa: E501
         """Fetch Usage Group By Instance Type  # noqa: E501
 
@@ -13969,17 +14213,17 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def get_dataset_api_v2_datasets_dataset_id_get(self, dataset_id, **kwargs):  # noqa: E501
-        """Get Dataset  # noqa: E501
+    def get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get(self, connection_id, **kwargs):  # noqa: E501
+        """Get Databricks Connection  # noqa: E501
 
-        Retrieves a Dataset object  # noqa: E501
+        Get a specific Databricks connection by ID  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_dataset_api_v2_datasets_dataset_id_get(dataset_id, async_req=True)
+        >>> thread = api.get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get(connection_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str dataset_id: (required)
+        :param str connection_id: (required)
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -13987,24 +14231,24 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: BackendServerApiProductRoutersDatasetsRouterDatasetResponse
+        :return: DatabricksconnectioninfoResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
-        return self.get_dataset_api_v2_datasets_dataset_id_get_with_http_info(dataset_id, **kwargs)  # noqa: E501
+        return self.get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get_with_http_info(connection_id, **kwargs)  # noqa: E501
 
-    def get_dataset_api_v2_datasets_dataset_id_get_with_http_info(self, dataset_id, **kwargs):  # noqa: E501
-        """Get Dataset  # noqa: E501
+    def get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get_with_http_info(self, connection_id, **kwargs):  # noqa: E501
+        """Get Databricks Connection  # noqa: E501
 
-        Retrieves a Dataset object  # noqa: E501
+        Get a specific Databricks connection by ID  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_dataset_api_v2_datasets_dataset_id_get_with_http_info(dataset_id, async_req=True)
+        >>> thread = api.get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get_with_http_info(connection_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str dataset_id: (required)
+        :param str connection_id: (required)
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -14014,7 +14258,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(BackendServerApiProductRoutersDatasetsRouterDatasetResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(DatabricksconnectioninfoResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -14022,7 +14266,7 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'dataset_id'
+            'connection_id'
         ]
         all_params.extend(
             [
@@ -14037,20 +14281,20 @@ class DefaultApi(object):
             if key not in all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method get_dataset_api_v2_datasets_dataset_id_get" % key
+                    " to method get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get" % key
                 )
             local_var_params[key] = val
         del local_var_params['kwargs']
-        # verify the required parameter 'dataset_id' is set
-        if self.api_client.client_side_validation and ('dataset_id' not in local_var_params or  # noqa: E501
-                                                        local_var_params['dataset_id'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `dataset_id` when calling `get_dataset_api_v2_datasets_dataset_id_get`")  # noqa: E501
+        # verify the required parameter 'connection_id' is set
+        if self.api_client.client_side_validation and ('connection_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['connection_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `connection_id` when calling `get_databricks_connection_api_v2_integrations_connections_databricks_connection_id_get`")  # noqa: E501
 
         collection_formats = {}
 
         path_params = {}
-        if 'dataset_id' in local_var_params:
-            path_params['dataset_id'] = local_var_params['dataset_id']  # noqa: E501
+        if 'connection_id' in local_var_params:
+            path_params['connection_id'] = local_var_params['connection_id']  # noqa: E501
 
         query_params = []
 
@@ -14068,14 +14312,14 @@ class DefaultApi(object):
         auth_settings = []  # noqa: E501
 
         return self.api_client.call_api(
-            '/api/v2/datasets/{dataset_id}', 'GET',
+            '/api/v2/integrations/connections/databricks/{connection_id}', 'GET',
             path_params,
             query_params,
             header_params,
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='BackendServerApiProductRoutersDatasetsRouterDatasetResponse',  # noqa: E501
+            response_type='DatabricksconnectioninfoResponse',  # noqa: E501
             auth_settings=auth_settings,
             async_req=local_var_params.get('async_req'),
             _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
@@ -14083,19 +14327,17 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def get_dataset_download_info_api_v2_datasets_download_info_get(self, name, **kwargs):  # noqa: E501
-        """Get Dataset Download Info  # noqa: E501
+    def get_dataset_api_v2_datasets_dataset_id_get(self, dataset_id, **kwargs):  # noqa: E501
+        """Get Dataset  # noqa: E501
 
-        Returns download info for a dataset.  # noqa: E501
+        Retrieves a Dataset object  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_dataset_download_info_api_v2_datasets_download_info_get(name, async_req=True)
+        >>> thread = api.get_dataset_api_v2_datasets_dataset_id_get(dataset_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str name: Name of the dataset to download. (required)
-        :param int version: Version of the dataset to download. If a negative integer is provided, the dataset returned is this many versions back of the latest version. Default: Latest version
-        :param str project_id: ID of the Anyscale project that the dataset belongs to.
+        :param str dataset_id: (required)
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -14103,26 +14345,142 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: ClouddatabucketpresignedurlresponseResponse
+        :return: BackendServerApiProductRoutersDatasetsRouterDatasetResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
-        return self.get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(name, **kwargs)  # noqa: E501
+        return self.get_dataset_api_v2_datasets_dataset_id_get_with_http_info(dataset_id, **kwargs)  # noqa: E501
 
-    def get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(self, name, **kwargs):  # noqa: E501
-        """Get Dataset Download Info  # noqa: E501
+    def get_dataset_api_v2_datasets_dataset_id_get_with_http_info(self, dataset_id, **kwargs):  # noqa: E501
+        """Get Dataset  # noqa: E501
 
-        Returns download info for a dataset.  # noqa: E501
+        Retrieves a Dataset object  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(name, async_req=True)
+        >>> thread = api.get_dataset_api_v2_datasets_dataset_id_get_with_http_info(dataset_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str name: Name of the dataset to download. (required)
-        :param int version: Version of the dataset to download. If a negative integer is provided, the dataset returned is this many versions back of the latest version. Default: Latest version
-        :param str project_id: ID of the Anyscale project that the dataset belongs to.
+        :param str dataset_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(BackendServerApiProductRoutersDatasetsRouterDatasetResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'dataset_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method get_dataset_api_v2_datasets_dataset_id_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'dataset_id' is set
+        if self.api_client.client_side_validation and ('dataset_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['dataset_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `dataset_id` when calling `get_dataset_api_v2_datasets_dataset_id_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'dataset_id' in local_var_params:
+            path_params['dataset_id'] = local_var_params['dataset_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/datasets/{dataset_id}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='BackendServerApiProductRoutersDatasetsRouterDatasetResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def get_dataset_download_info_api_v2_datasets_download_info_get(self, name, **kwargs):  # noqa: E501
+        """Get Dataset Download Info  # noqa: E501
+
+        Returns download info for a dataset.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.get_dataset_download_info_api_v2_datasets_download_info_get(name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str name: Name of the dataset to download. (required)
+        :param int version: Version of the dataset to download. If a negative integer is provided, the dataset returned is this many versions back of the latest version. Default: Latest version
+        :param str project_id: ID of the Anyscale project that the dataset belongs to.
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: ClouddatabucketpresignedurlresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(name, **kwargs)  # noqa: E501
+
+    def get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(self, name, **kwargs):  # noqa: E501
+        """Get Dataset Download Info  # noqa: E501
+
+        Returns download info for a dataset.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.get_dataset_download_info_api_v2_datasets_download_info_get_with_http_info(name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str name: Name of the dataset to download. (required)
+        :param int version: Version of the dataset to download. If a negative integer is provided, the dataset returned is this many versions back of the latest version. Default: Latest version
+        :param str project_id: ID of the Anyscale project that the dataset belongs to.
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -17783,6 +18141,7 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
         :param str session_name: The name of the session to fetch nodes for.
+        :param str text_filter: The text to filter the nodes by. Search for case-insensitive substring match in node id, name, node_ip, labels, or resources.
         :param list[NodeStatus] states_filter: The states to filter the nodes by.
         :param int page: The page number to fetch.
         :param int page_size: The number of nodes to fetch per page.
@@ -17811,6 +18170,7 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
         :param str session_name: The name of the session to fetch nodes for.
+        :param str text_filter: The text to filter the nodes by. Search for case-insensitive substring match in node id, name, node_ip, labels, or resources.
         :param list[NodeStatus] states_filter: The states to filter the nodes by.
         :param int page: The page number to fetch.
         :param int page_size: The number of nodes to fetch per page.
@@ -17833,6 +18193,7 @@ class DefaultApi(object):
         all_params = [
             'cluster_id',
             'session_name',
+            'text_filter',
             'states_filter',
             'page',
             'page_size'
@@ -17868,6 +18229,8 @@ class DefaultApi(object):
             query_params.append(('cluster_id', local_var_params['cluster_id']))  # noqa: E501
         if 'session_name' in local_var_params and local_var_params['session_name'] is not None:  # noqa: E501
             query_params.append(('session_name', local_var_params['session_name']))  # noqa: E501
+        if 'text_filter' in local_var_params and local_var_params['text_filter'] is not None:  # noqa: E501
+            query_params.append(('text_filter', local_var_params['text_filter']))  # noqa: E501
         if 'states_filter' in local_var_params and local_var_params['states_filter'] is not None:  # noqa: E501
             query_params.append(('states_filter', local_var_params['states_filter']))  # noqa: E501
             collection_formats['states_filter'] = 'multi'  # noqa: E501
@@ -17905,6 +18268,125 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def get_oauth_login_url_api_v2_integrations_oauth_login_get(self, integration_id, **kwargs):  # noqa: E501
+        """Get Oauth Login Url  # noqa: E501
+
+        Generate OAuth authorization URL with PKCE for third party integration  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.get_oauth_login_url_api_v2_integrations_oauth_login_get(integration_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str integration_id: ID of the registered third party integration connection (required)
+        :param bool validation: Whether this is a validation flow for the connection
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: OauthAuthUrlResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.get_oauth_login_url_api_v2_integrations_oauth_login_get_with_http_info(integration_id, **kwargs)  # noqa: E501
+
+    def get_oauth_login_url_api_v2_integrations_oauth_login_get_with_http_info(self, integration_id, **kwargs):  # noqa: E501
+        """Get Oauth Login Url  # noqa: E501
+
+        Generate OAuth authorization URL with PKCE for third party integration  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.get_oauth_login_url_api_v2_integrations_oauth_login_get_with_http_info(integration_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str integration_id: ID of the registered third party integration connection (required)
+        :param bool validation: Whether this is a validation flow for the connection
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(OauthAuthUrlResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'integration_id',
+            'validation'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method get_oauth_login_url_api_v2_integrations_oauth_login_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'integration_id' is set
+        if self.api_client.client_side_validation and ('integration_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['integration_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `integration_id` when calling `get_oauth_login_url_api_v2_integrations_oauth_login_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'integration_id' in local_var_params and local_var_params['integration_id'] is not None:  # noqa: E501
+            query_params.append(('integration_id', local_var_params['integration_id']))  # noqa: E501
+        if 'validation' in local_var_params and local_var_params['validation'] is not None:  # noqa: E501
+            query_params.append(('validation', local_var_params['validation']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/integrations/oauth/login', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='OauthAuthUrlResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def get_operator_events_api_v2_dataset_runs_operator_events_get(self, cluster_id, session_name, dataset_id, operator_id, **kwargs):  # noqa: E501
         """Get Operator Events  # noqa: E501
 
@@ -24236,6 +24718,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str name:
+        :param str paging_token:
+        :param int count:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -24261,6 +24745,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str name:
+        :param str paging_token:
+        :param int count:
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -24278,7 +24764,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'name'
+            'name',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -24298,6 +24786,10 @@ class DefaultApi(object):
             local_var_params[key] = val
         del local_var_params['kwargs']
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `list_api_keys_api_v2_users_api_keys_get`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `list_api_keys_api_v2_users_api_keys_get`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
@@ -24305,6 +24797,10 @@ class DefaultApi(object):
         query_params = []
         if 'name' in local_var_params and local_var_params['name'] is not None:  # noqa: E501
             query_params.append(('name', local_var_params['name']))  # noqa: E501
+        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
+            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
+        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
+            query_params.append(('count', local_var_params['count']))  # noqa: E501
 
         header_params = {}
 
@@ -24348,6 +24844,8 @@ class DefaultApi(object):
         :param str name:
         :param str created_by:
         :param str user_id:
+        :param str paging_token:
+        :param int count:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -24375,6 +24873,8 @@ class DefaultApi(object):
         :param str name:
         :param str created_by:
         :param str user_id:
+        :param str paging_token:
+        :param int count:
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -24394,7 +24894,9 @@ class DefaultApi(object):
         all_params = [
             'name',
             'created_by',
-            'user_id'
+            'user_id',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -24414,6 +24916,10 @@ class DefaultApi(object):
             local_var_params[key] = val
         del local_var_params['kwargs']
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `list_api_keys_for_admin_api_v2_users_admin_api_keys_get`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `list_api_keys_for_admin_api_v2_users_admin_api_keys_get`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
@@ -24425,6 +24931,10 @@ class DefaultApi(object):
             query_params.append(('created_by', local_var_params['created_by']))  # noqa: E501
         if 'user_id' in local_var_params and local_var_params['user_id'] is not None:  # noqa: E501
             query_params.append(('user_id', local_var_params['user_id']))  # noqa: E501
+        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
+            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
+        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
+            query_params.append(('count', local_var_params['count']))  # noqa: E501
 
         header_params = {}
 
@@ -24732,17 +25242,17 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def list_clouds_api_v2_clouds_get(self, **kwargs):  # noqa: E501
-        """List Clouds  # noqa: E501
+    def list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get(self, connection_id, **kwargs):  # noqa: E501
+        """List Catalogs In Connection  # noqa: E501
 
+        Lists all catalogs within a specific provider connection  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.list_clouds_api_v2_clouds_get(async_req=True)
+        >>> thread = api.list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get(connection_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str paging_token:
-        :param int count:
+        :param str connection_id: ID of the provider connection (required)
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -24750,24 +25260,24 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: CloudListResponse
+        :return: DatacatalogListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
-        return self.list_clouds_api_v2_clouds_get_with_http_info(**kwargs)  # noqa: E501
+        return self.list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get_with_http_info(connection_id, **kwargs)  # noqa: E501
 
-    def list_clouds_api_v2_clouds_get_with_http_info(self, **kwargs):  # noqa: E501
-        """List Clouds  # noqa: E501
+    def list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get_with_http_info(self, connection_id, **kwargs):  # noqa: E501
+        """List Catalogs In Connection  # noqa: E501
 
+        Lists all catalogs within a specific provider connection  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.list_clouds_api_v2_clouds_get_with_http_info(async_req=True)
+        >>> thread = api.list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get_with_http_info(connection_id, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param str paging_token:
-        :param int count:
+        :param str connection_id: ID of the provider connection (required)
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -24777,7 +25287,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(CloudListResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(DatacatalogListResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -24785,8 +25295,7 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'paging_token',
-            'count'
+            'connection_id'
         ]
         all_params.extend(
             [
@@ -24801,14 +25310,129 @@ class DefaultApi(object):
             if key not in all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method list_clouds_api_v2_clouds_get" % key
+                    " to method list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get" % key
                 )
             local_var_params[key] = val
         del local_var_params['kwargs']
+        # verify the required parameter 'connection_id' is set
+        if self.api_client.client_side_validation and ('connection_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['connection_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `connection_id` when calling `list_catalogs_in_connection_api_v2_data_catalogs_connections_connection_id_catalogs_get`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `list_clouds_api_v2_clouds_get`, must be a value less than or equal to `1000`")  # noqa: E501
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
+        collection_formats = {}
+
+        path_params = {}
+        if 'connection_id' in local_var_params:
+            path_params['connection_id'] = local_var_params['connection_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/data_catalogs/connections/{connection_id}/catalogs', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='DatacatalogListResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def list_clouds_api_v2_clouds_get(self, **kwargs):  # noqa: E501
+        """List Clouds  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_clouds_api_v2_clouds_get(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str paging_token:
+        :param int count:
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: CloudListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.list_clouds_api_v2_clouds_get_with_http_info(**kwargs)  # noqa: E501
+
+    def list_clouds_api_v2_clouds_get_with_http_info(self, **kwargs):  # noqa: E501
+        """List Clouds  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_clouds_api_v2_clouds_get_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str paging_token:
+        :param int count:
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(CloudListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'paging_token',
+            'count'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method list_clouds_api_v2_clouds_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `list_clouds_api_v2_clouds_get`, must be a value less than or equal to `1000`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `list_clouds_api_v2_clouds_get`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
@@ -25093,6 +25717,226 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def list_data_catalog_connections_api_v2_data_catalogs_connections_get(self, **kwargs):  # noqa: E501
+        """List Data Catalog Connections  # noqa: E501
+
+        Lists all data catalog provider connections.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_data_catalog_connections_api_v2_data_catalogs_connections_get(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: DatacatalogconnectionListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.list_data_catalog_connections_api_v2_data_catalogs_connections_get_with_http_info(**kwargs)  # noqa: E501
+
+    def list_data_catalog_connections_api_v2_data_catalogs_connections_get_with_http_info(self, **kwargs):  # noqa: E501
+        """List Data Catalog Connections  # noqa: E501
+
+        Lists all data catalog provider connections.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_data_catalog_connections_api_v2_data_catalogs_connections_get_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(DatacatalogconnectionListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method list_data_catalog_connections_api_v2_data_catalogs_connections_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/data_catalogs/connections', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='DatacatalogconnectionListResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def list_databricks_connections_api_v2_integrations_connections_databricks_get(self, **kwargs):  # noqa: E501
+        """List Databricks Connections  # noqa: E501
+
+        List all registered Databricks connections  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_databricks_connections_api_v2_integrations_connections_databricks_get(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param ValidationStatus validation_status:
+        :param ConnectionType connection_type:
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: ListDatabricksConnections
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.list_databricks_connections_api_v2_integrations_connections_databricks_get_with_http_info(**kwargs)  # noqa: E501
+
+    def list_databricks_connections_api_v2_integrations_connections_databricks_get_with_http_info(self, **kwargs):  # noqa: E501
+        """List Databricks Connections  # noqa: E501
+
+        List all registered Databricks connections  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_databricks_connections_api_v2_integrations_connections_databricks_get_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param ValidationStatus validation_status:
+        :param ConnectionType connection_type:
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(ListDatabricksConnections, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'validation_status',
+            'connection_type'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method list_databricks_connections_api_v2_integrations_connections_databricks_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'validation_status' in local_var_params and local_var_params['validation_status'] is not None:  # noqa: E501
+            query_params.append(('validation_status', local_var_params['validation_status']))  # noqa: E501
+        if 'connection_type' in local_var_params and local_var_params['connection_type'] is not None:  # noqa: E501
+            query_params.append(('connection_type', local_var_params['connection_type']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/integrations/connections/databricks', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='ListDatabricksConnections',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def list_dataset_versions_api_v2_datasets_versions_get(self, dataset_id, **kwargs):  # noqa: E501
         """List Dataset Versions  # noqa: E501
 
@@ -29429,6 +30273,129 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def oauth_callback_api_v2_integrations_oauth_callback_get(self, code, state, **kwargs):  # noqa: E501
+        """Oauth Callback  # noqa: E501
+
+        Handle OAuth callback from third party integration  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.oauth_callback_api_v2_integrations_oauth_callback_get(code, state, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str code: Authorization code from third party integration (required)
+        :param str state: State parameter for security (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: OauthconnectionresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.oauth_callback_api_v2_integrations_oauth_callback_get_with_http_info(code, state, **kwargs)  # noqa: E501
+
+    def oauth_callback_api_v2_integrations_oauth_callback_get_with_http_info(self, code, state, **kwargs):  # noqa: E501
+        """Oauth Callback  # noqa: E501
+
+        Handle OAuth callback from third party integration  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.oauth_callback_api_v2_integrations_oauth_callback_get_with_http_info(code, state, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str code: Authorization code from third party integration (required)
+        :param str state: State parameter for security (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(OauthconnectionresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'code',
+            'state'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method oauth_callback_api_v2_integrations_oauth_callback_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'code' is set
+        if self.api_client.client_side_validation and ('code' not in local_var_params or  # noqa: E501
+                                                        local_var_params['code'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `code` when calling `oauth_callback_api_v2_integrations_oauth_callback_get`")  # noqa: E501
+        # verify the required parameter 'state' is set
+        if self.api_client.client_side_validation and ('state' not in local_var_params or  # noqa: E501
+                                                        local_var_params['state'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `state` when calling `oauth_callback_api_v2_integrations_oauth_callback_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'code' in local_var_params and local_var_params['code'] is not None:  # noqa: E501
+            query_params.append(('code', local_var_params['code']))  # noqa: E501
+        if 'state' in local_var_params and local_var_params['state'] is not None:  # noqa: E501
+            query_params.append(('state', local_var_params['state']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/integrations/oauth/callback', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='OauthconnectionresponseResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def patch_workspace_api_v2_experimental_workspaces_workspace_id_patch(self, workspace_id, json_patch_operation, **kwargs):  # noqa: E501
         """Patch Workspace  # noqa: E501
 
@@ -31661,14 +32628,132 @@ class DefaultApi(object):
     def register_api_v2_kubernetes_manager_register_post_with_http_info(self, kubernetes_manager_registration_request, **kwargs):  # noqa: E501
         """Register  # noqa: E501
 
-        Register a kubernetes manager to receive the token.  # noqa: E501
+        Register a kubernetes manager to receive the token.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.register_api_v2_kubernetes_manager_register_post_with_http_info(kubernetes_manager_registration_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param KubernetesManagerRegistrationRequest kubernetes_manager_registration_request: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(KubernetesmanagerregistrationresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'kubernetes_manager_registration_request'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method register_api_v2_kubernetes_manager_register_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'kubernetes_manager_registration_request' is set
+        if self.api_client.client_side_validation and ('kubernetes_manager_registration_request' not in local_var_params or  # noqa: E501
+                                                        local_var_params['kubernetes_manager_registration_request'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `kubernetes_manager_registration_request` when calling `register_api_v2_kubernetes_manager_register_post`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'kubernetes_manager_registration_request' in local_var_params:
+            body_params = local_var_params['kubernetes_manager_registration_request']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/kubernetes_manager/register', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='KubernetesmanagerregistrationresponseResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def register_build_api_v2_anyscaled_register_build_post(self, build_registration, **kwargs):  # noqa: E501
+        """Register Build  # noqa: E501
+
+        Register a node to receive a per-build token.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.register_build_api_v2_anyscaled_register_build_post(build_registration, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param BuildRegistration build_registration: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: AnyscaledcredentialresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.register_build_api_v2_anyscaled_register_build_post_with_http_info(build_registration, **kwargs)  # noqa: E501
+
+    def register_build_api_v2_anyscaled_register_build_post_with_http_info(self, build_registration, **kwargs):  # noqa: E501
+        """Register Build  # noqa: E501
+
+        Register a node to receive a per-build token.  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.register_api_v2_kubernetes_manager_register_post_with_http_info(kubernetes_manager_registration_request, async_req=True)
+        >>> thread = api.register_build_api_v2_anyscaled_register_build_post_with_http_info(build_registration, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param KubernetesManagerRegistrationRequest kubernetes_manager_registration_request: (required)
+        :param BuildRegistration build_registration: (required)
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -31678,7 +32763,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(KubernetesmanagerregistrationresponseResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(AnyscaledcredentialresponseResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -31686,7 +32771,7 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'kubernetes_manager_registration_request'
+            'build_registration'
         ]
         all_params.extend(
             [
@@ -31701,14 +32786,14 @@ class DefaultApi(object):
             if key not in all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method register_api_v2_kubernetes_manager_register_post" % key
+                    " to method register_build_api_v2_anyscaled_register_build_post" % key
                 )
             local_var_params[key] = val
         del local_var_params['kwargs']
-        # verify the required parameter 'kubernetes_manager_registration_request' is set
-        if self.api_client.client_side_validation and ('kubernetes_manager_registration_request' not in local_var_params or  # noqa: E501
-                                                        local_var_params['kubernetes_manager_registration_request'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `kubernetes_manager_registration_request` when calling `register_api_v2_kubernetes_manager_register_post`")  # noqa: E501
+        # verify the required parameter 'build_registration' is set
+        if self.api_client.client_side_validation and ('build_registration' not in local_var_params or  # noqa: E501
+                                                        local_var_params['build_registration'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `build_registration` when calling `register_build_api_v2_anyscaled_register_build_post`")  # noqa: E501
 
         collection_formats = {}
 
@@ -31722,8 +32807,8 @@ class DefaultApi(object):
         local_var_files = {}
 
         body_params = None
-        if 'kubernetes_manager_registration_request' in local_var_params:
-            body_params = local_var_params['kubernetes_manager_registration_request']
+        if 'build_registration' in local_var_params:
+            body_params = local_var_params['build_registration']
         # HTTP header `Accept`
         header_params['Accept'] = self.api_client.select_header_accept(
             ['application/json'])  # noqa: E501
@@ -31736,14 +32821,14 @@ class DefaultApi(object):
         auth_settings = []  # noqa: E501
 
         return self.api_client.call_api(
-            '/api/v2/kubernetes_manager/register', 'POST',
+            '/api/v2/anyscaled/register_build', 'POST',
             path_params,
             query_params,
             header_params,
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='KubernetesmanagerregistrationresponseResponse',  # noqa: E501
+            response_type='AnyscaledcredentialresponseResponse',  # noqa: E501
             auth_settings=auth_settings,
             async_req=local_var_params.get('async_req'),
             _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
@@ -31751,17 +32836,17 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def register_build_api_v2_anyscaled_register_build_post(self, build_registration, **kwargs):  # noqa: E501
-        """Register Build  # noqa: E501
+    def register_databricks_connection_api_v2_integrations_connections_databricks_post(self, databricks_register_request, **kwargs):  # noqa: E501
+        """Register Databricks Connection  # noqa: E501
 
-        Register a node to receive a per-build token.  # noqa: E501
+        Register Databricks connection configuration  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.register_build_api_v2_anyscaled_register_build_post(build_registration, async_req=True)
+        >>> thread = api.register_databricks_connection_api_v2_integrations_connections_databricks_post(databricks_register_request, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param BuildRegistration build_registration: (required)
+        :param DatabricksRegisterRequest databricks_register_request: (required)
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -31769,24 +32854,24 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: AnyscaledcredentialresponseResponse
+        :return: DatabricksconnectionresponseResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
-        return self.register_build_api_v2_anyscaled_register_build_post_with_http_info(build_registration, **kwargs)  # noqa: E501
+        return self.register_databricks_connection_api_v2_integrations_connections_databricks_post_with_http_info(databricks_register_request, **kwargs)  # noqa: E501
 
-    def register_build_api_v2_anyscaled_register_build_post_with_http_info(self, build_registration, **kwargs):  # noqa: E501
-        """Register Build  # noqa: E501
+    def register_databricks_connection_api_v2_integrations_connections_databricks_post_with_http_info(self, databricks_register_request, **kwargs):  # noqa: E501
+        """Register Databricks Connection  # noqa: E501
 
-        Register a node to receive a per-build token.  # noqa: E501
+        Register Databricks connection configuration  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.register_build_api_v2_anyscaled_register_build_post_with_http_info(build_registration, async_req=True)
+        >>> thread = api.register_databricks_connection_api_v2_integrations_connections_databricks_post_with_http_info(databricks_register_request, async_req=True)
         >>> result = thread.get()
 
         :param async_req bool: execute request asynchronously
-        :param BuildRegistration build_registration: (required)
+        :param DatabricksRegisterRequest databricks_register_request: (required)
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -31796,7 +32881,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(AnyscaledcredentialresponseResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(DatabricksconnectionresponseResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -31804,7 +32889,7 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'build_registration'
+            'databricks_register_request'
         ]
         all_params.extend(
             [
@@ -31819,14 +32904,14 @@ class DefaultApi(object):
             if key not in all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method register_build_api_v2_anyscaled_register_build_post" % key
+                    " to method register_databricks_connection_api_v2_integrations_connections_databricks_post" % key
                 )
             local_var_params[key] = val
         del local_var_params['kwargs']
-        # verify the required parameter 'build_registration' is set
-        if self.api_client.client_side_validation and ('build_registration' not in local_var_params or  # noqa: E501
-                                                        local_var_params['build_registration'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `build_registration` when calling `register_build_api_v2_anyscaled_register_build_post`")  # noqa: E501
+        # verify the required parameter 'databricks_register_request' is set
+        if self.api_client.client_side_validation and ('databricks_register_request' not in local_var_params or  # noqa: E501
+                                                        local_var_params['databricks_register_request'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `databricks_register_request` when calling `register_databricks_connection_api_v2_integrations_connections_databricks_post`")  # noqa: E501
 
         collection_formats = {}
 
@@ -31840,8 +32925,8 @@ class DefaultApi(object):
         local_var_files = {}
 
         body_params = None
-        if 'build_registration' in local_var_params:
-            body_params = local_var_params['build_registration']
+        if 'databricks_register_request' in local_var_params:
+            body_params = local_var_params['databricks_register_request']
         # HTTP header `Accept`
         header_params['Accept'] = self.api_client.select_header_accept(
             ['application/json'])  # noqa: E501
@@ -31854,14 +32939,14 @@ class DefaultApi(object):
         auth_settings = []  # noqa: E501
 
         return self.api_client.call_api(
-            '/api/v2/anyscaled/register_build', 'POST',
+            '/api/v2/integrations/connections/databricks', 'POST',
             path_params,
             query_params,
             header_params,
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='AnyscaledcredentialresponseResponse',  # noqa: E501
+            response_type='DatabricksconnectionresponseResponse',  # noqa: E501
             auth_settings=auth_settings,
             async_req=local_var_params.get('async_req'),
             _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
@@ -35110,6 +36195,120 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get(self, organization_id, **kwargs):  # noqa: E501
+        """Sso Admin Link  # noqa: E501
+
+        Get the admin link for the organization's SSO configuration.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: SsoconfigResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get_with_http_info(organization_id, **kwargs)  # noqa: E501
+
+    def sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get_with_http_info(self, organization_id, **kwargs):  # noqa: E501
+        """Sso Admin Link  # noqa: E501
+
+        Get the admin link for the organization's SSO configuration.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get_with_http_info(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(SsoconfigResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'organization_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'organization_id' is set
+        if self.api_client.client_side_validation and ('organization_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['organization_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `organization_id` when calling `sso_admin_link_api_v2_organizations_organization_id_sso_admin_link_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'organization_id' in local_var_params:
+            path_params['organization_id'] = local_var_params['organization_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/organizations/{organization_id}/sso_admin_link', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='SsoconfigResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def sso_login_info_api_v2_users_sso_login_info_get(self, organization_id, **kwargs):  # noqa: E501
         """Sso Login Info  # noqa: E501
 
@@ -39045,6 +40244,133 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def update_template_api_v2_workspace_templates_template_id_put(self, template_id, update_workspace_template, **kwargs):  # noqa: E501
+        """Update Template  # noqa: E501
+
+        Update a workspace template  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_template_api_v2_workspace_templates_template_id_put(template_id, update_workspace_template, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str template_id: (required)
+        :param UpdateWorkspaceTemplate update_workspace_template: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: WorkspacetemplateResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.update_template_api_v2_workspace_templates_template_id_put_with_http_info(template_id, update_workspace_template, **kwargs)  # noqa: E501
+
+    def update_template_api_v2_workspace_templates_template_id_put_with_http_info(self, template_id, update_workspace_template, **kwargs):  # noqa: E501
+        """Update Template  # noqa: E501
+
+        Update a workspace template  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_template_api_v2_workspace_templates_template_id_put_with_http_info(template_id, update_workspace_template, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str template_id: (required)
+        :param UpdateWorkspaceTemplate update_workspace_template: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(WorkspacetemplateResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'template_id',
+            'update_workspace_template'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method update_template_api_v2_workspace_templates_template_id_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'template_id' is set
+        if self.api_client.client_side_validation and ('template_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['template_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `template_id` when calling `update_template_api_v2_workspace_templates_template_id_put`")  # noqa: E501
+        # verify the required parameter 'update_workspace_template' is set
+        if self.api_client.client_side_validation and ('update_workspace_template' not in local_var_params or  # noqa: E501
+                                                        local_var_params['update_workspace_template'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `update_workspace_template` when calling `update_template_api_v2_workspace_templates_template_id_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'template_id' in local_var_params:
+            path_params['template_id'] = local_var_params['template_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'update_workspace_template' in local_var_params:
+            body_params = local_var_params['update_workspace_template']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/workspace_templates/{template_id}', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='WorkspacetemplateResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post(self, session_command_id, **kwargs):  # noqa: E501
         """Upload Session Command Logs  # noqa: E501