lightning-sdk 2025.12.9__py3-none-any.whl → 2025.12.17__py3-none-any.whl

lightning_sdk/__version__.py

@@ -1,3 +1,3 @@
 """Version information for lightning_sdk."""
 
-__version__ = "2025.12.09"
+__version__ = "2025.12.17"

lightning_sdk/api/job_api.py

@@ -30,6 +30,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1LightningworkState,
     V1ListLightningworkResponse,
     V1UserRequestedComputeConfig,
+    V1Volume,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 from lightning_sdk.machine import Machine
@@ -253,7 +254,16 @@ class JobApiV2:
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> V1Job:
+        if scratch_disks is not None:
+            sanitized_scratch_disks = {}
+            for k, v in scratch_disks.items():
+                sanitized_k = k if k.startswith("/teamspace/scratch/") else f"/teamspace/scratch/{k}"
+                sanitized_scratch_disks[sanitized_k] = v
+        else:
+            sanitized_scratch_disks = None
+
         body = self._create_job_body(
             name=name,
             command=command,
@@ -271,6 +281,7 @@ class JobApiV2:
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
             reuse_snapshot=reuse_snapshot,
+            scratch_disks=sanitized_scratch_disks,
         )
 
         job: V1Job = self._client.jobs_service_create_job(project_id=teamspace_id, body=body)
@@ -295,6 +306,7 @@ class JobApiV2:
         reuse_snapshot: bool,
         max_runtime: Optional[int] = None,
         machine_image_version: Optional[str] = None,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> JobsServiceCreateJobBody:
         env_vars = []
         if env is not None:
@@ -316,6 +328,9 @@ class JobApiV2:
         if max_runtime:
             optional_spec_kwargs["requested_run_duration_seconds"] = str(max_runtime)
 
+        # don't do default dicts, as they'll be mutable. Create a fresh one here
+        scratch_disks = scratch_disks or {}
+
         spec = V1JobSpec(
             cloudspace_id=studio_id or "",
             cluster_id=cloud_account or "",
@@ -330,6 +345,7 @@ class JobApiV2:
             image_secret_ref=image_credentials or "",
             path_mappings=path_mappings_list,
             machine_image_version=machine_image_version,
+            volumes=[V1Volume(path=k, size_gb=v, ephemeral=True) for k, v in scratch_disks.items()],
             **optional_spec_kwargs,
         )
         return JobsServiceCreateJobBody(name=name, spec=spec)

lightning_sdk/api/org_api.py

@@ -1,4 +1,5 @@
 from lightning_sdk.lightning_cloud.openapi import (
+    V1CreateProjectRequest,
     V1Organization,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
@@ -20,3 +21,9 @@ class OrgApi:
     def _get_org_by_id(self, org_id: str) -> V1Organization:
         """Gets the organization from the given ID."""
         return self._client.organizations_service_get_organization(id=org_id)
+
+    def create_teamspace(self, name: str, organization_id: str) -> None:
+        """Creates a new teamspace."""
+        self._client.projects_service_create_project(
+            body=V1CreateProjectRequest(name=name, organization_id=organization_id, display_name=name)
+        )

lightning_sdk/api/studio_api.py

@@ -962,9 +962,9 @@ class StudioApi:
             interruptible=interruptible,
         )
 
-    def start_new_port(self, teamspace_id: str, studio_id: str, name: str, port: int, auto_start: bool = False) -> str:
+    def add_port(self, teamspace_id: str, studio_id: str, name: str, port: int, auto_start: bool = False) -> V1Endpoint:
         """Starts a new port to the given Studio."""
-        endpoint = self._client.endpoint_service_create_endpoint(
+        return self._client.endpoint_service_create_endpoint(
             project_id=teamspace_id,
             body=EndpointServiceCreateEndpointBody(
                 name=name,
@@ -974,7 +974,32 @@ class StudioApi:
                 ),
             ),
         )
-        return endpoint.urls[0]
+
+    def get_port_url(
+        self, teamspace_id: str, studio_id: str, port: Optional[int] = None, name: Optional[str] = None
+    ) -> str:
+        if port is None and name is None:
+            raise ValueError("Either 'port' or 'name' must be provided")
+
+        endpoints = self.list_ports(teamspace_id=teamspace_id, studio_id=studio_id)
+
+        for endpoint in endpoints:
+            if port is not None and port in endpoint.ports:
+                idx = endpoint.ports.index(port)
+                return endpoint.urls[idx]
+
+            if name is not None and endpoint.name == name:
+                return endpoint.urls[0] if endpoint.urls else None
+
+        identifier = f"port {port}" if port else f"name '{name}'"
+        raise ValueError(f"Endpoint with {identifier} not found")
+
+    def list_ports(self, teamspace_id: str, studio_id: str) -> List[V1Endpoint]:
+        """List ports that are exposed in the Studio."""
+        return self._client.endpoint_service_list_endpoints(
+            project_id=teamspace_id,
+            cloudspace_id=studio_id,
+        ).endpoints
 
     def create_assistant(self, studio_id: str, teamspace_id: str, port: int, assistant_name: str) -> V1Assistant:
         target_teamspace = self._client.projects_service_get_project(teamspace_id)

