lightning-sdk 0.1.30__py3-none-any.whl → 0.1.31__py3-none-any.whl

lightning_sdk/__init__.py

@@ -25,5 +25,5 @@ __all__ = [
     "Agent",
 ]
 
-__version__ = "0.1.30"
+__version__ = "0.1.31"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/teamspace_api.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Optional
 from lightning_sdk.api.utils import _download_model_files, _DummyBody, _ModelFileUploader
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
+    ModelIdVersionsBody,
     ModelsStoreApi,
     ProjectIdAgentsBody,
     ProjectIdModelsBody,
@@ -28,6 +29,7 @@ class TeamspaceApi:
 
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=7)
+        self._models: Optional[ModelsStoreApi] = None
 
     def get_teamspace(self, name: str, owner_id: str) -> V1Project:
         """Get the current teamspace from the owner."""
@@ -165,9 +167,36 @@ class TeamspaceApi:
         teamspace_id: str,
         cluster_id: str,
     ) -> V1ModelVersionArchive:
-        api = ModelsStoreApi(self._client.api_client)
-        body = ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private)
-        return api.models_store_create_model(body, project_id=teamspace_id)
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        # ask if such model already exists by listing models with specific name
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        if len(models) == 0:
+            return self._models.models_store_create_model(
+                body=ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private),
+                project_id=teamspace_id,
+            )
+        assert len(models) == 1, "Multiple models with the same name found"
+        return self._models.models_store_create_model_version(
+            body=ModelIdVersionsBody(cluster_id=cluster_id),
+            project_id=teamspace_id,
+            model_id=models[0].id,
+        )
+
+    def delete_model(self, name: str, version: Optional[str], teamspace_id: str) -> None:
+        """Delete a model or a version from the model store."""
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        assert len(models) == 1, "Multiple models with the same name found"
+        model_id = models[0].id
+        # decide if delete only version of whole model
+        if version:
+            if version == "latest":
+                version = models[0].latest_version
+            self._models.models_store_delete_model_version(project_id=teamspace_id, model_id=model_id, version=version)
+        else:
+            self._models.models_store_delete_model(project_id=teamspace_id, model_id=model_id)
 
     def upload_model_file(
         self,
@@ -213,8 +242,9 @@ class TeamspaceApi:
             )
 
     def complete_model_upload(self, model_id: str, version: str, teamspace_id: str) -> None:
