lightning-sdk 2026.1.27__py3-none-any.whl → 2026.2.3__py3-none-any.whl

lightning_sdk/__version__.py

@@ -1,3 +1,3 @@
 """Version information for lightning_sdk."""
 
-__version__ = "2026.01.27"
+__version__ = "2026.02.03"

lightning_sdk/api/k8s_api.py

@@ -86,8 +86,14 @@ class K8sClusterApi:
                     timestamp = datetime.fromisoformat(timestamp.replace("Z", "+00:00"))
                 hour = timestamp.replace(minute=0, second=0, microsecond=0)
 
+                # allocated GPUs cannot exceed total number of gpus
+                num_gpus = entry["num_gpus"]
+                allocated_gpus = entry["num_allocated_gpus"]
+                if allocated_gpus > num_gpus:
+                    allocated_gpus = num_gpus
+
                 # Store allocated GPUs for averaging
-                hourly_data[hour]["allocated_gpus"].append(entry["num_allocated_gpus"])
+                hourly_data[hour]["allocated_gpus"].append(allocated_gpus)
 
                 # Keep first entry for each hour (for other fields)
                 if hourly_data[hour]["first_entry"] is None:

lightning_sdk/api/studio_api.py

@@ -13,12 +13,15 @@ import requests
 from tqdm import tqdm
 
 from lightning_sdk.api.utils import (
+    _MAX_SIZE_MULTI_PART_CHUNK,
     _authenticate_and_get_token,
     _create_app,
     _DummyBody,
     _DummyResponse,
+    _FileUploader,
     _machine_to_compute_name,
     _sanitize_studio_remote_path,
+    _SinglePartFileUploader,
 )
 from lightning_sdk.api.utils import (
     _get_cloud_url as _cloud_url,
@@ -722,33 +725,36 @@ class StudioApi:
         remote_path: str,
         progress_bar: bool,
     ) -> None:
-        """Uploads file to given remote path on the studio."""
-        token = _authenticate_and_get_token(self._client)
-
-        query_params = {"token": token}
-        client_host = self._client.api_client.configuration.host
-        url = f"{client_host}/v1/projects/{teamspace_id}/artifacts/cloudspaces/{studio_id}/blobs/{remote_path}"
-
-        filesize = os.path.getsize(file_path)
-        with open(file_path, "rb") as f:
-            if progress_bar:
-                filesize = os.path.getsize(file_path)
-                with tqdm.wrapattr(
-                    f,
-                    "read",
-                    desc=f"Uploading {os.path.split(file_path)[1]}",
-                    total=filesize,
-                    unit="B",
-                    unit_scale=True,
-                    unit_divisor=1000,
-                ) as wrapped_file:
-                    r = requests.put(url, data=wrapped_file, params=query_params, timeout=30)
-            else:
-                r = requests.put(url, data=f, params=query_params, timeout=30)
+        """Uploads file to given remote path in the studio.
 
-        if r.status_code == 200:
-            return
-        raise RuntimeError(f"Failed to upload file '{file_path}' to the Studio. Status code: {r.status_code}")
+        Uses single-part upload for files <= 5MB, multipart upload for larger files.
+        """
+        file_size = os.path.getsize(file_path)
+        multipart_threshold = int(os.environ.get("LIGHTNING_MULTIPART_THRESHOLD", _MAX_SIZE_MULTI_PART_CHUNK))
+
+        if file_size <= multipart_threshold:
+            token = _authenticate_and_get_token(self._client)
+
+            query_params = {"token": token}
+            client_host = self._client.api_client.configuration.host
+            url = f"{client_host}/v1/projects/{teamspace_id}/artifacts/cloudspaces/{studio_id}/blobs/{remote_path}"
+
+            _SinglePartFileUploader(
+                client=self._client,
+                file_path=file_path,
+                url=url,
+                query_params=query_params,
+                progress_bar=progress_bar,
+            )()
+        else:
+            _FileUploader(
+                client=self._client,
+                teamspace_id=teamspace_id,
+                cloud_account=cloud_account,
+                file_path=file_path,
+                remote_path=_sanitize_studio_remote_path(remote_path, studio_id),
+                progress_bar=progress_bar,
+            )()
 
     def download_file(
         self,

lightning_sdk/api/teamspace_api.py

@@ -10,11 +10,15 @@ import requests
 from tqdm.auto import tqdm
 
 from lightning_sdk.api.utils import (
+    _MAX_SIZE_MULTI_PART_CHUNK,
     _authenticate_and_get_token,
     _download_model_files,
     _DummyBody,
+    _FileUploader,
     _get_model_version,
     _ModelFileUploader,
+    _resolve_teamspace_remote_path,
+    _SinglePartFileUploader,
 )
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
@@ -451,34 +455,40 @@ class TeamspaceApi:
         file_path: str,
         remote_path: str,
         progress_bar: bool,
+        headers: Optional[Dict[str, str]] = None,
     ) -> None:
-        """Uploads file to given remote path in the Teamspace drive /Uploads/."""
-        token = _authenticate_and_get_token(self._client)
+        """Uploads file to given remote path in the Teamspace drive.
 
-        query_params = {"token": token, "clusterId": cloud_account}
-        client_host = self._client.api_client.configuration.host
-        url = f"{client_host}/v1/projects/{teamspace_id}/artifacts/uploads/blobs/{remote_path}"
-
-        filesize = os.path.getsize(file_path)
-        with open(file_path, "rb") as f:
-            if progress_bar:
-                filesize = os.path.getsize(file_path)
-                with tqdm.wrapattr(
-                    f,
-                    "read",
-                    desc=f"Uploading {os.path.split(file_path)[1]}",
-                    total=filesize,
-                    unit="B",
-                    unit_scale=True,
-                    unit_divisor=1000,
-                ) as wrapped_file:
-                    r = requests.put(url, data=wrapped_file, params=query_params, timeout=30)
-            else:
-                r = requests.put(url, data=f, params=query_params, timeout=30)
-
-        if r.status_code == 200:
-            return
-        raise RuntimeError(f"Failed to upload file '{file_path}' to the Teamspace drive. Status code: {r.status_code}")
+        Uses single-part upload for files <= 5MB, multipart upload for larger files.
+        """
+        file_size = os.path.getsize(file_path)
+
+        multipart_threshold = int(os.environ.get("LIGHTNING_MULTIPART_THRESHOLD", _MAX_SIZE_MULTI_PART_CHUNK))
+
+        if file_size <= multipart_threshold:
+            token = _authenticate_and_get_token(self._client)
+
+            query_params = {"token": token, "clusterId": cloud_account}
+            client_host = self._client.api_client.configuration.host
+            url = f"{client_host}/v1/projects/{teamspace_id}/artifacts/blobs/{remote_path}"
+
+            _SinglePartFileUploader(
+                client=self._client,
+                file_path=file_path,
+                url=url,
+                query_params=query_params,
+                progress_bar=progress_bar,
+                headers=headers,
+            )()
+        else:
+            _FileUploader(
+                client=self._client,
+                teamspace_id=teamspace_id,
+                cloud_account=cloud_account,
+                file_path=file_path,
+                remote_path=_resolve_teamspace_remote_path(remote_path),
+                progress_bar=progress_bar,
+            )()
 
     def download_file(
         self,

lightning_sdk/api/utils.py

@@ -59,6 +59,66 @@ _MAX_BATCH_SIZE = 50
 _MAX_WORKERS = 10
 
 
+class _SinglePartFileUploader:
+    """A class handling upload files to studio and teamspace drive with new endpoint."""
+
+    def __init__(
+        self,
+        client: LightningClient,
+        file_path: str,
+        url: str,
+        query_params: Dict[str, str],
+        progress_bar: bool,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> None:
+        self.client = client
+        self.local_path = file_path
+        self.url = url
+        self.query_params = query_params
+        self.headers = headers
+        self.filesize = os.path.getsize(file_path)
+
+        if progress_bar:
+            self.progress_bar = tqdm(
+                desc=f"Uploading {os.path.split(file_path)[1]}",
+                total=self.filesize,
+                unit="B",
+                unit_scale=True,
+                unit_divisor=1000,
+                position=-1,
+                mininterval=1,
+            )
+        else:
+            self.progress_bar = None
+
+    def __call__(self) -> None:
+        self._upload_with_retry()
+
+    @backoff.on_exception(
+        backoff.expo, (requests.exceptions.HTTPError, requests.exceptions.RequestException), max_tries=10
+    )
+    def _upload_with_retry(self) -> None:
+        with open(self.local_path, "rb") as f:
+            if self.progress_bar is not None:
+                with tqdm.wrapattr(
+                    f,
+                    "read",
+                    desc=f"Uploading {os.path.split(self.local_path)[1]}",
+                    total=self.filesize,
+                    unit="B",
+                    unit_scale=True,
+                    unit_divisor=1000,
+                ) as wrapped_file:
+                    r = requests.put(
+                        self.url, data=wrapped_file, params=self.query_params, timeout=30, headers=self.headers
+                    )
+            else:
+                r = requests.put(self.url, data=f, params=self.query_params, timeout=30, headers=self.headers)
+
+        if r.status_code != 200:
+            raise RuntimeError(f"Failed to upload file '{self.local_path}'. Status code: {r.status_code}")
+
+
 class _FileUploader:
     """A class handling the upload to studios.
 
@@ -358,7 +418,7 @@ def _sanitize_studio_remote_path(path: str, studio_id: str) -> str:
 
 
 def _resolve_teamspace_remote_path(path: str) -> str:
-    return f"/Uploads/{path.replace('/teamspace/uploads/', '')}"
+    return f"{path.replace('/teamspace/', '')}"
 
 
 _DOWNLOAD_REQUEST_CHUNK_SIZE = 10 * _BYTES_PER_MB

lightning_sdk/cli/cp/teamspace_uploads.py

@@ -4,7 +4,7 @@ from pathlib import Path
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.cli.utils.filesystem import parse_teamspace_uploads_path, resolve_teamspace
+from lightning_sdk.cli.utils.filesystem import parse_teamspace_uploads_path, path_join, resolve_teamspace
 
 
 def cp_upload(
@@ -20,6 +20,8 @@ def cp_upload(
 
     teamspace_path_result = parse_teamspace_uploads_path(teamspace_path)
 
+    teamspace_path_result["destination"] = path_join("Uploads", teamspace_path_result["destination"])
+
     selected_teamspace = resolve_teamspace(teamspace_path_result["teamspace"], teamspace_path_result["owner"])
     console.print(f"Uploading to {selected_teamspace.owner.name}/{selected_teamspace.name}")
 
@@ -33,7 +35,7 @@ def cp_upload(
         if teamspace_path.endswith(("/", "\\")):
             # if destination ends with / or \, treat it as a directory
             file_name = os.path.basename(local_file_path)
-            teamspace_path_result["destination"] = os.path.join(teamspace_path_result["destination"], file_name)
+            teamspace_path_result["destination"] = path_join(teamspace_path_result["destination"], file_name)
         selected_teamspace.upload_file(
             local_file_path, teamspace_path_result["destination"], cloud_account=cloud_account
         )

lightning_sdk/cli/utils/filesystem.py

@@ -1,4 +1,5 @@
-from typing import Optional, TypedDict
+import os
+from typing import Any, Optional, TypedDict
 
 from lightning_sdk.cli.utils.owner_selection import OwnerMenu
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
@@ -14,6 +15,10 @@ class PathResult(TypedDict):
     destination: Optional[str]
 
 
+def path_join(*args: Any) -> str:
+    return os.path.join(*args).replace("\\", "/")
+
+
 def parse_studio_path(studio_path: str) -> PathResult:
     path_string = studio_path.removeprefix("lit://")
     if not path_string:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -141,6 +141,7 @@ 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_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
@@ -648,6 +649,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
@@ -1139,6 +1141,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

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -2470,6 +2470,127 @@ class ClusterServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cluster_service_get_cluster_capacity_reservation(self, body: 'ClusterServiceGetClusterCapacityReservationBody', project_id: 'str', cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterCapacityReservationResponse':  # noqa: E501
+        """capacity reservation  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_get_cluster_capacity_reservation(body, project_id, cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ClusterServiceGetClusterCapacityReservationBody body: (required)
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str id: capacity reservation id (required)
+        :return: V1GetClusterCapacityReservationResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cluster_service_get_cluster_capacity_reservation_with_http_info(body, project_id, cluster_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cluster_service_get_cluster_capacity_reservation_with_http_info(body, project_id, cluster_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def cluster_service_get_cluster_capacity_reservation_with_http_info(self, body: 'ClusterServiceGetClusterCapacityReservationBody', project_id: 'str', cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterCapacityReservationResponse':  # noqa: E501
+        """capacity reservation  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_get_cluster_capacity_reservation_with_http_info(body, project_id, cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ClusterServiceGetClusterCapacityReservationBody body: (required)
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str id: capacity reservation id (required)
+        :return: V1GetClusterCapacityReservationResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'cluster_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 cluster_service_get_cluster_capacity_reservation" % 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 `cluster_service_get_cluster_capacity_reservation`")  # 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 `cluster_service_get_cluster_capacity_reservation`")  # noqa: E501
+        # verify the required parameter 'cluster_id' is set
+        if ('cluster_id' not in params or
+                params['cluster_id'] is None):
+            raise ValueError("Missing the required parameter `cluster_id` when calling `cluster_service_get_cluster_capacity_reservation`")  # 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 `cluster_service_get_cluster_capacity_reservation`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cluster_id' in params:
+            path_params['clusterId'] = params['cluster_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/projects/{projectId}/clusters/{clusterId}/capacity-reservations/{id}', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetClusterCapacityReservationResponse',  # 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 cluster_service_get_cluster_credentials(self, **kwargs) -> 'V1GetClusterCredentialsResponse':  # noqa: E501
         """cluster_service_get_cluster_credentials  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/container_registry_service_api.py

@@ -454,3 +454,126 @@ class ContainerRegistryServiceApi(object):
             _preload_content=params.get('_preload_content', True),
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
+
+    def container_registry_service_update_container_registry(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1UpdateContainerRegistryResponse':  # noqa: E501
+        """container_registry_service_update_container_registry  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.container_registry_service_update_container_registry(cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str cluster_id: (required)
+        :param str id: (required)
+        :param str org_id:
+        :param str url:
+        :param str provider:
+        :param list[str] scopes_namespaces: for k8s org clusters
+        :param list[str] scopes_project_ids: for byoc cloud accounts
+        :return: V1UpdateContainerRegistryResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.container_registry_service_update_container_registry_with_http_info(cluster_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.container_registry_service_update_container_registry_with_http_info(cluster_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def container_registry_service_update_container_registry_with_http_info(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1UpdateContainerRegistryResponse':  # noqa: E501
+        """container_registry_service_update_container_registry  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.container_registry_service_update_container_registry_with_http_info(cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str cluster_id: (required)
+        :param str id: (required)
+        :param str org_id:
+        :param str url:
+        :param str provider:
+        :param list[str] scopes_namespaces: for k8s org clusters
+        :param list[str] scopes_project_ids: for byoc cloud accounts
+        :return: V1UpdateContainerRegistryResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['cluster_id', 'id', 'org_id', 'url', 'provider', 'scopes_namespaces', 'scopes_project_ids']  # 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 container_registry_service_update_container_registry" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'cluster_id' is set
+        if ('cluster_id' not in params or
+                params['cluster_id'] is None):
+            raise ValueError("Missing the required parameter `cluster_id` when calling `container_registry_service_update_container_registry`")  # 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 `container_registry_service_update_container_registry`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cluster_id' in params:
+            path_params['clusterId'] = params['cluster_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+        if 'org_id' in params:
+            query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'url' in params:
+            query_params.append(('url', params['url']))  # noqa: E501
+        if 'provider' in params:
+            query_params.append(('provider', params['provider']))  # noqa: E501
+        if 'scopes_namespaces' in params:
+            query_params.append(('scopes.namespaces', params['scopes_namespaces']))  # noqa: E501
+            collection_formats['scopes.namespaces'] = 'multi'  # noqa: E501
+        if 'scopes_project_ids' in params:
+            query_params.append(('scopes.projectIds', params['scopes_project_ids']))  # noqa: E501
+            collection_formats['scopes.projectIds'] = 'multi'  # noqa: E501
+
+        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/core/clusters/{clusterId}/container-registries/{id}', 'PATCH',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateContainerRegistryResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -88,6 +88,7 @@ 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_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
@@ -595,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
@@ -1086,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