lightning_sdk/api/teamspace_api.py

@@ -47,6 +47,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1UpstreamOpenAI,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
+from lightning_sdk.machine import Machine
 
 __all__ = ["TeamspaceApi"]
 
@@ -323,11 +324,47 @@ class TeamspaceApi:
         jobs = self._client.jobs_service_list_multi_machine_jobs(project_id=teamspace_id).multi_machine_jobs
         return apps, jobs
 
-    def list_machines(self, teamspace_id: str, cloud_account: str) -> List[V1ClusterAccelerator]:
-        response = self._client.cluster_service_list_project_cluster_accelerators(
-            project_id=teamspace_id, id=cloud_account
-        )
-        return response.accelerator
+    def list_machines(
+        self,
+        teamspace_id: str,
+        cloud_accounts: List[str],
+        machine: Optional[Machine] = None,
+        org_id: Optional[str] = None,
+    ) -> List[V1ClusterAccelerator]:
+        from lightning_sdk.api.cloud_account_api import CloudAccountApi
+
+        cloud_account_api = CloudAccountApi()
+        matched_accelerators = []
+        for ca in cloud_accounts:
+            try:
+                accelerators = cloud_account_api.list_cloud_account_accelerators(
+                    teamspace_id=teamspace_id,
+                    cloud_account_id=ca,
+                    org_id=org_id,
+                )
+                if not accelerators.accelerator:
+                    continue
+
+                if accelerators.accelerator:
+                    for cluster_machine in accelerators.accelerator:
+                        if not machine:
+                            matched_accelerators.append(cluster_machine)
+                            continue
+                        if (
+                            cluster_machine.resources.gpu == machine.accelerator_count
+                            or cluster_machine.resources.cpu == machine.accelerator_count
+                        ) and any(
+                            machine.family.lower() in s
+                            for s in (
+                                cluster_machine.slug,
+                                cluster_machine.slug_multi_cloud,
+                                cluster_machine.instance_id,
+                            )
+                        ):
+                            matched_accelerators.append(cluster_machine)
+            except Exception:
+                pass
+        return matched_accelerators
 
     def get_model(self, teamspace_id: str, model_id: Optional[str] = None, model_name: Optional[str] = None) -> V1Model:
         if model_id:

lightning_sdk/api/user_api.py

@@ -5,6 +5,7 @@ from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
     SecretServiceUpdateUserSecretBody,
     V1CloudSpace,
+    V1CreateProjectRequest,
     V1CreateUserSecretRequest,
     V1GetUserResponse,
     V1ListCloudSpacesResponse,
@@ -123,3 +124,7 @@ class UserApi:
         """
         pattern = r"^[A-Za-z_][A-Za-z0-9_]*$"
         return re.match(pattern, name) is not None
+
+    def create_teamspace(self, name: str) -> None:
+        """Creates a new teamspace."""
+        self._client.projects_service_create_project(body=V1CreateProjectRequest(name=name, display_name=name))

lightning_sdk/cli/legacy/download.py

@@ -242,7 +242,8 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
                     and (re.match(studio_name, st["name"]) or studio_name in st["name"]),
                     possible_studios,
                 )
-                if not len(possible_studios):
+                possible_studios = list(possible_studios)
+                if len(possible_studios) == 0:
                     raise ValueError(
                         f"Could not find Studio like '{studio}', please consider update your filtering pattern."
                     )

lightning_sdk/cli/legacy/studios_menu.py

@@ -31,7 +31,14 @@ class _StudiosMenu:
             title = "Please select a Studio of the following studios:"
 
         return TerminalMenu(
-            [f"{s['teamspace']}/{s['name']}" for s in possible_studios], title=title, clear_menu_on_exit=True
+            [
+                f"{s['org']}/{s['teamspace']}/{s['name']}"
+                if s.get("org") is not None
+                else f"{s['teamspace']}/{s['name']}"
+                for s in possible_studios
+            ],
+            title=title,
+            clear_menu_on_exit=True,
         )
 
     @staticmethod

lightning_sdk/job/base.py

@@ -87,11 +87,12 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "_BaseJob":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -192,7 +193,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
                 )
 
-            if entrypoint != "sh -c":
+            if entrypoint is not None:
                 raise ValueError("Specifying the entrypoint has no effect for jobs with Studio envs.")
 
         else:
@@ -218,6 +219,20 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
                 )
 
+            # command specified, so use the default entrypoint of sh -c
+            if command is not None and entrypoint is None:
+                entrypoint = "sh -c"
+
+            # entrypoint specifically set to empty string, so set to None here to fall back to the image entrypoint
+            elif entrypoint == "":  # noqa: SIM114
+                entrypoint = None
+
+            # entrypoint not specified, but also no command specified, so use the image entrypoint
+            elif entrypoint is None:
+                entrypoint = None
+
+            # all other cases, the entrypoint has been specifically set, so use it as is
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
         return inst._submit(
             machine=machine,
@@ -236,6 +251,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
             path_mappings=path_mappings,
             max_runtime=max_runtime,
             reuse_snapshot=reuse_snapshot,
+            scratch_disks=scratch_disks,
         )
 
     @abstractmethod
@@ -253,10 +269,11 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "_BaseJob":
         """Submit a new job to the Lightning AI platform.
 
