lightning-sdk 2026.1.22__py3-none-any.whl → 2026.1.30__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py

@@ -1436,6 +1436,119 @@ class JobsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def jobs_service_duplicate_deployment(self, body: 'JobsServiceDuplicateDeploymentBody', project_id: 'str', source_deployment_id: 'str', **kwargs) -> 'V1Deployment':  # noqa: E501
+        """jobs_service_duplicate_deployment  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.jobs_service_duplicate_deployment(body, project_id, source_deployment_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param JobsServiceDuplicateDeploymentBody body: (required)
+        :param str project_id: Required (required)
+        :param str source_deployment_id: Required (required)
+        :return: V1Deployment
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.jobs_service_duplicate_deployment_with_http_info(body, project_id, source_deployment_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.jobs_service_duplicate_deployment_with_http_info(body, project_id, source_deployment_id, **kwargs)  # noqa: E501
+            return data
+
+    def jobs_service_duplicate_deployment_with_http_info(self, body: 'JobsServiceDuplicateDeploymentBody', project_id: 'str', source_deployment_id: 'str', **kwargs) -> 'V1Deployment':  # noqa: E501
+        """jobs_service_duplicate_deployment  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.jobs_service_duplicate_deployment_with_http_info(body, project_id, source_deployment_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param JobsServiceDuplicateDeploymentBody body: (required)
+        :param str project_id: Required (required)
+        :param str source_deployment_id: Required (required)
+        :return: V1Deployment
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'source_deployment_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method jobs_service_duplicate_deployment" % 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 `jobs_service_duplicate_deployment`")  # 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 `jobs_service_duplicate_deployment`")  # noqa: E501
+        # verify the required parameter 'source_deployment_id' is set
+        if ('source_deployment_id' not in params or
+                params['source_deployment_id'] is None):
+            raise ValueError("Missing the required parameter `source_deployment_id` when calling `jobs_service_duplicate_deployment`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'source_deployment_id' in params:
+            path_params['sourceDeploymentId'] = params['source_deployment_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/deployments/{sourceDeploymentId}/duplicate', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1Deployment',  # 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 jobs_service_find_job(self, project_id: 'str', **kwargs) -> 'V1Job':  # noqa: E501
         """jobs_service_find_job  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py

@@ -1950,6 +1950,119 @@ class OrganizationsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def organizations_service_update_org_role(self, body: 'OrganizationsServiceUpdateOrgRoleBody', org_id: 'str', id: 'str', **kwargs) -> 'V1OrgRole':  # noqa: E501
+        """organizations_service_update_org_role  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.organizations_service_update_org_role(body, org_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param OrganizationsServiceUpdateOrgRoleBody body: (required)
+        :param str org_id: (required)
+        :param str id: (required)
+        :return: V1OrgRole
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.organizations_service_update_org_role_with_http_info(body, org_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.organizations_service_update_org_role_with_http_info(body, org_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def organizations_service_update_org_role_with_http_info(self, body: 'OrganizationsServiceUpdateOrgRoleBody', org_id: 'str', id: 'str', **kwargs) -> 'V1OrgRole':  # noqa: E501
+        """organizations_service_update_org_role  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.organizations_service_update_org_role_with_http_info(body, org_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param OrganizationsServiceUpdateOrgRoleBody body: (required)
+        :param str org_id: (required)
+        :param str id: (required)
+        :return: V1OrgRole
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'org_id', 'id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method organizations_service_update_org_role" % 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 `organizations_service_update_org_role`")  # noqa: E501
+        # verify the required parameter 'org_id' is set
+        if ('org_id' not in params or
+                params['org_id'] is None):
+            raise ValueError("Missing the required parameter `org_id` when calling `organizations_service_update_org_role`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `organizations_service_update_org_role`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'org_id' in params:
+            path_params['orgId'] = params['org_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/orgs/{orgId}/roles/{id}', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1OrgRole',  # 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 organizations_service_update_organization(self, body: 'OrganizationsServiceUpdateOrganizationBody', id: 'str', **kwargs) -> 'V1Organization':  # noqa: E501
         """Update organization  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py

@@ -872,6 +872,7 @@ class StorageServiceApi(object):
         :param str cluster_id:
         :param bool local_index: if true, fetch from the local index
         :param bool include_folder_index: if true, include the folder index in the response
+        :param bool only_direct_children: if true, only include direct children of the prefix
         :return: V1GetArtifactsPageResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -900,12 +901,13 @@ class StorageServiceApi(object):
         :param str cluster_id:
         :param bool local_index: if true, fetch from the local index
         :param bool include_folder_index: if true, include the folder index in the response
+        :param bool only_direct_children: if true, only include direct children of the prefix
         :return: V1GetArtifactsPageResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index']  # noqa: E501
+        all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index', 'only_direct_children']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -946,6 +948,8 @@ class StorageServiceApi(object):
             query_params.append(('localIndex', params['local_index']))  # noqa: E501
         if 'include_folder_index' in params:
             query_params.append(('includeFolderIndex', params['include_folder_index']))  # noqa: E501
+        if 'only_direct_children' in params:
+            query_params.append(('onlyDirectChildren', params['only_direct_children']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -81,7 +81,6 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_clo
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_version_publication_body import CloudSpaceServiceUpdateCloudSpaceVersionPublicationBody
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_visibility_body import CloudSpaceServiceUpdateCloudSpaceVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudy_service_update_user_cloudy_settings_body import CloudyServiceUpdateUserCloudySettingsBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_add_container_registry_body import ClusterServiceAddContainerRegistryBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_capacity_reservation_body import ClusterServiceCreateClusterCapacityReservationBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_proxy_body import ClusterServiceCreateClusterProxyBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_usage_restriction_body import ClusterServiceCreateClusterUsageRestrictionBody
@@ -89,16 +88,19 @@ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_machine
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_org_cluster_capacity_reservation_body import ClusterServiceCreateOrgClusterCapacityReservationBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project_cluster_body import ClusterServiceCreateProjectClusterBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_server_alert_body import ClusterServiceCreateServerAlertBody
+from lightning_sdk.lightning_cloud.openapi.models.cluster_service_get_cluster_capacity_reservation_body import ClusterServiceGetClusterCapacityReservationBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_interrupt_server_body import ClusterServiceInterruptServerBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_purchase_capacity_block_body import ClusterServicePurchaseCapacityBlockBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_refresh_container_registry_credentials_body import ClusterServiceRefreshContainerRegistryCredentialsBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_server_check_in_body import ClusterServiceServerCheckInBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_sleep_server_body import ClusterServiceSleepServerBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_body import ClusterServiceUpdateClusterBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_usage_restriction_body import ClusterServiceUpdateClusterUsageRestrictionBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_project_cluster_body import ClusterServiceUpdateProjectClusterBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_validate_container_registry_body import ClusterServiceValidateContainerRegistryBody
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_config_ecr import ContainerRegistryConfigECR
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_provider import ContainerRegistryProvider
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_create_container_registry_body import ContainerRegistryServiceCreateContainerRegistryBody
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_refresh_container_registry_credentials_body import ContainerRegistryServiceRefreshContainerRegistryCredentialsBody
 from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_service_create_data_connection_body import DataConnectionServiceCreateDataConnectionBody
@@ -126,6 +128,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment_body import JobsServiceCreateDeploymentBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_job_body import JobsServiceCreateJobBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_multi_machine_job_body import JobsServiceCreateMultiMachineJobBody
+from lightning_sdk.lightning_cloud.openapi.models.jobs_service_duplicate_deployment_body import JobsServiceDuplicateDeploymentBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_deployment_routing_telemetry_body import JobsServiceReportDeploymentRoutingTelemetryBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_job_system_metrics_body import JobsServiceReportJobSystemMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_logs_activity_body import JobsServiceReportLogsActivityBody
@@ -190,6 +193,7 @@ from lightning_sdk.lightning_cloud.openapi.models.organizations_service_auto_joi
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_body import OrganizationsServiceCreateOrgMembershipBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_role_binding_body import OrganizationsServiceCreateOrgMembershipRoleBindingBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_role_body import OrganizationsServiceCreateOrgRoleBody
+from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_org_role_body import OrganizationsServiceUpdateOrgRoleBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_body import OrganizationsServiceUpdateOrganizationBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_credits_auto_replenish_body import OrganizationsServiceUpdateOrganizationCreditsAutoReplenishBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_validate_auto_join_domain_body import OrganizationsServiceValidateAutoJoinDomainBody
@@ -251,7 +255,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import
 from lightning_sdk.lightning_cloud.openapi.models.v1_abort_storage_transfer_response import V1AbortStorageTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_accelerator_quota_info import V1AcceleratorQuotaInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_ack_user_storage_violation_response import V1AckUserStorageViolationResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_add_container_registry_response import V1AddContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_add_job_timing_response import V1AddJobTimingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_affiliate_link import V1AffiliateLink
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_complete_part_response import V1AgentCompletePartResponse
@@ -370,9 +373,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_rea
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_container_metrics import V1ContainerMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry import V1ContainerRegistry
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_info import V1ContainerRegistryInfo
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_integration import V1ContainerRegistryIntegration
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_status import V1ContainerRegistryStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_config import V1ContainerRegistryConfig
+from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_scopes import V1ContainerRegistryScopes
 from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
 from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
 from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -391,6 +393,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_k
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_response import V1CreateClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_request import V1CreateClusterRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response import V1CreateClusterResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_container_registry_response import V1CreateContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_git_credentials_request import V1CreateGitCredentialsRequest
@@ -477,6 +480,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_multi_machine_job_response import V1DeleteMultiMachineJobResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_cluster_capacity_reservation_response import V1DeleteOrgClusterCapacityReservationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_response import V1DeleteOrgMembershipResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
@@ -526,6 +530,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_paramet
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_type import V1DeploymentTemplateParameterType
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_summary import V1DeploymentTemplateSummary
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_type import V1DeploymentTemplateType
+from lightning_sdk.lightning_cloud.openapi.models.v1_describe_org_cluster_capacity_reservation_response import V1DescribeOrgClusterCapacityReservationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_download_job_logs_response import V1DownloadJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_download_lightningapp_instance_logs_response import V1DownloadLightningappInstanceLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive import V1Drive
@@ -536,9 +541,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config import V1ECRRegistryConfig
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config_input import V1ECRRegistryConfigInput
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_details import V1ECRRegistryDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
@@ -594,6 +596,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_required_ba
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_transfer_estimate_response import V1GetCloudSpaceTransferEstimateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_capacity_reservation_response import V1GetClusterCapacityReservationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_health_response import V1GetClusterHealthResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
@@ -759,7 +762,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_response import V1ListFilesystemMMTsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_metrics_response import V1ListFilesystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
@@ -876,6 +879,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics import V1Metrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_stream import V1MetricsStream
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1MetricsTags
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
+from lightning_sdk.lightning_cloud.openapi.models.v1_mithril_direct_v1 import V1MithrilDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_metrics import V1ModelMetrics
@@ -1084,6 +1088,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_visibili
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_container_registry_response import V1UpdateContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_like_response import V1UpdateConversationLikeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_message_content_response import V1UpdateConversationMessageContentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_message_like_response import V1UpdateConversationMessageLikeResponse
@@ -1126,7 +1131,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_requested_flow_compute
 from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_response import V1UserSLURMJobActionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_validate_container_registry_response import V1ValidateContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse

lightning_sdk/lightning_cloud/openapi/models/{cluster_service_refresh_container_registry_credentials_body.py → cluster_service_get_cluster_capacity_reservation_body.py}

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

lightning_sdk/lightning_cloud/openapi/models/{v1_container_registry_integration.py → container_registry_config_ecr.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1ContainerRegistryIntegration(object):
+class ContainerRegistryConfigECR(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -41,40 +41,66 @@ class V1ContainerRegistryIntegration(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'registries': 'list[V1ContainerRegistry]'
+        'last_refresh_at': 'datetime',
+        'token_refresh_seconds': 'str'
     }
 
     attribute_map = {
-        'registries': 'registries'
+        'last_refresh_at': 'lastRefreshAt',
+        'token_refresh_seconds': 'tokenRefreshSeconds'
     }
 
-    def __init__(self, registries: 'list[V1ContainerRegistry]' =None):  # noqa: E501
-        """V1ContainerRegistryIntegration - a model defined in Swagger"""  # noqa: E501
-        self._registries = None
+    def __init__(self, last_refresh_at: 'datetime' =None, token_refresh_seconds: 'str' =None):  # noqa: E501
+        """ContainerRegistryConfigECR - a model defined in Swagger"""  # noqa: E501
+        self._last_refresh_at = None
+        self._token_refresh_seconds = None
         self.discriminator = None
-        if registries is not None:
-            self.registries = registries
+        if last_refresh_at is not None:
+            self.last_refresh_at = last_refresh_at
+        if token_refresh_seconds is not None:
+            self.token_refresh_seconds = token_refresh_seconds
 
     @property
-    def registries(self) -> 'list[V1ContainerRegistry]':
-        """Gets the registries of this V1ContainerRegistryIntegration.  # noqa: E501
+    def last_refresh_at(self) -> 'datetime':
+        """Gets the last_refresh_at of this ContainerRegistryConfigECR.  # noqa: E501
 
 
-        :return: The registries of this V1ContainerRegistryIntegration.  # noqa: E501
-        :rtype: list[V1ContainerRegistry]
+        :return: The last_refresh_at of this ContainerRegistryConfigECR.  # noqa: E501
+        :rtype: datetime
         """
-        return self._registries
+        return self._last_refresh_at
 
-    @registries.setter
-    def registries(self, registries: 'list[V1ContainerRegistry]'):
-        """Sets the registries of this V1ContainerRegistryIntegration.
+    @last_refresh_at.setter
+    def last_refresh_at(self, last_refresh_at: 'datetime'):
+        """Sets the last_refresh_at of this ContainerRegistryConfigECR.
 
 
-        :param registries: The registries of this V1ContainerRegistryIntegration.  # noqa: E501
-        :type: list[V1ContainerRegistry]
+        :param last_refresh_at: The last_refresh_at of this ContainerRegistryConfigECR.  # noqa: E501
+        :type: datetime
         """
 
-        self._registries = registries
+        self._last_refresh_at = last_refresh_at
+
+    @property
+    def token_refresh_seconds(self) -> 'str':
+        """Gets the token_refresh_seconds of this ContainerRegistryConfigECR.  # noqa: E501
+
+
+        :return: The token_refresh_seconds of this ContainerRegistryConfigECR.  # noqa: E501
+        :rtype: str
+        """
+        return self._token_refresh_seconds
+
+    @token_refresh_seconds.setter
+    def token_refresh_seconds(self, token_refresh_seconds: 'str'):
+        """Sets the token_refresh_seconds of this ContainerRegistryConfigECR.
+
+
+        :param token_refresh_seconds: The token_refresh_seconds of this ContainerRegistryConfigECR.  # noqa: E501
+        :type: str
+        """
+
+        self._token_refresh_seconds = token_refresh_seconds
 
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
@@ -97,7 +123,7 @@ class V1ContainerRegistryIntegration(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1ContainerRegistryIntegration, dict):
+        if issubclass(ContainerRegistryConfigECR, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -111,13 +137,13 @@ class V1ContainerRegistryIntegration(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1ContainerRegistryIntegration') -> bool:
+    def __eq__(self, other: 'ContainerRegistryConfigECR') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1ContainerRegistryIntegration):
+        if not isinstance(other, ContainerRegistryConfigECR):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1ContainerRegistryIntegration') -> bool:
+    def __ne__(self, other: 'ContainerRegistryConfigECR') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other

lightning_sdk/lightning_cloud/openapi/models/{v1_container_registry_status.py → container_registry_provider.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1ContainerRegistryStatus(object):
+class ContainerRegistryProvider(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -37,10 +37,14 @@ class V1ContainerRegistryStatus(object):
     """
     allowed enum values
     """
-    UNSPECIFIED = "CONTAINER_REGISTRY_STATUS_UNSPECIFIED"
-    PENDING = "CONTAINER_REGISTRY_STATUS_PENDING"
-    VALIDATED = "CONTAINER_REGISTRY_STATUS_VALIDATED"
-    ERROR = "CONTAINER_REGISTRY_STATUS_ERROR"
+    UNSPECIFIED = "Unspecified"
+    ECR = "ECR"
+    LITCR = "LitCR"
+    GHCR = "GHCR"
+    DOCKERHUB = "DockerHub"
+    GCR = "GCR"
+    GAR = "GAR"
+    ACR = "ACR"
     """
     Attributes:
       swagger_types (dict): The key is attribute name
@@ -55,7 +59,7 @@ class V1ContainerRegistryStatus(object):
     }
 
     def __init__(self):  # noqa: E501
-        """V1ContainerRegistryStatus - a model defined in Swagger"""  # noqa: E501
+        """ContainerRegistryProvider - a model defined in Swagger"""  # noqa: E501
         self.discriminator = None
 
     def to_dict(self) -> dict:
@@ -79,7 +83,7 @@ class V1ContainerRegistryStatus(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1ContainerRegistryStatus, dict):
+        if issubclass(ContainerRegistryProvider, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -93,13 +97,13 @@ class V1ContainerRegistryStatus(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1ContainerRegistryStatus') -> bool:
+    def __eq__(self, other: 'ContainerRegistryProvider') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1ContainerRegistryStatus):
+        if not isinstance(other, ContainerRegistryProvider):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1ContainerRegistryStatus') -> bool:
+    def __ne__(self, other: 'ContainerRegistryProvider') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other