lightning-sdk 0.1.47__py3-none-any.whl → 0.1.49__py3-none-any.whl

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

@@ -290,7 +290,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_capacity_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_key_response import V1DeleteClusterEncryptionKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_response import V1DeleteContainerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
@@ -305,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
@@ -757,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

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py

@@ -42,7 +42,6 @@ class V1ClusterSpec(object):
     """
     swagger_types = {
         'auth_token': 'str',
-        'available_instance_types': 'list[str]',
         'aws_v1': 'V1AWSDirectV1',
         'cluster_type': 'V1ClusterType',
         'deletion_options': 'V1ClusterDeletionOptions',
@@ -63,7 +62,6 @@ class V1ClusterSpec(object):
 
     attribute_map = {
         'auth_token': 'authToken',
-        'available_instance_types': 'availableInstanceTypes',
         'aws_v1': 'awsV1',
         'cluster_type': 'clusterType',
         'deletion_options': 'deletionOptions',
@@ -82,10 +80,9 @@ class V1ClusterSpec(object):
         'vultr_v1': 'vultrV1'
     }
 
-    def __init__(self, auth_token: 'str' =None, available_instance_types: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
+    def __init__(self, auth_token: 'str' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
         """V1ClusterSpec - a model defined in Swagger"""  # noqa: E501
         self._auth_token = None
-        self._available_instance_types = None
         self._aws_v1 = None
         self._cluster_type = None
         self._deletion_options = None
@@ -105,8 +102,6 @@ class V1ClusterSpec(object):
         self.discriminator = None
         if auth_token is not None:
             self.auth_token = auth_token
-        if available_instance_types is not None:
-            self.available_instance_types = available_instance_types
         if aws_v1 is not None:
             self.aws_v1 = aws_v1
         if cluster_type is not None:
@@ -161,29 +156,6 @@ class V1ClusterSpec(object):
 
         self._auth_token = auth_token
 
-    @property
-    def available_instance_types(self) -> 'list[str]':
-        """Gets the available_instance_types of this V1ClusterSpec.  # noqa: E501
-
-        available_instance_types is a list of instance types that are available for the cluster. This is just a soft filter to prevent users from using instances that we haven't prepared for. If the list is empty, no filtering is done.  # noqa: E501
-
-        :return: The available_instance_types of this V1ClusterSpec.  # noqa: E501
-        :rtype: list[str]
-        """
-        return self._available_instance_types
-
-    @available_instance_types.setter
-    def available_instance_types(self, available_instance_types: 'list[str]'):
-        """Sets the available_instance_types of this V1ClusterSpec.
-
-        available_instance_types is a list of instance types that are available for the cluster. This is just a soft filter to prevent users from using instances that we haven't prepared for. If the list is empty, no filtering is done.  # noqa: E501
-
-        :param available_instance_types: The available_instance_types of this V1ClusterSpec.  # noqa: E501
-        :type: list[str]
-        """
-
-        self._available_instance_types = available_instance_types
-
     @property
     def aws_v1(self) -> 'V1AWSDirectV1':
         """Gets the aws_v1 of this V1ClusterSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/{v1_delete_container_response.py → v1_delete_lit_repository_response.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1DeleteContainerResponse(object):
+class V1DeleteLitRepositoryResponse(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -47,7 +47,7 @@ class V1DeleteContainerResponse(object):
     }
 
     def __init__(self):  # noqa: E501
-        """V1DeleteContainerResponse - a model defined in Swagger"""  # noqa: E501
+        """V1DeleteLitRepositoryResponse - a model defined in Swagger"""  # noqa: E501
         self.discriminator = None
 
     def to_dict(self) -> dict:
@@ -71,7 +71,7 @@ class V1DeleteContainerResponse(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1DeleteContainerResponse, dict):
+        if issubclass(V1DeleteLitRepositoryResponse, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -85,13 +85,13 @@ class V1DeleteContainerResponse(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1DeleteContainerResponse') -> bool:
+    def __eq__(self, other: 'V1DeleteLitRepositoryResponse') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1DeleteContainerResponse):
+        if not isinstance(other, V1DeleteLitRepositoryResponse):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1DeleteContainerResponse') -> bool:
+    def __ne__(self, other: 'V1DeleteLitRepositoryResponse') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py

@@ -47,6 +47,7 @@ class V1GoogleCloudDirectV1(object):
         'credentials_service_account_email': 'str',
         'primary_region': 'str',
         'project_id': 'str',
+        'project_sa_enabled': 'bool',
         'regions': 'list[str]',
         'service_account_email': 'str',
         'source_cidr_ips': 'list[str]'
@@ -59,12 +60,13 @@ class V1GoogleCloudDirectV1(object):
         'credentials_service_account_email': 'credentialsServiceAccountEmail',
         'primary_region': 'primaryRegion',
         'project_id': 'projectId',
+        'project_sa_enabled': 'projectSaEnabled',
         'regions': 'regions',
         'service_account_email': 'serviceAccountEmail',
         'source_cidr_ips': 'sourceCidrIps'
     }
 
-    def __init__(self, bucket_name: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
+    def __init__(self, bucket_name: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, project_sa_enabled: 'bool' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
         """V1GoogleCloudDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._bucket_name = None
         self._compute_project_role = None
@@ -72,6 +74,7 @@ class V1GoogleCloudDirectV1(object):
         self._credentials_service_account_email = None
         self._primary_region = None
         self._project_id = None
+        self._project_sa_enabled = None
         self._regions = None
         self._service_account_email = None
         self._source_cidr_ips = None
@@ -88,6 +91,8 @@ class V1GoogleCloudDirectV1(object):
             self.primary_region = primary_region
         if project_id is not None:
             self.project_id = project_id
+        if project_sa_enabled is not None:
+            self.project_sa_enabled = project_sa_enabled
         if regions is not None:
             self.regions = regions
         if service_account_email is not None:
@@ -223,6 +228,29 @@ class V1GoogleCloudDirectV1(object):
 
         self._project_id = project_id
 
+    @property
+    def project_sa_enabled(self) -> 'bool':
+        """Gets the project_sa_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+
+        If enabled, we will set up and use (for compute machines) service accounts with fine-graned permissions, to allow access objects under the teamspace prefix only. This flag assumed to be removed once this is dogfooded well ehough.  # noqa: E501
+
+        :return: The project_sa_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+        :rtype: bool
+        """
+        return self._project_sa_enabled
+
+    @project_sa_enabled.setter
+    def project_sa_enabled(self, project_sa_enabled: 'bool'):
+        """Sets the project_sa_enabled of this V1GoogleCloudDirectV1.
+
+        If enabled, we will set up and use (for compute machines) service accounts with fine-graned permissions, to allow access objects under the teamspace prefix only. This flag assumed to be removed once this is dogfooded well ehough.  # noqa: E501
+
+        :param project_sa_enabled: The project_sa_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+        :type: bool
+        """
+
+        self._project_sa_enabled = project_sa_enabled
+
     @property
     def regions(self) -> 'list[str]':
         """Gets the regions of this V1GoogleCloudDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py

@@ -41,19 +41,24 @@ class V1LambdaLabsDirectV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'credentials_secret_id': 'str'
+        'credentials_secret_id': 'str',
+        'parent_cluster_id': 'str'
     }
 
     attribute_map = {
-        'credentials_secret_id': 'credentialsSecretId'
+        'credentials_secret_id': 'credentialsSecretId',
+        'parent_cluster_id': 'parentClusterId'
     }
 
-    def __init__(self, credentials_secret_id: 'str' =None):  # noqa: E501
+    def __init__(self, credentials_secret_id: 'str' =None, parent_cluster_id: 'str' =None):  # noqa: E501
         """V1LambdaLabsDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._credentials_secret_id = None
+        self._parent_cluster_id = None
         self.discriminator = None
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
+        if parent_cluster_id is not None:
+            self.parent_cluster_id = parent_cluster_id
 
     @property
     def credentials_secret_id(self) -> 'str':
@@ -78,6 +83,29 @@ class V1LambdaLabsDirectV1(object):
 
         self._credentials_secret_id = credentials_secret_id
 
+    @property
+    def parent_cluster_id(self) -> 'str':
+        """Gets the parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+
+        Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that.  # noqa: E501
+
+        :return: The parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._parent_cluster_id
+
+    @parent_cluster_id.setter
+    def parent_cluster_id(self, parent_cluster_id: 'str'):
+        """Sets the parent_cluster_id of this V1LambdaLabsDirectV1.
+
+        Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that.  # noqa: E501
+
+        :param parent_cluster_id: The parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._parent_cluster_id = parent_cluster_id
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}