lightning-sdk 0.1.40__py3-none-any.whl → 0.1.42__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py

@@ -1009,6 +1009,111 @@ class LitLoggerServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_logger_service_get_metrics_stream(self, **kwargs) -> 'V1MetricsStream':  # noqa: E501
+        """lit_logger_service_get_metrics_stream  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_logger_service_get_metrics_stream(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_name:
+        :param str project_owner_name:
+        :param str name:
+        :param str id:
+        :param int version_number:
+        :return: V1MetricsStream
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.lit_logger_service_get_metrics_stream_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_logger_service_get_metrics_stream_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def lit_logger_service_get_metrics_stream_with_http_info(self, **kwargs) -> 'V1MetricsStream':  # noqa: E501
+        """lit_logger_service_get_metrics_stream  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_logger_service_get_metrics_stream_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_name:
+        :param str project_owner_name:
+        :param str name:
+        :param str id:
+        :param int version_number:
+        :return: V1MetricsStream
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_name', 'project_owner_name', 'name', 'id', 'version_number']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method lit_logger_service_get_metrics_stream" % key
+                )
+            params[key] = val
+        del params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'project_name' in params:
+            query_params.append(('projectName', params['project_name']))  # noqa: E501
+        if 'project_owner_name' in params:
+            query_params.append(('projectOwnerName', params['project_owner_name']))  # noqa: E501
+        if 'name' in params:
+            query_params.append(('name', params['name']))  # noqa: E501
+        if 'id' in params:
+            query_params.append(('id', params['id']))  # noqa: E501
+        if 'version_number' in params:
+            query_params.append(('versionNumber', params['version_number']))  # 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(
+            '/v1/metrics-streams', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1MetricsStream',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def lit_logger_service_get_shared_metrics_stream(self, id: 'str', **kwargs) -> 'V1MetricsStream':  # noqa: E501
         """lit_logger_service_get_shared_metrics_stream  # noqa: E501
 
@@ -1429,6 +1534,119 @@ class LitLoggerServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
+        """lit_logger_service_update_metrics_stream_visibility  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_logger_service_update_metrics_stream_visibility(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateMetricsStreamVisibilityResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def lit_logger_service_update_metrics_stream_visibility_with_http_info(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
+        """lit_logger_service_update_metrics_stream_visibility  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateMetricsStreamVisibilityResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method lit_logger_service_update_metrics_stream_visibility" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `lit_logger_service_update_metrics_stream_visibility`")  # noqa: E501
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `lit_logger_service_update_metrics_stream_visibility`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `lit_logger_service_update_metrics_stream_visibility`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # 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(
+            '/v1/projects/{projectId}/metrics-streams/{id}/visibility', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateMetricsStreamVisibilityResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def lit_logger_service_update_shared_metrics_stream(self, body: 'LitloggermetricsIdBody', id: 'str', **kwargs) -> 'V1UpdateSharedMetricsStreamResponse':  # noqa: E501
         """lit_logger_service_update_shared_metrics_stream  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -943,6 +943,115 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_get_model_by_name(self, project_owner_name: 'str', project_name: 'str', model_name: 'str', **kwargs) -> 'V1Model':  # noqa: E501
+        """models_store_get_model_by_name  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_get_model_by_name(project_owner_name, project_name, model_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_owner_name: (required)
+        :param str project_name: (required)
+        :param str model_name: (required)
+        :return: V1Model
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_get_model_by_name_with_http_info(self, project_owner_name: 'str', project_name: 'str', model_name: 'str', **kwargs) -> 'V1Model':  # noqa: E501
+        """models_store_get_model_by_name  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_get_model_by_name_with_http_info(project_owner_name, project_name, model_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_owner_name: (required)
+        :param str project_name: (required)
+        :param str model_name: (required)
+        :return: V1Model
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_owner_name', 'project_name', 'model_name']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method models_store_get_model_by_name" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_owner_name' is set
+        if ('project_owner_name' not in params or
+                params['project_owner_name'] is None):
+            raise ValueError("Missing the required parameter `project_owner_name` when calling `models_store_get_model_by_name`")  # noqa: E501
+        # verify the required parameter 'project_name' is set
+        if ('project_name' not in params or
+                params['project_name'] is None):
+            raise ValueError("Missing the required parameter `project_name` when calling `models_store_get_model_by_name`")  # noqa: E501
+        # verify the required parameter 'model_name' is set
+        if ('model_name' not in params or
+                params['model_name'] is None):
+            raise ValueError("Missing the required parameter `model_name` when calling `models_store_get_model_by_name`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_owner_name' in params:
+            path_params['projectOwnerName'] = params['project_owner_name']  # noqa: E501
+        if 'project_name' in params:
+            path_params['projectName'] = params['project_name']  # noqa: E501
+        if 'model_name' in params:
+            path_params['modelName'] = params['model_name']  # 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(
+            '/v1/projects/{projectOwnerName}/{projectName}/models/{modelName}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1Model',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def models_store_get_model_file_upload_urls(self, body: 'UploadIdPartsBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse':  # noqa: E501
         """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts.  # noqa: E501
 
@@ -1294,6 +1403,123 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_get_model_files_url(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1GetModelFilesUrlResponse':  # noqa: E501
+        """GetModelFileUrl responds with a pre-signed url for a given file  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_get_model_files_url(project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (required)
+        :param str page_size:
+        :param str offset:
+        :return: V1GetModelFilesUrlResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.models_store_get_model_files_url_with_http_info(project_id, model_id, version, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_get_model_files_url_with_http_info(project_id, model_id, version, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_get_model_files_url_with_http_info(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1GetModelFilesUrlResponse':  # noqa: E501
+        """GetModelFileUrl responds with a pre-signed url for a given file  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_get_model_files_url_with_http_info(project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (required)
+        :param str page_size:
+        :param str offset:
+        :return: V1GetModelFilesUrlResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'model_id', 'version', 'page_size', 'offset']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method models_store_get_model_files_url" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `models_store_get_model_files_url`")  # noqa: E501
+        # verify the required parameter 'model_id' is set
+        if ('model_id' not in params or
+                params['model_id'] is None):
+            raise ValueError("Missing the required parameter `model_id` when calling `models_store_get_model_files_url`")  # noqa: E501
+        # verify the required parameter 'version' is set
+        if ('version' not in params or
+                params['version'] is None):
+            raise ValueError("Missing the required parameter `version` when calling `models_store_get_model_files_url`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'model_id' in params:
+            path_params['modelId'] = params['model_id']  # noqa: E501
+        if 'version' in params:
+            path_params['version'] = params['version']  # noqa: E501
+
+        query_params = []
+        if 'page_size' in params:
+            query_params.append(('pageSize', params['page_size']))  # noqa: E501
+        if 'offset' in params:
+            query_params.append(('offset', params['offset']))  # 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(
+            '/v1/projects/{projectId}/models/{modelId}/versions/{version}/files', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetModelFilesUrlResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def models_store_get_model_version(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
         """GetModelVersion used to get specific model version details  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py

@@ -643,6 +643,7 @@ class SecretServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str type:
         :return: V1ListSecretsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -664,12 +665,13 @@ class SecretServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str type:
         :return: V1ListSecretsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'type']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -696,6 +698,8 @@ class SecretServiceApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'type' in params:
+            query_params.append(('type', params['type']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py

@@ -53,6 +53,11 @@ class SnowflakeServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str username:
+        :param str password:
+        :param str account:
+        :param str private_key_file:
+        :param str private_key_password:
         :return: V1CheckSnowflakeConnectionResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -74,12 +79,17 @@ class SnowflakeServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str username:
+        :param str password:
+        :param str account:
+        :param str private_key_file:
+        :param str private_key_password:
         :return: V1CheckSnowflakeConnectionResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'username', 'password', 'account', 'private_key_file', 'private_key_password']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -106,6 +116,16 @@ class SnowflakeServiceApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'username' in params:
+            query_params.append(('username', params['username']))  # noqa: E501
+        if 'password' in params:
+            query_params.append(('password', params['password']))  # noqa: E501
+        if 'account' in params:
+            query_params.append(('account', params['account']))  # noqa: E501
+        if 'private_key_file' in params:
+            query_params.append(('privateKeyFile', params['private_key_file']))  # noqa: E501
+        if 'private_key_password' in params:
+            query_params.append(('privateKeyPassword', params['private_key_password']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -89,6 +89,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBo
 from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
+from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -409,6 +410,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
@@ -424,6 +426,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_response import V1GetUserStorageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
+from lightning_sdk.lightning_cloud.openapi.models.v1_header import V1Header
 from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_exec import V1HealthCheckExec
 from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get import V1HealthCheckHttpGet
 from lightning_sdk.lightning_cloud.openapi.models.v1_ids_logger_metrics import V1IdsLoggerMetrics
@@ -521,6 +524,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
@@ -575,9 +579,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_stream import V1Met
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1MetricsTags
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
 from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
+from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
 from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job import V1MultiMachineJob
+from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_event import V1MultiMachineJobEvent
+from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_event_type import V1MultiMachineJobEventType
+from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_fault_tolerance import V1MultiMachineJobFaultTolerance
+from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_fault_tolerance_strategy import V1MultiMachineJobFaultToleranceStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_state import V1MultiMachineJobState
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_status import V1MultiMachineJobStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics import V1NamedGetLoggerMetrics
@@ -687,6 +696,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1Sys
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
+from lightning_sdk.lightning_cloud.openapi.models.v1_trainium_system_metrics import V1TrainiumSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_project_balance_response import V1TransferProjectBalanceResponse
@@ -703,6 +713,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
@@ -732,6 +743,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_request import V1ValidateManagedEndpointRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_response import V1ValidateManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse