lightning-sdk 0.1.53__py3-none-any.whl → 0.1.55__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1096,6 +1096,7 @@ class ClusterServiceApi(object):
         :param async_req bool
         :param str cluster_id: (required)
         :param str id: (required)
+        :param str org_id:
         :return: V1DeleteClusterUsageRestrictionResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1118,12 +1119,13 @@ class ClusterServiceApi(object):
         :param async_req bool
         :param str cluster_id: (required)
         :param str id: (required)
+        :param str org_id:
         :return: V1DeleteClusterUsageRestrictionResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['cluster_id', 'id']  # noqa: E501
+        all_params = ['cluster_id', 'id', 'org_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1156,6 +1158,8 @@ class ClusterServiceApi(object):
             path_params['id'] = params['id']  # noqa: E501
 
         query_params = []
+        if 'org_id' in params:
+            query_params.append(('orgId', params['org_id']))  # noqa: E501
 
         header_params = {}
 
@@ -2311,6 +2315,7 @@ class ClusterServiceApi(object):
 
         :param async_req bool
         :param str cluster_id: (required)
+        :param str org_id:
         :return: V1ListClusterUsageRestrictionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -2332,12 +2337,13 @@ class ClusterServiceApi(object):
 
         :param async_req bool
         :param str cluster_id: (required)
+        :param str org_id:
         :return: V1ListClusterUsageRestrictionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['cluster_id']  # noqa: E501
+        all_params = ['cluster_id', 'org_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -2364,6 +2370,8 @@ class ClusterServiceApi(object):
             path_params['clusterId'] = params['cluster_id']  # noqa: E501
 
         query_params = []
+        if 'org_id' in params:
+            query_params.append(('orgId', params['org_id']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py

@@ -342,6 +342,115 @@ class LitRegistryServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest(self, project_id: 'str', lit_repo_name: 'str', full_hash_digest: 'str', **kwargs) -> 'V1LitRegistryArtifact':  # noqa: E501
+        """lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest  # 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_registry_repository_image_artifact_version_by_digest(project_id, lit_repo_name, full_hash_digest, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :param str full_hash_digest: (required)
+        :return: V1LitRegistryArtifact
+                 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_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, **kwargs)  # noqa: E501
+            return data
+
+    def lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(self, project_id: 'str', lit_repo_name: 'str', full_hash_digest: 'str', **kwargs) -> 'V1LitRegistryArtifact':  # noqa: E501
+        """lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest  # 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_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str lit_repo_name: (required)
+        :param str full_hash_digest: (required)
+        :return: V1LitRegistryArtifact
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'lit_repo_name', 'full_hash_digest']  # 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_registry_repository_image_artifact_version_by_digest" % 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_registry_repository_image_artifact_version_by_digest`")  # 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_registry_repository_image_artifact_version_by_digest`")  # noqa: E501
+        # verify the required parameter 'full_hash_digest' is set
+        if ('full_hash_digest' not in params or
+                params['full_hash_digest'] is None):
+            raise ValueError("Missing the required parameter `full_hash_digest` when calling `lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest`")  # 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
+        if 'full_hash_digest' in params:
+            path_params['fullHashDigest'] = params['full_hash_digest']  # 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}/artifacts/{fullHashDigest}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1LitRegistryArtifact',  # 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_repository(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1LitRepository':  # noqa: E501
         """lit_registry_service_get_lit_repository  # noqa: E501
 
@@ -443,6 +552,107 @@ class LitRegistryServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def lit_registry_service_list_lit_registry_repository_image_artifact_versions(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1ListLitRegistryRepositoryImageArtifactVersionsResponse':  # noqa: E501
+        """lit_registry_service_list_lit_registry_repository_image_artifact_versions  # 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_list_lit_registry_repository_image_artifact_versions(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: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
+                 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_list_lit_registry_repository_image_artifact_versions_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(project_id, lit_repo_name, **kwargs)  # noqa: E501
+            return data
+
+    def lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1ListLitRegistryRepositoryImageArtifactVersionsResponse':  # noqa: E501
+        """lit_registry_service_list_lit_registry_repository_image_artifact_versions  # 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_list_lit_registry_repository_image_artifact_versions_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: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
+                 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_list_lit_registry_repository_image_artifact_versions" % 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_list_lit_registry_repository_image_artifact_versions`")  # 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_list_lit_registry_repository_image_artifact_versions`")  # 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}/artifacts', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListLitRegistryRepositoryImageArtifactVersionsResponse',  # 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_update_lit_repository_metadata(self, body: 'LitregistryLitRepoNameBody', project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1UpdateLitRepositoryResponse':  # noqa: E501
         """lit_registry_service_update_lit_repository_metadata  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -534,6 +534,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instances
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_response import V1ListLightningworkEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
@@ -569,6 +570,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
+from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_project import V1LitRegistryProject
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_repository import V1LitRepository
 from lightning_sdk.lightning_cloud.openapi.models.v1_locked_resource import V1LockedResource

lightning_sdk/lightning_cloud/openapi/models/cluster_id_usagerestrictions_body.py

@@ -43,25 +43,30 @@ class ClusterIdUsagerestrictionsBody(object):
     swagger_types = {
         'instance_type': 'str',
         'max_instances': 'int',
+        'org_id': 'str',
         'restriction_type': 'str'
     }
 
     attribute_map = {
         'instance_type': 'instanceType',
         'max_instances': 'maxInstances',
+        'org_id': 'orgId',
         'restriction_type': 'restrictionType'
     }
 
-    def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, restriction_type: 'str' =None):  # noqa: E501
+    def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, org_id: 'str' =None, restriction_type: 'str' =None):  # noqa: E501
         """ClusterIdUsagerestrictionsBody - a model defined in Swagger"""  # noqa: E501
         self._instance_type = None
         self._max_instances = None
+        self._org_id = None
         self._restriction_type = None
         self.discriminator = None
         if instance_type is not None:
             self.instance_type = instance_type
         if max_instances is not None:
             self.max_instances = max_instances
+        if org_id is not None:
+            self.org_id = org_id
         if restriction_type is not None:
             self.restriction_type = restriction_type
 
@@ -107,6 +112,27 @@ class ClusterIdUsagerestrictionsBody(object):
 
         self._max_instances = max_instances
 
+    @property
+    def org_id(self) -> 'str':
+        """Gets the org_id of this ClusterIdUsagerestrictionsBody.  # noqa: E501
+
+
+        :return: The org_id of this ClusterIdUsagerestrictionsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._org_id
+
+    @org_id.setter
+    def org_id(self, org_id: 'str'):
+        """Sets the org_id of this ClusterIdUsagerestrictionsBody.
+
+
+        :param org_id: The org_id of this ClusterIdUsagerestrictionsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._org_id = org_id
+
     @property
     def restriction_type(self) -> 'str':
         """Gets the restriction_type of this ClusterIdUsagerestrictionsBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/usagerestrictions_id_body.py

@@ -43,25 +43,30 @@ class UsagerestrictionsIdBody(object):
     swagger_types = {
         'instance_type': 'str',
         'max_instances': 'int',
+        'org_id': 'str',
         'restriction_type': 'str'
     }
 
     attribute_map = {
         'instance_type': 'instanceType',
         'max_instances': 'maxInstances',
+        'org_id': 'orgId',
         'restriction_type': 'restrictionType'
     }
 
-    def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, restriction_type: 'str' =None):  # noqa: E501
+    def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, org_id: 'str' =None, restriction_type: 'str' =None):  # noqa: E501
         """UsagerestrictionsIdBody - a model defined in Swagger"""  # noqa: E501
         self._instance_type = None
         self._max_instances = None
+        self._org_id = None
         self._restriction_type = None
         self.discriminator = None
         if instance_type is not None:
             self.instance_type = instance_type
         if max_instances is not None:
             self.max_instances = max_instances
+        if org_id is not None:
+            self.org_id = org_id
         if restriction_type is not None:
             self.restriction_type = restriction_type
 
@@ -107,6 +112,27 @@ class UsagerestrictionsIdBody(object):
 
         self._max_instances = max_instances
 
+    @property
+    def org_id(self) -> 'str':
+        """Gets the org_id of this UsagerestrictionsIdBody.  # noqa: E501
+
+
+        :return: The org_id of this UsagerestrictionsIdBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._org_id
+
+    @org_id.setter
+    def org_id(self, org_id: 'str'):
+        """Sets the org_id of this UsagerestrictionsIdBody.
+
+
+        :param org_id: The org_id of this UsagerestrictionsIdBody.  # noqa: E501
+        :type: str
+        """
+
+        self._org_id = org_id
+
     @property
     def restriction_type(self) -> 'str':
         """Gets the restriction_type of this UsagerestrictionsIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py

@@ -37,9 +37,12 @@ class V1CloudProvider(object):
     """
     allowed enum values
     """
+    CLOUD_PROVIDER_UNSPECIFIED = "CLOUD_PROVIDER_UNSPECIFIED"
     AWS = "AWS"
     GCP = "GCP"
     VULTR = "VULTR"
+    LAMBDA_LABS = "LAMBDA_LABS"
+    SLURM = "SLURM"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py

@@ -66,6 +66,7 @@ class V1ClusterAccelerator(object):
         'local_disk_supported': 'bool',
         'max_available_quota': 'str',
         'non_spot': 'bool',
+        'provider': 'V1CloudProvider',
         'quota_checked_at': 'datetime',
         'quota_code': 'str',
         'quota_name': 'str',
@@ -109,6 +110,7 @@ class V1ClusterAccelerator(object):
         'local_disk_supported': 'localDiskSupported',
         'max_available_quota': 'maxAvailableQuota',
         'non_spot': 'nonSpot',
+        'provider': 'provider',
         'quota_checked_at': 'quotaCheckedAt',
         'quota_code': 'quotaCode',
         'quota_name': 'quotaName',
@@ -126,7 +128,7 @@ class V1ClusterAccelerator(object):
         'spot_quota_page_url': 'spotQuotaPageUrl'
     }
 
-    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, dws_only: 'bool' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None, spot_quota_code: 'str' =None, spot_quota_name: 'str' =None, spot_quota_page_url: 'str' =None):  # noqa: E501
+    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, dws_only: 'bool' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, provider: 'V1CloudProvider' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None, spot_quota_code: 'str' =None, spot_quota_name: 'str' =None, spot_quota_page_url: 'str' =None):  # noqa: E501
         """V1ClusterAccelerator - a model defined in Swagger"""  # noqa: E501
         self._accelerator_type = None
         self._allowed_resources = None
@@ -153,6 +155,7 @@ class V1ClusterAccelerator(object):
         self._local_disk_supported = None
         self._max_available_quota = None
         self._non_spot = None
+        self._provider = None
         self._quota_checked_at = None
         self._quota_code = None
         self._quota_name = None
@@ -219,6 +222,8 @@ class V1ClusterAccelerator(object):
             self.max_available_quota = max_available_quota
         if non_spot is not None:
             self.non_spot = non_spot
+        if provider is not None:
+            self.provider = provider
         if quota_checked_at is not None:
             self.quota_checked_at = quota_checked_at
         if quota_code is not None:
@@ -775,6 +780,27 @@ class V1ClusterAccelerator(object):
 
         self._non_spot = non_spot
 
+    @property
+    def provider(self) -> 'V1CloudProvider':
+        """Gets the provider of this V1ClusterAccelerator.  # noqa: E501
+
+
+        :return: The provider of this V1ClusterAccelerator.  # noqa: E501
+        :rtype: V1CloudProvider
+        """
+        return self._provider
+
+    @provider.setter
+    def provider(self, provider: 'V1CloudProvider'):
+        """Sets the provider of this V1ClusterAccelerator.
+
+
+        :param provider: The provider of this V1ClusterAccelerator.  # noqa: E501
+        :type: V1CloudProvider
+        """
+
+        self._provider = provider
+
     @property
     def quota_checked_at(self) -> 'datetime':
         """Gets the quota_checked_at of this V1ClusterAccelerator.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job.py

@@ -53,6 +53,7 @@ class V1Job(object):
         'project_id': 'str',
         'ready_at': 'datetime',
         'release_id': 'str',
+        'restart_count': 'int',
         'server_error': 'str',
         'spec': 'V1JobSpec',
         'started_at': 'datetime',
@@ -80,6 +81,7 @@ class V1Job(object):
         'project_id': 'projectId',
         'ready_at': 'readyAt',
         'release_id': 'releaseId',
+        'restart_count': 'restartCount',
         'server_error': 'serverError',
         'spec': 'spec',
         'started_at': 'startedAt',
@@ -94,7 +96,7 @@ class V1Job(object):
         'user_logs_started_at': 'userLogsStartedAt'
     }
 
-    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, deployment_id: 'str' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_healthy: 'bool' =None, message: 'str' =None, multi_machine_job_id: 'str' =None, name: 'str' =None, overprovisioned: 'bool' =None, project_id: 'str' =None, ready_at: 'datetime' =None, release_id: 'str' =None, restart_count: 'int' =None, server_error: 'str' =None, spec: 'V1JobSpec' =None, started_at: 'datetime' =None, state: 'str' =None, stopped_at: 'datetime' =None, timings: 'dict(str, V1JobTiming)' =None, total_cost: 'float' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None, user_logs_started_at: 'datetime' =None):  # noqa: E501
         """V1Job - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
         self._deployment_id = None
@@ -108,6 +110,7 @@ class V1Job(object):
         self._project_id = None
         self._ready_at = None
         self._release_id = None
+        self._restart_count = None
         self._server_error = None
         self._spec = None
         self._started_at = None
@@ -145,6 +148,8 @@ class V1Job(object):
             self.ready_at = ready_at
         if release_id is not None:
             self.release_id = release_id
+        if restart_count is not None:
+            self.restart_count = restart_count
         if server_error is not None:
             self.server_error = server_error
         if spec is not None:
@@ -424,6 +429,27 @@ class V1Job(object):
 
         self._release_id = release_id
 
+    @property
+    def restart_count(self) -> 'int':
+        """Gets the restart_count of this V1Job.  # noqa: E501
+
+
+        :return: The restart_count of this V1Job.  # noqa: E501
+        :rtype: int
+        """
+        return self._restart_count
+
+    @restart_count.setter
+    def restart_count(self, restart_count: 'int'):
+        """Sets the restart_count of this V1Job.
+
+
+        :param restart_count: The restart_count of this V1Job.  # noqa: E501
+        :type: int
+        """
+
+        self._restart_count = restart_count
+
     @property
     def server_error(self) -> 'str':
         """Gets the server_error of this V1Job.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -59,6 +59,7 @@ class V1JobSpec(object):
         'path_mappings': 'list[V1PathMapping]',
         'readiness_probe': 'V1JobHealthCheckConfig',
         'regions': 'list[str]',
+        'requested_run_duration_seconds': 'str',
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
@@ -85,6 +86,7 @@ class V1JobSpec(object):
         'path_mappings': 'pathMappings',
         'readiness_probe': 'readinessProbe',
         'regions': 'regions',
+        'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
@@ -92,7 +94,7 @@ class V1JobSpec(object):
         'volumes': 'volumes'
     }
 
-    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -112,6 +114,7 @@ class V1JobSpec(object):
         self._path_mappings = None
         self._readiness_probe = None
         self._regions = None
+        self._requested_run_duration_seconds = None
         self._resources = None
         self._restart_policy = None
         self._run_id = None
@@ -154,6 +157,8 @@ class V1JobSpec(object):
             self.readiness_probe = readiness_probe
         if regions is not None:
             self.regions = regions
+        if requested_run_duration_seconds is not None:
+            self.requested_run_duration_seconds = requested_run_duration_seconds
         if resources is not None:
             self.resources = resources
         if restart_policy is not None:
@@ -543,6 +548,27 @@ class V1JobSpec(object):
 
         self._regions = regions
 
+    @property
+    def requested_run_duration_seconds(self) -> 'str':
+        """Gets the requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+
+
+        :return: The requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._requested_run_duration_seconds
+
+    @requested_run_duration_seconds.setter
+    def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
+        """Sets the requested_run_duration_seconds of this V1JobSpec.
+
+
+        :param requested_run_duration_seconds: The requested_run_duration_seconds of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._requested_run_duration_seconds = requested_run_duration_seconds
+
     @property
     def resources(self) -> 'V1Resources':
         """Gets the resources of this V1JobSpec.  # noqa: E501