@@ -289,7 +306,12 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 within it.
                 Note that the connection needs to be added to the teamspace already in order for it to be found.
                 Only supported for jobs with a docker image compute environment.
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             max_runtime: the duration (in seconds) for which to allocate the machine.

lightning_sdk/job/job.py

@@ -97,13 +97,14 @@ class Job(_BaseJob):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "Job":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -131,8 +132,10 @@ class Job(_BaseJob):
         cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
             Required if the registry is part of a cloud provider (e.g. ECR).
         entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-            just runs the provided command in a standard shell.
-            To use the pre-defined entrypoint of the provided image, set this to an empty string.
+            just runs the provided command in a standard shell if a command is provided.
+            If no command is provided, it will run the pre-defined entrypoint of the provided image.
+            To use the pre-defined entrypoint of the provided image with a specified command,
+            set this to an empty string.
             Only applicable when submitting docker jobs.
         path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
             represents the data-connection name and the path inside that connection.
@@ -172,6 +175,7 @@ class Job(_BaseJob):
             max_runtime=max_runtime,
             cluster=cluster,
             reuse_snapshot=reuse_snapshot,
+            scratch_disks=scratch_disks,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -191,12 +195,13 @@ class Job(_BaseJob):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "Job":
         """Submit a new job to the Lightning AI platform.
 
@@ -219,7 +224,12 @@ class Job(_BaseJob):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
@@ -255,6 +265,7 @@ class Job(_BaseJob):
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
             reuse_snapshot=reuse_snapshot,
+            scratch_disks=scratch_disks,
         )
         return self
 

lightning_sdk/job/v1.py

@@ -55,6 +55,7 @@ class _JobV1(_BaseJob):
         interruptible: bool = False,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "_BaseJob":
         """Start a new async workload from your studio.
 
@@ -107,10 +108,11 @@ class _JobV1(_BaseJob):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "_JobV1":
         """Submit a job to run on a machine.
 
@@ -125,7 +127,10 @@ class _JobV1(_BaseJob):
             image_credentials: The image credentials for the job (not supported).
             cloud_account_auth: Whether to use cloud account authentication for the job (not supported).
             entrypoint: The entrypoint of your docker container (not supported).
-                Defaults to `sh -c` which just runs the provided command in a standard shell.
+                Defaults to `sh -c` which just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: The mappings from data connection inside your container (not supported)

lightning_sdk/job/v2.py

@@ -1,3 +1,4 @@
+from pathlib import PurePath
 from typing import TYPE_CHECKING, Dict, Optional, Union
 
 from lightning_sdk.api.job_api import JobApiV2
@@ -48,12 +49,13 @@ class _JobV2(_BaseJob):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
         reuse_snapshot: bool = True,
+        scratch_disks: Optional[Dict[str, int]] = None,
     ) -> "_JobV2":
         """Submit a new job to the Lightning AI platform.
 
@@ -72,6 +74,14 @@ class _JobV2(_BaseJob):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
+                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                Only applicable when submitting docker jobs.
             artifacts_local: The path of inside the docker container, you want to persist images from.
                 CAUTION: When setting this to "/", it will effectively erase your container.
                 Only supported for jobs with a docker image compute environment.
@@ -90,6 +100,11 @@ class _JobV2(_BaseJob):
                 Defaults to 3h
             reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
                 submitted. Turning this off may result in longer job startup times. Defaults to True.
+            scratch_disks: Dictionary of scratch disks to add to the job. The keys are the path that the disk
+                will be mounted to, relative to the /teamspace/scratch directory. The values are the size of
+                the volume in GiB. For example, { "data": 100 } will add a 100GiB volume available under
+                /teamspace/scratch/data.
+
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -115,6 +130,30 @@ class _JobV2(_BaseJob):
             default_cloud_account=self._teamspace.default_cloud_account,
         )
 
