lightning-sdk 0.1.46__py3-none-any.whl → 0.1.48__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py

@@ -148,6 +148,107 @@ class LitRegistryServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_registry_service_delete_lit_repository(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1DeleteLitRepositoryResponse':  # noqa: E501
+        """lit_registry_service_delete_lit_repository  # 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_registry_service_delete_lit_repository(project_id, lit_repo_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :return: V1DeleteLitRepositoryResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.lit_registry_service_delete_lit_repository_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_registry_service_delete_lit_repository_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+            return data
+
+    def lit_registry_service_delete_lit_repository_with_http_info(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1DeleteLitRepositoryResponse':  # noqa: E501
+        """lit_registry_service_delete_lit_repository  # 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_registry_service_delete_lit_repository_with_http_info(project_id, lit_repo_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :return: V1DeleteLitRepositoryResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'lit_repo_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 lit_registry_service_delete_lit_repository" % 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 `lit_registry_service_delete_lit_repository`")  # noqa: E501
+        # verify the required parameter 'lit_repo_name' is set
+        if ('lit_repo_name' not in params or
+                params['lit_repo_name'] is None):
+            raise ValueError("Missing the required parameter `lit_repo_name` when calling `lit_registry_service_delete_lit_repository`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'lit_repo_name' in params:
+            path_params['litRepoName'] = params['lit_repo_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/{projectId}/lit-registry/{litRepoName}', 'DELETE',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1DeleteLitRepositoryResponse',  # 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_registry_service_get_lit_project_registry(self, project_id: 'str', **kwargs) -> 'V1LitRegistryProject':  # noqa: E501
         """lit_registry_service_get_lit_project_registry  # noqa: E501
 
@@ -240,3 +341,104 @@ class LitRegistryServiceApi(object):
             _preload_content=params.get('_preload_content', True),
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
+
+    def lit_registry_service_get_lit_repository(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1LitRepository':  # noqa: E501
+        """lit_registry_service_get_lit_repository  # 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_registry_service_get_lit_repository(project_id, lit_repo_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :return: V1LitRepository
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.lit_registry_service_get_lit_repository_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_registry_service_get_lit_repository_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+            return data
+
+    def lit_registry_service_get_lit_repository_with_http_info(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1LitRepository':  # noqa: E501
+        """lit_registry_service_get_lit_repository  # 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_registry_service_get_lit_repository_with_http_info(project_id, lit_repo_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :return: V1LitRepository
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'lit_repo_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 lit_registry_service_get_lit_repository" % 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 `lit_registry_service_get_lit_repository`")  # noqa: E501
+        # verify the required parameter 'lit_repo_name' is set
+        if ('lit_repo_name' not in params or
+                params['lit_repo_name'] is None):
+            raise ValueError("Missing the required parameter `lit_repo_name` when calling `lit_registry_service_get_lit_repository`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'lit_repo_name' in params:
+            path_params['litRepoName'] = params['lit_repo_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/{projectId}/lit-registry/{litRepoName}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1LitRepository',  # 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)

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -294,7 +294,7 @@ class ModelsStoreApi(object):
             collection_formats=collection_formats)
 
     def models_store_create_model(self, body: 'ProjectIdModelsBody', project_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
-        """CreateModel creates a new version of a model and marks the upload as incomplete. Files are expected to be uploaded using the multi-part upload APIs below, followed by a completion request after which the model will be fully visible (can list, delete etc.)  # noqa: E501
+        """CreateModel creates a new model.  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -316,7 +316,7 @@ class ModelsStoreApi(object):
             return data
 
     def models_store_create_model_with_http_info(self, body: 'ProjectIdModelsBody', project_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
-        """CreateModel creates a new version of a model and marks the upload as incomplete. Files are expected to be uploaded using the multi-part upload APIs below, followed by a completion request after which the model will be fully visible (can list, delete etc.)  # noqa: E501
+        """CreateModel creates a new model.  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -399,7 +399,7 @@ class ModelsStoreApi(object):
             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
+        """CreateModelVersion creates a new version of a model and marks the upload as incomplete. Files are expected to be uploaded using the multi-part upload APIs below, followed by a completion request after which the model will be fully visible (can list, delete etc.)  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -422,7 +422,7 @@ class ModelsStoreApi(object):
             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
+        """CreateModelVersion creates a new version of a model and marks the upload as incomplete. Files are expected to be uploaded using the multi-part upload APIs below, followed by a completion request after which the model will be fully visible (can list, delete etc.)  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1827,8 +1827,129 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_set_model_default_version(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model':  # noqa: E501
+        """models_store_set_model_default_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_set_model_default_version(body, project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VersionDefaultBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (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_set_model_default_version_with_http_info(body, project_id, model_id, version, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_set_model_default_version_with_http_info(body, project_id, model_id, version, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_set_model_default_version_with_http_info(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model':  # noqa: E501
+        """models_store_set_model_default_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_set_model_default_version_with_http_info(body, project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VersionDefaultBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (required)
+        :return: V1Model
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'model_id', 'version']  # 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_set_model_default_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_set_model_default_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_set_model_default_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_set_model_default_version`")  # 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_set_model_default_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
+        if 'version' in params:
+            path_params['version'] = params['version']  # 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/{version}/default', 'PUT',
+            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_update_model(self, body: 'ModelsModelIdBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1Model':  # noqa: E501
-        """models_store_update_model  # noqa: E501
+        """UpdateModel updates the model, in particular the model's name.  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1851,7 +1972,7 @@ class ModelsStoreApi(object):
             return data
 
     def models_store_update_model_with_http_info(self, body: 'ModelsModelIdBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1Model':  # noqa: E501
-        """models_store_update_model  # noqa: E501
+        """UpdateModel updates the model, in particular the model's name.  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1940,8 +2061,129 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_update_model_version(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """UpdateModelVersion updates the model version, in particular the model version's 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_update_model_version(body, project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VersionsVersionBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (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_update_model_version_with_http_info(body, project_id, model_id, version, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_update_model_version_with_http_info(body, project_id, model_id, version, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_update_model_version_with_http_info(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """UpdateModelVersion updates the model version, in particular the model version's 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_update_model_version_with_http_info(body, project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VersionsVersionBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (required)
+        :return: V1ModelVersionArchive
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'model_id', 'version']  # 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_update_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_update_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_update_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_update_model_version`")  # 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_update_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
+        if 'version' in params:
+            path_params['version'] = params['version']  # 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/{version}', 'PUT',
+            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_update_model_visibility(self, body: 'ModelIdVisibilityBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1UpdateModelVisibilityResponse':  # noqa: E501
-        """models_store_update_model_visibility  # noqa: E501
+        """UpdateModelVisibility updates the model version, mainly switch between public and private  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1964,7 +2206,7 @@ class ModelsStoreApi(object):
             return data
 
     def models_store_update_model_visibility_with_http_info(self, body: 'ModelIdVisibilityBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1UpdateModelVisibilityResponse':  # noqa: E501
-        """models_store_update_model_visibility  # noqa: E501
+        """UpdateModelVisibility updates the model version, mainly switch between public and private  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -304,6 +304,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instanc
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_endpoint_response import V1DeleteManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
@@ -756,6 +757,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1Vu
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
 from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
 from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody
+from lightning_sdk.lightning_cloud.openapi.models.version_default_body import VersionDefaultBody
 from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body import VersionUploadsBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
 from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody

lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py

@@ -41,6 +41,7 @@ class V1AWSDirectV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'ami_owner_id': 'str',
         'bucket_endpoint': 'str',
         'bucket_name': 'str',
         'cloud_formation_disabled': 'bool',
@@ -55,6 +56,7 @@ class V1AWSDirectV1(object):
     }
 
     attribute_map = {
+        'ami_owner_id': 'amiOwnerId',
         'bucket_endpoint': 'bucketEndpoint',
         'bucket_name': 'bucketName',
         'cloud_formation_disabled': 'cloudFormationDisabled',
@@ -68,8 +70,9 @@ class V1AWSDirectV1(object):
         'use_launch_templates': 'useLaunchTemplates'
     }
 
-    def __init__(self, bucket_endpoint: 'str' =None, bucket_name: 'str' =None, cloud_formation_disabled: 'bool' =None, efs_enabled: 'bool' =None, external_id: 'str' =None, primary_region: 'str' =None, regional_load_balancers: 'list[V1RegionalLoadBalancer]' =None, regions: 'list[str]' =None, role_arn: 'str' =None, source_cidr_ips: 'list[str]' =None, use_launch_templates: 'bool' =None):  # noqa: E501
+    def __init__(self, ami_owner_id: 'str' =None, bucket_endpoint: 'str' =None, bucket_name: 'str' =None, cloud_formation_disabled: 'bool' =None, efs_enabled: 'bool' =None, external_id: 'str' =None, primary_region: 'str' =None, regional_load_balancers: 'list[V1RegionalLoadBalancer]' =None, regions: 'list[str]' =None, role_arn: 'str' =None, source_cidr_ips: 'list[str]' =None, use_launch_templates: 'bool' =None):  # noqa: E501
         """V1AWSDirectV1 - a model defined in Swagger"""  # noqa: E501
+        self._ami_owner_id = None
         self._bucket_endpoint = None
         self._bucket_name = None
         self._cloud_formation_disabled = None
@@ -82,6 +85,8 @@ class V1AWSDirectV1(object):
         self._source_cidr_ips = None
         self._use_launch_templates = None
         self.discriminator = None
+        if ami_owner_id is not None:
+            self.ami_owner_id = ami_owner_id
         if bucket_endpoint is not None:
             self.bucket_endpoint = bucket_endpoint
         if bucket_name is not None:
@@ -105,6 +110,27 @@ class V1AWSDirectV1(object):
         if use_launch_templates is not None:
             self.use_launch_templates = use_launch_templates
 
+    @property
+    def ami_owner_id(self) -> 'str':
+        """Gets the ami_owner_id of this V1AWSDirectV1.  # noqa: E501
+
+
+        :return: The ami_owner_id of this V1AWSDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._ami_owner_id
+
+    @ami_owner_id.setter
+    def ami_owner_id(self, ami_owner_id: 'str'):
+        """Sets the ami_owner_id of this V1AWSDirectV1.
+
+
+        :param ami_owner_id: The ami_owner_id of this V1AWSDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._ami_owner_id = ami_owner_id
+
     @property
     def bucket_endpoint(self) -> 'str':
         """Gets the bucket_endpoint of this V1AWSDirectV1.  # noqa: E501