-        api = ModelsStoreApi(self._client.api_client)
-        api.models_store_complete_model_upload(
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        self._models.models_store_complete_model_upload(
             body=_DummyBody(),
             project_id=teamspace_id,
             model_id=model_id,

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -104,6 +104,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCo
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
 from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
+from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body import IdExecuteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body1 import IdExecuteBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_fork_body import IdForkBody
@@ -133,6 +134,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifa
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
+from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -359,6 +361,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1D
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_status import V1DeploymentStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_strategy import V1DeploymentStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template import V1DeploymentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_engagement_response import V1DeploymentTemplateEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import V1DeploymentTemplateGalleryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter import V1DeploymentTemplateParameter
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_placement import V1DeploymentTemplateParameterPlacement
@@ -591,6 +594,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_org_role import V1OrgRole
 from lightning_sdk.lightning_cloud.openapi.models.v1_organization import V1Organization
 from lightning_sdk.lightning_cloud.openapi.models.v1_owner_type import V1OwnerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1PackageManager
+from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -1955,7 +1955,7 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1964,7 +1964,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse
@@ -1978,7 +1978,7 @@ class CloudSpaceServiceApi(object):
             (data) = self.cloud_space_service_engage_cloud_space_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
             return data
 
-    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1987,7 +1987,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1106,10 +1106,10 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1134,16 +1134,16 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'capacity_block_duration_hours', 'start_date', 'timezone', 'org_id']  # noqa: E501
+        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_days']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1180,14 +1180,14 @@ class ClusterServiceApi(object):
             query_params.append(('instanceType', params['instance_type']))  # noqa: E501
         if 'instance_count' in params:
             query_params.append(('instanceCount', params['instance_count']))  # noqa: E501
-        if 'capacity_block_duration_hours' in params:
-            query_params.append(('capacityBlockDurationHours', params['capacity_block_duration_hours']))  # noqa: E501
         if 'start_date' in params:
             query_params.append(('startDate', params['start_date']))  # noqa: E501
         if 'timezone' in params:
             query_params.append(('timezone', params['timezone']))  # noqa: E501
         if 'org_id' in params:
             query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'capacity_block_duration_days' in params:
+            query_params.append(('capacityBlockDurationDays', params['capacity_block_duration_days']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py

@@ -140,6 +140,111 @@ class DeploymentTemplatesServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def deployment_templates_service_engage_deployment_template(self, body: 'IdEngageBody', id: 'str', **kwargs) -> 'V1DeploymentTemplateEngagementResponse':  # noqa: E501
+        """deployment_templates_service_engage_deployment_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.deployment_templates_service_engage_deployment_template(body, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdEngageBody body: (required)
+        :param str id: (required)
+        :return: V1DeploymentTemplateEngagementResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.deployment_templates_service_engage_deployment_template_with_http_info(body, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.deployment_templates_service_engage_deployment_template_with_http_info(body, id, **kwargs)  # noqa: E501
+            return data
+
+    def deployment_templates_service_engage_deployment_template_with_http_info(self, body: 'IdEngageBody', id: 'str', **kwargs) -> 'V1DeploymentTemplateEngagementResponse':  # noqa: E501
+        """deployment_templates_service_engage_deployment_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.deployment_templates_service_engage_deployment_template_with_http_info(body, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdEngageBody body: (required)
+        :param str id: (required)
+        :return: V1DeploymentTemplateEngagementResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', '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 deployment_templates_service_engage_deployment_template" % 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 `deployment_templates_service_engage_deployment_template`")  # 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 `deployment_templates_service_engage_deployment_template`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        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/deployment-templates/{id}/engage', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1DeploymentTemplateEngagementResponse',  # 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 deployment_templates_service_get_deployment_template(self, id: 'str', **kwargs) -> 'V1DeploymentTemplate':  # noqa: E501
         """deployment_templates_service_get_deployment_template  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py

@@ -365,6 +365,7 @@ class JobsServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str id: (required)
+        :param bool delete_templates:
         :return: V1DeleteDeploymentResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -387,12 +388,13 @@ class JobsServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str id: (required)
+        :param bool delete_templates:
         :return: V1DeleteDeploymentResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'id']  # noqa: E501
+        all_params = ['project_id', 'id', 'delete_templates']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -425,6 +427,8 @@ class JobsServiceApi(object):
             path_params['id'] = params['id']  # noqa: E501
 
         query_params = []
+        if 'delete_templates' in params:
+            query_params.append(('deleteTemplates', params['delete_templates']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -28,7 +28,6 @@ import six
 from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
 
 if TYPE_CHECKING:
-    from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
 class ModelsStoreApi(object):
@@ -398,6 +397,119 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_create_model_version(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """models_store_create_model_version  # 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_create_model_version(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVersionsBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1ModelVersionArchive
+                 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_create_model_version_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_create_model_version_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_create_model_version_with_http_info(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """models_store_create_model_version  # 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_create_model_version_with_http_info(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVersionsBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1ModelVersionArchive
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'model_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 models_store_create_model_version" % 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 `models_store_create_model_version`")  # 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 `models_store_create_model_version`")  # 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_create_model_version`")  # 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
+
+        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}/models/{modelId}/versions', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ModelVersionArchive',  # 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_create_multi_part_upload(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse':  # noqa: E501
         """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files.  # noqa: E501
 
@@ -1280,6 +1392,7 @@ class ModelsStoreApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str name:
         :return: V1ListModelsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1301,12 +1414,13 @@ class ModelsStoreApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str name:
         :return: V1ListModelsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'name']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1333,6 +1447,8 @@ class ModelsStoreApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'name' in params:
+            query_params.append(('name', params['name']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -69,6 +69,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCo
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
 from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
+from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body import IdExecuteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body1 import IdExecuteBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_fork_body import IdForkBody
@@ -98,6 +99,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifa
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
+from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -324,6 +326,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1D
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_status import V1DeploymentStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_strategy import V1DeploymentStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template import V1DeploymentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_engagement_response import V1DeploymentTemplateEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import V1DeploymentTemplateGalleryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter import V1DeploymentTemplateParameter
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_placement import V1DeploymentTemplateParameterPlacement
@@ -556,6 +559,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_org_role import V1OrgRole
 from lightning_sdk.lightning_cloud.openapi.models.v1_organization import V1Organization
 from lightning_sdk.lightning_cloud.openapi.models.v1_owner_type import V1OwnerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1PackageManager
+from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin

lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py

@@ -41,7 +41,7 @@ class ClusterIdCapacityblockBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'capacity_block_duration_hours': 'int',
+        'capacity_block_duration_days': 'int',
         'instance_count': 'int',
         'instance_type': 'str',
         'org_id': 'str',
@@ -50,7 +50,7 @@ class ClusterIdCapacityblockBody(object):
     }
 
     attribute_map = {
-        'capacity_block_duration_hours': 'capacityBlockDurationHours',
+        'capacity_block_duration_days': 'capacityBlockDurationDays',
         'instance_count': 'instanceCount',
         'instance_type': 'instanceType',
         'org_id': 'orgId',
@@ -58,17 +58,17 @@ class ClusterIdCapacityblockBody(object):
         'timezone': 'timezone'
     }
 
-    def __init__(self, capacity_block_duration_hours: 'int' =None, instance_count: 'int' =None, instance_type: 'str' =None, org_id: 'str' =None, start_date: 'datetime' =None, timezone: 'str' =None):  # noqa: E501
+    def __init__(self, capacity_block_duration_days: 'int' =None, instance_count: 'int' =None, instance_type: 'str' =None, org_id: 'str' =None, start_date: 'datetime' =None, timezone: 'str' =None):  # noqa: E501
         """ClusterIdCapacityblockBody - a model defined in Swagger"""  # noqa: E501
-        self._capacity_block_duration_hours = None
+        self._capacity_block_duration_days = None
         self._instance_count = None
         self._instance_type = None
         self._org_id = None
         self._start_date = None
         self._timezone = None
         self.discriminator = None
-        if capacity_block_duration_hours is not None:
-            self.capacity_block_duration_hours = capacity_block_duration_hours
+        if capacity_block_duration_days is not None:
+            self.capacity_block_duration_days = capacity_block_duration_days
         if instance_count is not None:
             self.instance_count = instance_count
         if instance_type is not None:
@@ -81,25 +81,25 @@ class ClusterIdCapacityblockBody(object):
             self.timezone = timezone
 
     @property
-    def capacity_block_duration_hours(self) -> 'int':
-        """Gets the capacity_block_duration_hours of this ClusterIdCapacityblockBody.  # noqa: E501
+    def capacity_block_duration_days(self) -> 'int':
+        """Gets the capacity_block_duration_days of this ClusterIdCapacityblockBody.  # noqa: E501
 
 
-        :return: The capacity_block_duration_hours of this ClusterIdCapacityblockBody.  # noqa: E501
+        :return: The capacity_block_duration_days of this ClusterIdCapacityblockBody.  # noqa: E501
         :rtype: int
         """
-        return self._capacity_block_duration_hours
+        return self._capacity_block_duration_days
 
-    @capacity_block_duration_hours.setter
-    def capacity_block_duration_hours(self, capacity_block_duration_hours: 'int'):
-        """Sets the capacity_block_duration_hours of this ClusterIdCapacityblockBody.
+    @capacity_block_duration_days.setter
+    def capacity_block_duration_days(self, capacity_block_duration_days: 'int'):
+        """Sets the capacity_block_duration_days of this ClusterIdCapacityblockBody.
 
 
-        :param capacity_block_duration_hours: The capacity_block_duration_hours of this ClusterIdCapacityblockBody.  # noqa: E501
+        :param capacity_block_duration_days: The capacity_block_duration_days of this ClusterIdCapacityblockBody.  # noqa: E501
         :type: int
         """
 
-        self._capacity_block_duration_hours = capacity_block_duration_hours
+        self._capacity_block_duration_days = capacity_block_duration_days
 
     @property
     def instance_count(self) -> 'int':