lightning-sdk 0.1.41__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/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
@@ -522,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
@@ -576,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
@@ -688,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
@@ -704,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
@@ -733,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

lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py

@@ -43,6 +43,7 @@ class DeploymenttemplatesIdBody(object):
     swagger_types = {
         'about_page_content': 'str',
         'categories': 'list[str]',
+        'cloud_space_id': 'str',
         'description': 'str',
         'featured': 'bool',
         'image_url': 'str',
@@ -61,6 +62,7 @@ class DeploymenttemplatesIdBody(object):
     attribute_map = {
         'about_page_content': 'aboutPageContent',
         'categories': 'categories',
+        'cloud_space_id': 'cloudSpaceId',
         'description': 'description',
         'featured': 'featured',
         'image_url': 'imageUrl',
@@ -76,10 +78,11 @@ class DeploymenttemplatesIdBody(object):
         'visibility': 'visibility'
     }
 
-    def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'V1DeploymentMetrics' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pricing: 'V1ApiPricingSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None):  # noqa: E501
+    def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, cloud_space_id: 'str' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'V1DeploymentMetrics' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pricing: 'V1ApiPricingSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None):  # noqa: E501
         """DeploymenttemplatesIdBody - a model defined in Swagger"""  # noqa: E501
         self._about_page_content = None
         self._categories = None
+        self._cloud_space_id = None
         self._description = None
         self._featured = None
         self._image_url = None
@@ -98,6 +101,8 @@ class DeploymenttemplatesIdBody(object):
             self.about_page_content = about_page_content
         if categories is not None:
             self.categories = categories
+        if cloud_space_id is not None:
+            self.cloud_space_id = cloud_space_id
         if description is not None:
             self.description = description
         if featured is not None:
@@ -167,6 +172,27 @@ class DeploymenttemplatesIdBody(object):
 
         self._categories = categories
 
+    @property
+    def cloud_space_id(self) -> 'str':
+        """Gets the cloud_space_id of this DeploymenttemplatesIdBody.  # noqa: E501
+
+
+        :return: The cloud_space_id of this DeploymenttemplatesIdBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloud_space_id
+
+    @cloud_space_id.setter
+    def cloud_space_id(self, cloud_space_id: 'str'):
+        """Sets the cloud_space_id of this DeploymenttemplatesIdBody.
+
+
+        :param cloud_space_id: The cloud_space_id of this DeploymenttemplatesIdBody.  # noqa: E501
+        :type: str
+        """
+
+        self._cloud_space_id = cloud_space_id
+
     @property
     def description(self) -> 'str':
         """Gets the description of this DeploymenttemplatesIdBody.  # noqa: E501