+        if scratch_disks is not None and len(scratch_disks) > 0:
+            if studio is None:
+                raise ValueError("scratch_disks are only supported within a studio job")
+
+            if len(scratch_disks) > 5:
+                raise ValueError("scratch_disk may only contain up to 5 elements")
+
+            for path, size in scratch_disks.items():
+                if size > 50000:
+                    raise ValueError("scratch_disk size cannot exceed 50TiB")
+
+                path = PurePath(path)
+
+                if path.is_absolute():
+                    # For compatibility with Python 3.8, which doesn't provide
+                    # pathlib.PurePath.is_relative_to.
+                    try:
+                        path.relative_to("/teamspace/scratch")
+                    except ValueError:
+                        raise ValueError("scratch_disk paths must be relative to /teamspace/scratch") from None
+
+                if ".." in path.parts:
+                    raise ValueError("scratch_disk path cannot contain '..'")
+
         submitted = self._job_api.submit_job(
             name=self.name,
             command=command,
@@ -133,6 +172,7 @@ class _JobV2(_BaseJob):
             path_mappings=path_mappings,
             max_runtime=max_runtime,
             reuse_snapshot=reuse_snapshot,
+            scratch_disks=scratch_disks,
         )
         self._job = submitted
         self._name = submitted.name

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -42,6 +42,7 @@ from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api impor
 from lightning_sdk.lightning_cloud.openapi.api.incidents_service_api import IncidentsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.k8_s_cluster_service_api import K8SClusterServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.kubernetes_virtual_machine_service_api import KubernetesVirtualMachineServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
@@ -132,6 +133,7 @@ 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
@@ -140,11 +142,13 @@ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_server_alert_body import ClusterServiceCreateServerAlertBody
 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_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.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
@@ -191,6 +195,8 @@ from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_create_ku
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_render_kubernetes_template_body import K8SClusterServiceRenderKubernetesTemplateBody
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_report_k8_s_cluster_metrics_body import K8SClusterServiceReportK8SClusterMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_update_kubernetes_template_body import K8SClusterServiceUpdateKubernetesTemplateBody
+from lightning_sdk.lightning_cloud.openapi.models.kubernetes_virtual_machine_service_create_kubernetes_virtual_machine_body import KubernetesVirtualMachineServiceCreateKubernetesVirtualMachineBody
+from lightning_sdk.lightning_cloud.openapi.models.kubernetes_virtual_machine_service_update_kubernetes_virtual_machine_body import KubernetesVirtualMachineServiceUpdateKubernetesVirtualMachineBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_update_lightningapp_instance_body import LightningappInstanceServiceUpdateLightningappInstanceBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_update_lightningapp_instance_release_body import LightningappInstanceServiceUpdateLightningappInstanceReleaseBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_upload_lightningapp_instance_artifact_body import LightningappInstanceServiceUploadLightningappInstanceArtifactBody
@@ -295,6 +301,7 @@ 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
@@ -412,6 +419,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1Com
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
 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_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
@@ -481,6 +492,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_k
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restriction_response import V1DeleteClusterUsageRestrictionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_registry_response import V1DeleteContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
@@ -494,6 +506,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_incident_response im
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_template_response import V1DeleteKubernetesTemplateResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_virtual_machine_response import V1DeleteKubernetesVirtualMachineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_license_response import V1DeleteLicenseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_response import V1DeleteLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
@@ -569,6 +582,9 @@ 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
@@ -712,6 +728,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod import V1KubernetesPod
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template import V1KubernetesTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_virtual_machine import V1KubernetesVirtualMachine
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_config import V1KubevirtConfig
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_provider_configuration import V1KubevirtProviderConfiguration
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_vm_configuration import V1KubevirtVmConfiguration
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_vm_resources import V1KubevirtVmResources
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_license import V1License
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
@@ -764,6 +785,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_metrics_response import V1ListContainerMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_registries_response import V1ListContainerRegistriesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversation_message_actions_response import V1ListConversationMessageActionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversations_response import V1ListConversationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifacts_response import V1ListDataConnectionArtifactsResponse
@@ -783,7 +805,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_mm_ts_response import V1ListFilesystemMMTsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_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
@@ -803,6 +825,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kai_scheduler_queues_metrics_response import V1ListKaiSchedulerQueuesMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_pods_response import V1ListKubernetesPodsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_virtual_machines_response import V1ListKubernetesVirtualMachinesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_license_response import V1ListLicenseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_artifacts_response import V1ListLightningappInstanceArtifactsResponse
@@ -981,6 +1004,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_container_registry_credentials_response import V1RefreshContainerRegistryCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -1141,6 +1165,7 @@ 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